在配置文件中使用主要 Python 脚本的变量

Use variables of main Python script in a config file

我想在主 Python 脚本中间使用配置文件。 在这个文件中,有一些变量已经在主脚本中初始化了。

主脚本:

# -*- coding: utf-8 -*-

import imp
import optparse
from time import strftime, strptime
from datetime import date, timedelta

[...]

date_mois = date_de_ref.strftime("%m")
date_annee = date_de_ref.strftime("%Y")
date_moins2j = (date_de_ref - timedelta(days=2)).strftime("%d/%m/%Y")
date_moins7j = (date_de_ref - timedelta(days=7)).strftime("%d/%m/%Y")

imp.load_source("conf_file", "part_1.config")
import conf_file

for task in conf_file.list_tasks:
    print task[1]

print "The end!"

"part_1.config":

list_tasks = [
    ["B", "BAT001", "Info batch 1"],
    ["B", "BAT002 DEBUT=21/01/2015", "Info batch 2"],
    ["B", "BAT003 FIN=" + date_moins2j, "Info batch 3"],
]

如果我这样执行,我得到:

Traceback (most recent call last):
  File "ordo_na.py", line 48, in <module>
    imp.load_source("conf_file", "part_1.config")
  File "part_1.config", line 9, in <module>
    ["B", "BAT003 FIN=" + date_moins2j, ""],
NameError: name 'date_moins2j' is not defined

我试过:

import ordo_na

list_tasks = [
    ["B", "BAT001", ""],
    ["B", "BAT002 DEBUT=21/13/12", ""],
    ["B", "BAT003 FIN=" + ordo_na.date_moins2j, ""],
]

主脚本好像是这样执行了两次

如何在配置文件中使用主脚本中的变量?
我想尽可能避免使用 import ordo_na 行,并尽可能保持配置文件最干净。

顺便说下我用的是Python2.6.6(无法升级到较新的版本)

谢谢!

模块是独立的,您无法从导入模块轻松访问导入模块的变量(我认为您正在尝试这样做)。

考虑:

part_1.config

def get_tasks(date_moins2j):

    list_tasks = [
        ["B", "BAT001", ""],
        ["B", "BAT002 DEBUT=21/13/12", ""],
        ["B", "BAT003 FIN=" + date_moins2j, ""],
    ]

    return list_tasks

主脚本

import imp

imp.load_source("conf_file", "part_1.config")
import conf_file

print( conf_file.get_tasks("foo") )

或者,您可以将 part_1.config 重命名为 my_config.py 之类的名称,然后只需使用 import:

主脚本

import my_config as conf_file

pprint.pprint( conf_file.get_tasks("foo") )

如果您不想处理显式传递参数,您总是可以传递导入模块(主脚本)的全局字典,然后访问它,那么您的文件将如下所示:

part_1.config

def get_tasks(g):

    list_tasks = [
        ["B", "BAT001", ""],
        ["B", "BAT002 DEBUT=21/13/12", ""],
        ["B", "BAT003 FIN=" + g['date_moins2j'], ""],
    ]

    return list_tasks

主脚本

import imp

imp.load_source("conf_file", "part_1.config")
import conf_file

date_moins2j = "foo"

pprint( conf_file.get_tasks(globals()) )