

def upgradedb():
    logging.info('Creating tables')
    current_dir = os.path.dirname(os.path.abspath(__file__))
    package_dir = os.path.normpath(os.path.join(current_dir, '..'))
    directory = os.path.join(package_dir, 'migrations')
    config = Config(os.path.join(package_dir, 'alembic.ini'))
    config.set_main_option('script_location', directory)
    config.set_main_option('sqlalchemy.url', configuration.get('core', 'SQL_ALCHEMY_CONN'))
    command.upgrade(config, 'heads')

dataset/ETHPy150Open airbnb/airflow/airflow/utils/db.py upgradedb/original