Làm cách nào để nhập mô-đun là __main__?

Question

Tôi có một mô-đun có các thông số

if __name__ == '__main__': 
    do stuff... 

thành ngữ mới nhất.

Tôi muốn nhập từ mô-đun khác và đánh lừa nó để chạy mã đó. Có cách nào để làm điều này không?

Tôi nên đề cập, vì lý do tôi sẽ không đi vào đây, tôi không ở vị trí để thay đổi mã trong mô-đun đã nhập. Tôi cần phải bằng cách nào đó sửa đổi các thủ tục nhập khẩu để tên của nó là chính khi nhập khẩu, có lẽ bằng cách sử dụng ihooks hoặc tương tự.

Answer 1

Có, thực thi kịch bản thay vì nhập khẩu nó. Nhưng tôi coi đây là một giải pháp cực kỳ đáng sợ.

Tuy nhiên mô hình lý tưởng sẽ là:

def do_stuff(): 
    ... stuff happens ... 

if __name__ == '__main__': 
    do_stuff() 

cách mà bạn có thể làm:

from mymodule import do_stuff 
do_stuff() 

EDIT: trả lời sau khi làm rõ về việc không thể chỉnh sửa mã module.

Tôi sẽ không bao giờ đề xuất điều này trong bất kỳ mã sản xuất nào, đây là giải pháp "sử dụng rủi ro riêng".

import mymodule 

with open(os.path.splitext(mymodule.__file__)[0] + ".py") as fh: 
    exec fh.read() 

Answer 2

Đặt mã đó vào một hàm và gọi mã đó từ mô-đun bạn cũng nhập mã đó vào.

def stuff(): 
    ... 

if __name__ == '__main__': 
    stuff() 

Và sau đó trong các mô-đun bạn đang nhập nó vào:

import module 
module.stuff() 

Answer 3

Đặt nó trong một hàm:

def _main(): 
    do stuff 

if __name__ == '__main__': 
    main() 

Answer 4

Mã trong một đoạn thơ chính thường bao giờ làm cho tinh thần để chạy trực tiếp. Nếu bạn muốn chạy nó sau đó sử dụng subprocess để chạy nó trong một trình thông dịch Python khác.

Answer 5

Dưới đây là một ví dụ về một mô-đun chính bằng Python:

#! /usr/bin/env python 
import sys 
import os 

def main(arg1, arg2, arg3): 
    print(arg1, arg2, arg3) 

if __name__ == "__main__": 
    main(*sys.argv) 

Nhưng bạn cũng có thể bao gồm

def main(): 
    #The module contains Python code specific to the library module, 
    #like tests, and follow the module with this: 

if __name__ == "__main__": 
    main(*sys.argv) 

trong bất kỳ mô-đun bạn muốn chạy như chính.

Ví dụ: nếu bạn có mô-đun thư viện, bạn luôn có thể sử dụng cấu trúc này để thực thi một cái gì đó cụ thể như kiểm tra.

Answer 6

Như đã nêu trong các câu trả lời khác, đây là một ý tưởng tồi và bạn nên giải quyết vấn đề theo cách khác.

Bất kể, cách Python có phải nó là như thế này:

import runpy 
result = runpy._run_module_as_main("your.module.name"))