改写从导入模块

Question

导入模块的内置功能，我有一个应用程序，我想，当情况发生时替换某些功能，例如：

condition_check.py：

Flag = True 
import ctypes # An included library with Python install. 
import inspect 
def MsgBox(msg): 
    ctypes.windll.user32.MessageBoxA(0, msg, 'MsgBox', 1) 
def check(): 
    global print 
    if Flag: 
     def print(msg): 
       MsgBox(msg) 
    else: 
     pass 

为主。潘岳：

## works 
from condition_check import * 
MsgBox('this is msgbox') 
print('this is a print') 

## does not work 
import condition_check 
condition_check.MsgBox('this is msgbox') 
print('this is a print') 

我明白condition_check.py是压倒一切的自己的print代替main.py“ s print。我相信检查库可用于此目的，但我无法查找一个示例。

Answer 1

我假设你正在使用Python 3.如果你是，你只需要设置内置模块的属性。

import builtins 
import ctypes 

original = builtins.print 
Flag = True 

def MsgBox(msg): 
    ctypes.windll.user32.MessageBoxA(0, msg, 'MsgBox', 1) 

def check(): 
    if Flag: 
     builtins.print = MsgBox 
    else: 
     builtins.print = original 

我会注意，但是，有两件事情：

1. Flag是不是有两个原因一个好听的名字：，它是不是在所有的描述。一面旗帜仅仅意味着它是True或False;它没有说明它的用途。 Python的官方风格指南（PEP 8）推荐使用snake_case，而不使用PascalCase作为常规变量。 PascalCase只能用于类。

2. PEP 8不推荐使用通配符导入（from <module> import *），因为它们使命名空间中的名称不清楚，导致读者和自动化工具混淆。 （几乎与Imports一节中的确切引用相同。）

3. 您不需要覆盖print函数。更好的方法是重写sys.stdout到控制通向哪里流：

   import ctypes 
   import sys 
   
   def MsgBox(msg): 
       ctypes.windll.user32.MessageBoxA(0, msg, 'MsgBox', 1) 
   
   class Printer: 
       def __init__(self, original, alternate, use_alternate): 
        self.original = original 
        self.alternate = alternate 
        self.use_alternate = use_alternate 
   
       def write(self, msg): 
        if self.use_alternate: 
         return self.alternate(msg) 
        return self.original(msg) 
   
   sys.stdout = printer = Printer(sys.stdout.write, MsgBox, True) 
   

   你的标志，然后printer.use_alternate。除了易于控制之外，这也与Python 2兼容，即使Python 2 print是一个声明。这确实有保持该print添加新行的轻微的缺点，但一个随时可以使用的东西的alternate像lambda msg: MsgBox(msg.strip())