在模块方法前自动执行代码

时间:2012-08-13 22:41:58

标签: python python-requests

我正在使用requests模块,但问题更为通用。

有没有办法在从导入的模块调用方法之前自动执行代码?

这会让我更容易编写代码。目前我担心锤击友好的网络服务,所以除非我能找到答案,否则我将不得不实施自己的控制。


我这样想:

import requests as requests2

...然后在代码中的某个位置,定义一个requests()函数,带有特殊参数或一些神奇的未发现语法。在运行自己的代码之后,特殊的调制会将方法调用转发到别名为requests2的实际模块。

可以这样做吗?

2 个答案:

答案 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']