mirror of
https://github.com/yyhuni/xingrin.git
synced 2026-02-01 20:23:23 +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()
|