如何在自定义SnowflakeHook中导出可持久生效的环境变量

os.environ 对环境变量的修改仅作用于当前Python进程及其派生的子进程，修改结果不会持久化到进程外。每次初始化SnowflakeHook如果是在新进程中执行，之前写入的DEV_DATABASE_CREATED变量会随着上一个进程的退出完全消失，所以会触发KeyError重复执行建库逻辑，最终抛出数据库已存在的报错。

• 方案1（最推荐）：直接查询Snowflake判断库是否存在
  完全不需要依赖本地标记，直接查询Snowflake实例中对应库是否存在，从根源避免标记和实际状态不一致的问题，示例逻辑：

env = os.environ['ENV']
user = os.environ['USER'].replace('.', '_')
if env == 'dev':
    logging.info('Development environment detected')
    dev_db_name = f'{env}_{user}'
    # 直接查询库是否存在，return_result根据Hook实际返回参数调整
    check_db = self.run(f"SHOW DATABASES LIKE '{dev_db_name}'", return_result=True)
    if not check_db:
        self.run(f"""CREATE DATABASE {dev_db_name} CLONE {self.database}""")
        logging.info(f'Dev database {dev_db_name} created')
    else:
        logging.info('Dev database already exists')
    self.run(f"USE DATABASE {dev_db_name};")
    self.run(sql, autocommit, parameters)

• 方案2：用本地文件持久化标记
  如果不想额外查询数据库，可以把创建标记存在本地文件中，只要文件不删除标记就不会丢失。如果是Docker部署，需要将文件所在目录挂载为持久化卷，避免容器重建后文件丢失：

import os
MARKER_FILE = ".dev_db_created"
env = os.environ['ENV']
user = os.environ['USER'].replace('.', '_')
if env == 'dev':
    logging.info('Development environment detected')
    dev_db_name = f'{env}_{user}'
    if os.path.exists(MARKER_FILE) and open(MARKER_FILE).read().strip() == 'True':
        logging.info('Dev database already exists')
    else:
        self.run(f"""CREATE DATABASE {dev_db_name} CLONE {self.database}""")
        with open(MARKER_FILE, 'w') as f:
            f.write('True')
        logging.info(f'Dev database {dev_db_name} created')
    self.run(f"USE DATABASE {dev_db_name};")
    self.run(sql, autocommit, parameters)

• 方案3：持久化到系统环境变量（不推荐）
  如果必须用环境变量实现，可以在创建库后将变量写入系统级或用户级的环境变量配置文件（比如Linux下的~/.bashrc、/etc/profile，Docker场景下写入.env文件供docker-compose加载），但这种方式修改后需要重新加载环境变量才会生效，适配性远不如前两种方案。

内容的提问来源于stack exchange，提问作者Javier Lopez Tomas