40 lines
831 B
Python
40 lines
831 B
Python
#
|
|
# started from http://www.logilab.org/blogentry/78354
|
|
#
|
|
|
|
from logilab.astng import MANAGER
|
|
from logilab.astng.builder import ASTNGBuilder
|
|
|
|
def hashlib_transform(module):
|
|
if module.name == 'hashlib':
|
|
fake = ASTNGBuilder(MANAGER).string_build('''
|
|
|
|
class fakehash(object):
|
|
digest_size = -1
|
|
def __init__(self, value): pass
|
|
def digest(self):
|
|
return u''
|
|
def hexdigest(self):
|
|
return u''
|
|
def update(self, value): pass
|
|
|
|
class md5(fakehash):
|
|
pass
|
|
|
|
class sha1(fakehash):
|
|
pass
|
|
|
|
class sha256(fakehash):
|
|
pass
|
|
|
|
''')
|
|
for hashfunc in ('sha256', 'sha1', 'md5'):
|
|
module.locals[hashfunc] = fake.locals[hashfunc]
|
|
|
|
def register(linter):
|
|
"""called when loaded by pylint --load-plugins, register our tranformation
|
|
function here
|
|
"""
|
|
MANAGER.register_transformer(hashlib_transform)
|
|
|