mirror of
https://github.com/yyhuni/xingrin.git
synced 2026-02-08 15:34:59 +08:00
43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
|
"""
|
|||
|
|
Prefect Flow Django 环境初始化模块
|
|||
|
|
|
|||
|
|
在所有 Prefect Flow 文件开头导入此模块即可自动配置 Django 环境
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
|
|||
|
|
def setup_django_for_prefect():
|
|||
|
|
"""
|
|||
|
|
为 Prefect Flow 配置 Django 环境
|
|||
|
|
|
|||
|
|
此函数会：
|
|||
|
|
1. 添加项目根目录到 Python 路径
|
|||
|
|
2. 设置 DJANGO_SETTINGS_MODULE 环境变量
|
|||
|
|
3. 调用 django.setup() 初始化 Django
|
|||
|
|
|
|||
|
|
使用方式：
|
|||
|
|
from apps.common.prefect_django_setup import setup_django_for_prefect
|
|||
|
|
setup_django_for_prefect()
|
|||
|
|
"""
|
|||
|
|
# 获取项目根目录（backend 目录）
|
|||
|
|
current_dir = os.path.dirname(os.path.abspath(__file__))
|
|||
|
|
backend_dir = os.path.join(current_dir, '../..')
|
|||
|
|
backend_dir = os.path.abspath(backend_dir)
|
|||
|
|
|
|||
|
|
# 添加到 Python 路径
|
|||
|
|
if backend_dir not in sys.path:
|
|||
|
|
sys.path.insert(0, backend_dir)
|
|||
|
|
|
|||
|
|
# 配置 Django
|
|||
|
|
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'config.settings')
|
|||
|
|
|
|||
|
|
# 初始化 Django
|
|||
|
|
import django
|
|||
|
|
django.setup()
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 自动执行初始化（导入即生效）
|
|||
|
|
setup_django_for_prefect()
|