我正在使用requests模块,但问题更为通用。
有没有办法在从导入的模块调用方法之前自动执行代码?
这会让我更容易编写代码。目前我担心锤击友好的网络服务,所以除非我能找到答案,否则我将不得不实施自己的控制。
我这样想:
import requests as requests2
...然后在代码中的某个位置,定义一个requests()
函数,带有特殊参数或一些神奇的未发现语法。在运行自己的代码之后,特殊的调制会将方法调用转发到别名为requests2
的实际模块。
可以这样做吗?
答案 0 :(得分:3)
你有正确的想法。为执行任何操作的“请求”创建代理,然后将方法调用转发给实际的requests
模块。例如
class RequestProxy(object):
def __init__(self):
import requests as _requests
self._requests = _requests
def __getattribute__(self, attr):
run_custom_code()
return getattr(self._requests, attr)
requests = RequestProxy()
答案 1 :(得分:2)
你的意思是在另一个模块中为每个函数添加装饰器?您可以使用类来模拟它:
class RequestsProxy(object):
def __init__(self):
self.special_sauce_decorator = special_sauce_indeed()
self._requests = __import__("requests")
def __getattr__(self, attrname):
val = getattr(self._requests, attrname)
if callable(val):
return self.special_sauce_decorator(val)
return val
使用示例:
>>> def special_sauce_indeed():
def decorator(f):
def wrapped(*args, **kwargs):
print 'wrapped'
return f(*args, **kwargs)
return wrapped
return decorator
>>> class OsProxy(object):
def __init__(self):
self.special_sauce_decorator = special_sauce_indeed()
self._requests = __import__("os")
def __getattr__(self, attrname):
val = getattr(self._requests, attrname)
if callable(val):
return self.special_sauce_decorator(val)
return val
>>> os = OsProxy()
>>> os.listdir(".")
wrapped
['DLLs', 'Doc', 'faiojerf.py', 'func_counter_test.py', 'include', 'inet_time.py', 'kcol.py', 'Lib', 'libs', 'LICENSE.txt', 'memoize_test.py', 'minpy.py', 'NEWS.txt', 'numpy-wininst.log', 'paren_test.py', 'PIL-wininst.log', 'psycopg2-wininst.log', 'python.exe', 'pythonw.exe', 'pywin32-wininst.log', 'README.txt', 'Removenumpy.exe', 'RemovePIL.exe', 'Removepsycopg2.exe', 'Removepywin32.exe', 'Removescipy.exe', 'Removesetuptools.exe', 'scipy-wininst.log', 'Scripts', 'setuptools-wininst.log', 'slots.py', 'so1.py', 'staticvar.py', 'summing.py', 'taojiwjiot.,py', 'tcl', 'templol.py', 'test.py', 'thunkify_test.py', 'TicketNumberGenerator.py', 'Tools', 'w9xpopen.exe', 'wordcount.py']