Использование собственных объектов модуля в __main__.py - PullRequest
Новая
       4

Использование собственных объектов модуля в __main__.py

41 голосов
/ 05 августа 2010

Я пытаюсь получить доступ к данным модуля изнутри его __main__.py.

Структура выглядит следующим образом: 

mymod/
    __init__.py
    __main__.py

Теперь, если я выставлю переменную в __init__.py следующим образом: 

__all__ = ['foo']
foo = {'bar': 'baz'}

Как получить доступ к foo из __main__.py?

Ответы [ 5 ]

23 голосов
/ 05 августа 2010

Вам нужно либо иметь пакет уже в sys.path, добавить каталог, содержащий mymod в sys.path в __main__.py, либо использовать переключатель -m.

Для добавления mymod путь будет выглядеть примерно так (в __main__.py): 

import sys
import os
path = os.path.dirname(sys.modules[__name__].__file__)
path = os.path.join(path, '..')
sys.path.insert(0, path)
from myprog import function_you_referenced_from_init_file

При использовании переключателя -m будет выглядеть: 

python -m mymod

См. этот ответ для дальнейшего обсуждения.

4 голосов
/ 19 июля 2012

Проблема, с которой я сталкиваюсь чаще всего, заключается в том, что я часто хочу запустить файл __init__.py в качестве сценария для тестирования функций, но их не следует запускать при загрузке пакета.Существует полезный обходной путь для различных путей выполнения между python <package>/__init__.py и python -m <package>.

  • $ python -m <module> выполняет <package>/__main__.py.__init__.py не загружен.
  • $ python <package>/__init__.py просто выполняет скрипт __init__.py как обычный скрипт.


Проблема

Когда мы хотим, чтобы __init__.py имел предложение if __name__ == '__main__': ..., которое использует материал из __main__.py.Мы не можем импортировать __main__.py, потому что он всегда будет импортировать __main__.pyc с пути интерпретатора.( Если только … мы прибегаем к хакам импорта с абсолютным путем, которые могут вызвать много других беспорядков).


Решение Решение:)

Используйте два файла сценария для модуля __main__: 

<package>/
         __init__.py
         __main__.py
         main.py

# __init__.py

# ...
# some code, including module methods and __all__ definitions

__all__ = ['foo', 'bar']
bar = {'key': 'value'}
def foo():
    return bar
# ...
if __name__ == '__main__':
    from main import main
    main.main()

# __main__.py

# some code...such as:
import sys
if (len(sys.argv) > 1 and sys.argv[1].lower() == 'option1'):
    from main import main()
    main('option1')
elif (len(sys.argv) > 1 and sys.argv[1].lower() == 'option2'):
    from main import main()
    main('option2')
else:
    # do something else?
    print 'invalid option. please use "python -m <package> option1|option2"'

# main.py

def main(opt = None):
    if opt == 'option1':
        from __init__ import foo
        print foo()
    elif opt == 'option2':
        from __init__ import bar
        print bar.keys()
    elif opt is None:
        print 'called from __init__'

Импорт в main.py, вероятно, не идеален в том случае, если мы запускаем из __init__.py, поскольку мы перезагружаем их в локальную область видимости другого модуля, несмотря на то, что уже загружаем их в __init__.py, но явная загрузка должнаИзбегайте круговой нагрузки.Если вы снова загрузите весь модуль __init__ в свой main.py, он не будет загружен как __main__, поэтому должен быть безопасным для круговой загрузки.

2 голосов
/ 05 августа 2010

Модуль __init__ из пакета действует как элемент самого пакета, поэтому объекты импортируются непосредственно из mymod: 

from mymod import foo

Или 

from . import foo

Если вы хотите быть кратким, тогда читайте о относительном импорте . Вам, как всегда, нужно убедиться, что вы не вызываете модуль, например, как mymod/__main__.py, поскольку это не позволит Python обнаруживать mymod как пакет. Вы можете рассмотреть distutils.

1 голос
/ 10 мая 2012

Если вы запустите модуль с python -m mymod, то код в __main__.py сможет импортировать из остальной части модуля без добавления модуля в sys.path.

0 голосов
/ 19 марта 2012

Структура каталога модуля выглядит следующим образом: 

py/
   __init__.py
   __main__.py

__ init __. Py 

#!/usr/bin/python3
#
# __init__.py
#

__all__ = ['foo']
foo = {'bar': 'baz'}
info = { "package": __package__,
         "name": __name__,
         "locals": [x for x in locals().copy()] }
print(info)

__ main __. Py 

#!/usr/bin/python3
#
# __main__.py
#

info = { "package": __package__,
         "name": __name__,
         "locals": [x for x in locals().copy()] }
print(info)
from . import info as pyinfo
print({"pyinfo: ": pyinfo})

Выполнить модуль как скрипт, используяфлаг -m

$ python -m py

# the printout from the 'print(info)' command in __init__.py
{'name': 'py', 'locals': ['__all__', '__builtins__', '__file__', '__package__', '__path__', '__name__', 'foo', '__doc__'], 'package': None}
# the printout from the 'print(info)' command in __main__.py
{'name': '__main__', 'locals': ['__builtins__', '__name__', '__file__', '__loader__', '__doc__', '__package__'], 'package': 'py'}
# the printout from the 'print(pyinfo)' command in __main__.py
{'pyinfo: ': {'name': 'py', 'locals': ['__all__', '__builtins__', '__file__', '__package__', '__path__', '__name__', 'foo', '__doc__'], 'package': None}}
...