检查脚本是独立运行还是从django界面运行

时间:2014-07-01 10:03:27

标签: python django command-line-interface

ALL,

我开发了一个从CLI运行的脚本。脚本的目的是从csv文件加载数据并将其推送到mySQL。 现在,修改脚本以便从Django运行,以便用户在浏览器中单击一个名为"浏览..."的按钮,选择csv文件,然后将读取文件和数据用我的脚本推送到mySQL。

现在我的问题是,对于数据库,有一个表应该包含2个字段:一个是本地文件名,另一个是用户机器上用于Web界面的文件名(django)。

所以我的问题很简单:是否可以确定我是从CLI还是从Django Web应用程序启动脚本?

谢谢。

1 个答案:

答案 0 :(得分:0)

在python中,如果当前脚本作为主程序运行,则特殊的python变量__name__将被设置为字符串"__main__"

因此,要检测脚本是否直接从命令行运行,例如python my_script.py,只需使用if语句,例如

if __name__ == '__main__':
    do_stuff()