Чи можу я в Python викликати main () імпортованого модуля?

У Python у мене є модуль myModule.py, де я визначаю кілька функцій і main () , який приймає кілька аргументів командного рядка.

Зазвичай я називаю це main () із сценарію bash. Тепер я хотів би помістити все в невеликий пакет , тому я подумав, що, можливо, я можу перетворити мій простий скрипт bash на скрипт Python і помістити його в пакет.

Отже, як мені насправді викликати функцію main () myModule.py з функції main () MyFormerBashScript.py? Чи можу я це зробити? Як йому передати будь-які аргументи ?

Це просто функція. Імпортуйте його і зателефонуйте:

import myModule

myModule.main()

Якщо вам потрібно проаналізувати аргументи, у вас є два варіанти:

• Проаналізуйте їх main(), але передайте sys.argvяк параметр (весь код нижче в тому самому модулі myModule):

  def main(args):
      # parse arguments using optparse or argparse or what have you
  
  if __name__ == '__main__':
      import sys
      main(sys.argv[1:])
  

  Тепер ви можете імпортувати та дзвонити myModule.main(['arg1', 'arg2', 'arg3'])з іншого модуля.

• Є main()приймати параметри, які вже розібрані (знову весь код в myModuleмодулі):

  def main(foo, bar, baz='spam'):
      # run with already parsed arguments
  
  if __name__ == '__main__':
      import sys
      # parse sys.argv[1:] using optparse or argparse or what have you
      main(foovalue, barvalue, **dictofoptions)
  

  і імпортувати та викликати в myModule.main(foovalue, barvalue, baz='ham')іншому місці та передаючи аргументи python за потреби.

Фокус тут полягає у виявленні, коли ваш модуль використовується як скрипт; коли ви запускаєте файл python як основний скрипт ( python filename.py), жоден importоператор не використовується, тому python викликає цей модуль "__main__". Але якщо той самий filename.pyкод трактується як модуль ( import filename), тоді python використовує його як ім'я модуля. В обох випадках змінна __name__встановлена, і тестування на неї повідомляє, як запускався ваш код.

Відповідь Мартієна має сенс, але в ньому бракувало чогось вирішального, що може здатися очевидним для інших, але мені було важко це зрозуміти.

У версії, де ви використовуєте argparse, вам потрібно мати цей рядок в основному тексті.

args = parser.parse_args(args)

Зазвичай, коли ви використовуєте argparse просто в сценарії, ви просто пишете

args = parser.parse_args()

і parse_args знаходять аргументи з командного рядка. Але в цьому випадку основна функція не має доступу до аргументів командного рядка, тому вам слід сказати argparse, якими є аргументи.

Ось приклад

import argparse
import sys

def x(x_center, y_center):
    print "X center:", x_center
    print "Y center:", y_center

def main(args):
    parser = argparse.ArgumentParser(description="Do something.")
    parser.add_argument("-x", "--xcenter", type=float, default= 2, required=False)
    parser.add_argument("-y", "--ycenter", type=float, default= 4, required=False)
    args = parser.parse_args(args)
    x(args.xcenter, args.ycenter)

if __name__ == '__main__':
    main(sys.argv[1:])

Якщо припустити, що ви вказали цю назву mytest.py, щоб запустити її, ви можете зробити будь-яку з них із командного рядка

python ./mytest.py -x 8
python ./mytest.py -x 8 -y 2
python ./mytest.py 

який повертається відповідно

X center: 8.0
Y center: 4

або

X center: 8.0
Y center: 2.0

або

X center: 2
Y center: 4

Або якщо ви хочете запустити з іншого сценарію python, ви можете це зробити

import mytest
mytest.main(["-x","7","-y","6"]) 

який повертається

X center: 7.0
Y center: 6.0

Це залежить. Якщо основний код захищений ifяк у:

if __name__ == '__main__':
    ...main code...

тоді ні, ви не можете змусити Python виконати це, оскільки ви не можете впливати на автоматичну змінну __name__.

Але коли весь код знаходиться у функції, тоді це можливо. Спробуйте

import myModule

myModule.main()

Це працює навіть тоді, коли модуль захищає себе за допомогою __all__.

from myModule import *може не зробити вас mainвидимим, тому вам дійсно потрібно імпортувати сам модуль.

У мене була така сама потреба у використанні argparse. Річ у parse_argsфункції argparse.ArgumentParserекземпляра об'єкта неявно бере свої аргументи за замовчуванням з sys.args. Обійстя, слідуючи рядку Мартіна, полягає у тому, щоб зробити це явним, тому ви можете змінити аргументи, які ви передаєте parse_argsяк бажання.

def main(args):
    # some stuff
    parser = argparse.ArgumentParser()
    # some other stuff
    parsed_args = parser.parse_args(args)
    # more stuff with the args

if __name__ == '__main__':
    import sys
    main(sys.argv[1:])

Ключовим моментом є передача аргументів для parse_argsфункціонування. Пізніше, щоб використовувати головне, ви просто робите те, що розповідає Мартін.

Відповідь, яку я шукав, отримав відповідь тут: Як використовувати python argparse з аргументами, відмінними від sys.argv?

Якщо main.pyі parse_args()написано таким чином, то синтаксичний аналіз можна зробити красиво

# main.py
import argparse
def parse_args():
    parser = argparse.ArgumentParser(description="")
    parser.add_argument('--input', default='my_input.txt')
    return parser

def main(args):
    print(args.input)

if __name__ == "__main__":
    parser = parse_args()
    args = parser.parse_args()
    main(args)

Потім ви можете викликати main()та аналізувати аргументи з parser.parse_args(['--input', 'foobar.txt'])ним в іншому скрипті python:

# temp.py
from main import main, parse_args
parser = parse_args()
args = parser.parse_args([]) # note the square bracket
# to overwrite default, use parser.parse_args(['--input', 'foobar.txt'])
print(args) # Namespace(input='my_input.txt')
main(args)

Припускаючи, що ви також намагаєтесь передати аргументи командного рядка.

import sys
import myModule


def main():
    # this will just pass all of the system arguments as is
    myModule.main(*sys.argv)

    # all the argv but the script name
    myModule.main(*sys.argv[1:])