jprdonnelly/airbyte
1
0
Fork 0
mirror of synced 2025-12-26 14:02:10 -05:00
Code Issues Packages Projects Releases Wiki Activity

read module env variables in main (#735)

Browse Source
This commit is contained in:
Charles
2020-10-29 10:13:46 -07:00
committed by GitHub
parent f408bce2dc
commit 2f238d6d35
2 changed files with 11 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
airbyte-integrations/bases/base-python/base_python/entrypoint.py
View File

@@ -33,11 +33,6 @@ from airbyte_protocol import AirbyteMessage, Status, Type
from .integration import ConfigContainer, Source
from .logger import AirbyteLogger
impl_module = os.environ.get("AIRBYTE_IMPL_MODULE", Source.__module__)
impl_class = os.environ.get("AIRBYTE_IMPL_PATH", Source.__name__)
module = importlib.import_module(impl_module)
impl = getattr(module, impl_class)
logger = AirbyteLogger()
@@ -130,10 +125,16 @@ class AirbyteEntrypoint(object):
def launch(source, args):
AirbyteEntrypoint(source).start(args)
def main():
impl_module = os.environ.get("AIRBYTE_IMPL_MODULE", Source.__module__)
impl_class = os.environ.get("AIRBYTE_IMPL_PATH", Source.__name__)
module = importlib.import_module(impl_module)
impl = getattr(module, impl_class)
# set up and run entrypoint
source = impl()