我正在尝试修补python的内置str
,以便跟踪所有str
分配的计数。我遇到了一些问题,并且想知道是否有人能够看到我做错了什么,或者是否可以通过python3中的猴子修补本地实现这一点? (以下在python 2.7.12中工作正常)
$ python
Python 3.5.2 (default, Nov 23 2017, 16:37:01)
[GCC 5.4.0 20160609] on linux
我首先尝试修补str
,好像它是一个函数:
def patch_str_allocations():
old_str = str
def mystr(*args, **kwargs):
return old_str(*args, **kwargs)
builtins.str = mystr
def test():
logger = logging.getLogger(__name__)
patch_str_allocations()
logger.debug(str('test'))
但是,当然这会使字符串用于isinstance
logger.debug(route)
File "/usr/lib/python3.5/logging/__init__.py", line 1267, in debug
self._log(DEBUG, msg, args, **kwargs)
File "/usr/lib/python3.5/logging/__init__.py", line 1403, in _log
fn, lno, func, sinfo = self.findCaller(stack_info)
File "/usr/lib/python3.5/logging/__init__.py", line 1360, in findCaller
filename = os.path.normcase(co.co_filename)
File "/home/ubuntu/.virtualenvs/papm/lib/python3.5/posixpath.py", line 52, in normcase
if not isinstance(s, (bytes, str)):
TypeError: isinstance() arg 2 must be a type or tuple of types
然后我尝试了一种基于类的方法:
class StrAllocator(str):
oldstr = None
def __new__(cls, *args, **kwargs):
return StrAllocator.oldstr.__new__(cls, *args, **kwargs)
@property
def __class__(self):
return str
def patch_str_allocations():
StrAllocator.oldstr = str
builtins.str = StrAllocator
在正常的str构造中,这工作正常但仍然遇到一些问题:
class StrAllocatorTestCase(unittest.TestCase):
def test_log(self):
t1 = str('t1')
logger = logging.getLogger(__name__)
patch_str_allocations()
t2 = str('t2')
print(type(t1))
print(type(t2))
print(isinstance(t1, str))
print(isinstance(t2, StrAllocator))
print(isinstance(t2, str))
logger.debug(str('test'))
$ nosetests tests.test_str_allocator:StrAllocatorTestCase.test_log -s
<class 'str'>
<class 'pythonapm.instruments.allocations.StrAllocator'>
False
True
True
E
======================================================================
ERROR: test_log (tests.instruments.test_str_allocator.StrAllocatorTestCase)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/vagrant_data/github.com/dm03514/python-apm/tests/instruments/test_str_allocator.py", line 30, in test_log
logger.debug(str('test'))
File "/usr/lib/python3.5/logging/__init__.py", line 1267, in debug
self._log(DEBUG, msg, args, **kwargs)
File "/usr/lib/python3.5/logging/__init__.py", line 1403, in _log
fn, lno, func, sinfo = self.findCaller(stack_info)
File "/usr/lib/python3.5/logging/__init__.py", line 1360, in findCaller
filename = os.path.normcase(co.co_filename)
File "/home/ubuntu/.virtualenvs/papm/lib/python3.5/posixpath.py", line 54, in normcase
"not '{}'".format(s.__class__.__name__))
TypeError: normcase() argument must be str or bytes, not 'str'
----------------------------------------------------------------------
Ran 1 test in 0.003s
与isstring
上的sre_compile一样检查
File "/home/ubuntu/.virtualenvs/papm/lib/python3.5/site-packages/flask/app.py", line 1250, in decorator [0/9965]
self.add_url_rule(rule, endpoint, f, **options)
File "/home/ubuntu/.virtualenvs/papm/lib/python3.5/site-packages/flask/app.py", line 66, in wrapper_func
return f(self, *args, **kwargs)
File "/home/ubuntu/.virtualenvs/papm/lib/python3.5/site-packages/flask/app.py", line 1216, in add_url_rule
self.url_map.add(rule)
File "/home/ubuntu/.virtualenvs/papm/lib/python3.5/site-packages/werkzeug/routing.py", line 1215, in add
rule.bind(self)
File "/home/ubuntu/.virtualenvs/papm/lib/python3.5/site-packages/werkzeug/routing.py", line 687, in bind
self.compile()
File "/home/ubuntu/.virtualenvs/papm/lib/python3.5/site-packages/werkzeug/routing.py", line 751, in compile
self._regex = re.compile(regex, re.UNICODE)
File "/home/ubuntu/.virtualenvs/papm/lib/python3.5/re.py", line 224, in compile
return _compile(pattern, flags)
File "/home/ubuntu/.virtualenvs/papm/lib/python3.5/re.py", line 292, in _compile
raise TypeError("first argument must be string or compiled pattern")
TypeError: first argument must be string or compiled pattern
任何人都能看到失踪的东西吗? (除了我对描述符和python类的理解:p)
从REPL上面的示例Works,但它在nose和unittests的上下文中不起作用......
⟫ ipython
Python 3.5.2 (default, Nov 23 2017, 16:37:01)
Type 'copyright', 'credits' or 'license' for more information
IPython 6.4.0 -- An enhanced Interactive Python. Type '?' for help.
In [1]: import logging
In [2]: import builtins
In [3]: class StrAllocator(str):
...: oldstr = None
...:
...: def __new__(cls, *args, **kwargs):
...: return StrAllocator.oldstr.__new__(cls, *args, **kwargs)
...:
...: @property
...: def __class__(self):
...: return str
...:
...:
In [4]: def patch_str_allocations(): [6/9733]
...: StrAllocator.oldstr = str
...: builtins.str = StrAllocator
...:
In [5]: def test_log():
...: t1 = str('t1')
...: logger = logging.getLogger(__name__)
...: patch_str_allocations()
...: t2 = str('t2')
...: print(type(t1))
...: print(type(t2))
...: print(isinstance(t1, str))
...: print(isinstance(t2, StrAllocator))
...: print(isinstance(t2, str))
...: logger.debug(str('test'))
...:
In [6]: test_log()
<class 'str'>
<class '__main__.StrAllocator'>
False
True
True
答案 0 :(得分:3)
如果您坚持使用自己的功能对内置str
进行猴子修补,为什么不对isinstance()
进行修补,以确保它将您的功能视为内置str
?类似的东西:
def patch_str_allocations():
old_str = str
old_isinstance = builtins.isinstance
def mystr(*args, **kwargs):
return old_str(*args, **kwargs)
def my_isinstance(o, t):
if t is mystr:
t = old_str
return old_isinstance(o, t)
builtins.str = mystr
builtins.isinstance = my_isinstance
您可能还想检查t
的{{1}}是否为元组,并对其进行迭代,以确保将my_isinstance()
替换为mystr
。