def main()

in run_goose.py [0:0]
44 lines of code
9 McCabe index (conditional complexity)

def main():
    parser = argparse.ArgumentParser(description="Run goose migrations")
    parser.add_argument("--only-if-empty-db", default=False, action="store_true")
    parser.add_argument("--wait", type=int, default=30, help="Wait for connection for X seconds")
    args = parser.parse_args()

    db_connection_string = f'postgresql://{quote(os.environ["MF_METADATA_DB_USER"])}:'\
        f'{quote(os.environ["MF_METADATA_DB_PSWD"])}@{os.environ["MF_METADATA_DB_HOST"]}:'\
        f'{os.environ["MF_METADATA_DB_PORT"]}/{os.environ["MF_METADATA_DB_NAME"]}'

    ssl_mode = os.environ.get("MF_METADATA_DB_SSL_MODE")
    ssl_cert_path = os.environ.get("MF_METADATA_DB_SSL_CERT_PATH")
    ssl_key_path = os.environ.get("MF_METADATA_DB_SSL_KEY_PATH")
    ssl_root_cert_path = os.environ.get("MF_METADATA_DB_SSL_ROOT_CERT")

    if ssl_mode in ['allow', 'prefer', 'require', 'verify-ca', 'verify-full']:
        ssl_query = f'sslmode={ssl_mode}'
        if ssl_cert_path is not None:
            ssl_query = f'{ssl_query}&sslcert={ssl_cert_path}'
        if ssl_key_path is not None:
            ssl_query = f'{ssl_query}&sslkey={ssl_key_path}'
        if ssl_root_cert_path is not None:
            ssl_query = f'{ssl_query}&sslrootcert={ssl_root_cert_path}'
    else:
        ssl_query = f'sslmode=disable'

    db_connection_string = f'{db_connection_string}?{ssl_query}'

    if args.wait:
        wait_for_postgres(db_connection_string, timeout_seconds=args.wait)

    if args.only_if_empty_db:
        if check_if_goose_table_exists(db_connection_string):
            print(
                f"Skipping migrations since --only-if-empty-db flag is used",
                file=sys.stderr,
            )
            sys.exit(0)

    p = Popen(
        [
            "goose",
            "-dir",
            "/root/services/migration_service/migration_files/",
            "postgres",
            db_connection_string,
            "up",
        ]
    )
    if p.wait() != 0:
        raise Exception("Failed to run initial migration")