python - scrapy 可以在设置中设置自定义变量吗？

Question

我想在 settings.py 中设置一个变量 ENV，例如:

import sys, os
path_django_site = os.path.join(os.path.dirname(__file__), "../../")  
sys.path.append(path_django_site)
os.environ['DJANGO_SETTINGS_MODULE'] = 'pjt1.settings.dev'  
ENV = os.environ['DJANGO_SETTINGS_MODULE']

所以在我的爬虫项目中，我可以这样写代码:

if ENV =='pjt1.settings.dev':
    obj = Group.objects.filter(id__lte=10)
    print 'this is test'

elif ENV =='pjt.settings.prod':
    obj = Group.objects.all()
    print 'This is normal environvent'

有可能吗？请指导我。谢谢你。

Answer 1

我不确定我是否理解实际问题，但您可以在 settings.py 中设置任何变量并像这样使用它:

# your wsgi file probably includes something like this
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "pjt1.settings.dev")

# settings.py
ENV = os.environ['DJANGO_SETTINGS_MODULE']

# other files
from django.conf import settings
if settings.ENV == 'pjt1.settings.dev':
    ...