Как использовать «if __name__ == '__main__':» в файле .bat? - PullRequest
0 голосов
/ 31 января 2020

У меня есть многопроцессорная функция, которая прекрасно работает в Spyder, например:

if __name__ == '__main__':
    global results
    p = Pool(20)
    results = p.map(get_api_item, date_list)
    p.terminate()
    p.join()

result = pd.concat(results)

path = r'<path>'
result.to_csv(os.path.join(path,r'api_item.csv'), index=False)

Когда я запускаю этот файл .py в файле .bat ...

@echo off
"E:\Python\Anaconda3\python.exe" "E:\Projects\api.py" %*
pause

. ..Я получаю эту ошибку:

Traceback (most recent call last):
  File "<string>", line 1, in <module>
  File "E:\Python\Anaconda3\lib\multiprocessing\spawn.py", line 105, in spawn_ma
in
    exitcode = _main(fd)
  File "E:\Python\Anaconda3\lib\multiprocessing\spawn.py", line 114, in _main
    prepare(preparation_data)
  File "E:\Python\Anaconda3\lib\multiprocessing\spawn.py", line 225, in prepare
    _fixup_main_from_path(data['init_main_from_path'])
  File "E:\Python\Anaconda3\lib\multiprocessing\spawn.py", line 277, in _fixup_m
ain_from_path
    run_name="__mp_main__")
  File "E:\Python\Anaconda3\lib\runpy.py", line 263, in run_path
    pkg_name=pkg_name, script_name=fname)
  File "E:\Python\Anaconda3\lib\runpy.py", line 96, in _run_module_code
    mod_name, mod_spec, pkg_name, script_name)
  File "E:\Python\Anaconda3\lib\runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "<path to api>\api.py", line 29, in <module>
    result = pd.concat(results)
NameError: name 'results' is not defined

Я думаю, это потому, что служба (файл .bat), вероятно, не вызывает ее __main__. Как мне исправить это?

1 Ответ

1 голос
/ 31 января 2020

Пожалуйста, прочитайте и поймите предупреждение в документах mutiprocessing (https://docs.python.org/3.8/library/multiprocessing.html), начиная с:

Безопасный импорт основного модуля

Убедитесь, что основной модуль может быть безопасно импортирован новым интерпретатором Python, не вызывая непреднамеренных побочных эффектов (таких как запуск нового процесса).

Ваш модуль будет импортирован другими процессами, поэтому вы не сможете код на глобальном (самом внешнем) уровне в зависимости от кода в вашем блоке __main__.

...