ALL,
我开发了一个从CLI运行的脚本。脚本的目的是从csv文件加载数据并将其推送到mySQL。 现在,修改脚本以便从Django运行,以便用户在浏览器中单击一个名为"浏览..."的按钮,选择csv文件,然后将读取文件和数据用我的脚本推送到mySQL。
现在我的问题是,对于数据库,有一个表应该包含2个字段:一个是本地文件名,另一个是用户机器上用于Web界面的文件名(django)。
所以我的问题很简单:是否可以确定我是从CLI还是从Django Web应用程序启动脚本?
谢谢。
答案 0 :(得分:0)
在python中,如果当前脚本作为主程序运行,则特殊的python变量__name__
将被设置为字符串"__main__"
。
因此,要检测脚本是否直接从命令行运行,例如python my_script.py
,只需使用if
语句,例如
if __name__ == '__main__':
do_stuff()