blob: 66068d2a9ed2a024781d32ba37d61a9b562ff563 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
try:
from logilab.astng import MANAGER, scoped_nodes, node_classes
PYLINT=0
except ImportError:
from astroid import MANAGER, scoped_nodes, node_classes
PYLINT=1
def ssl_transform(module):
if module.name == 'ssl':
for proto in ('SSLv23', 'TLSv1'):
module.locals['PROTOCOL_%s' % proto] = [node_classes.Const()]
def register(linter):
if PYLINT == 0:
MANAGER.register_transformer(ssl_transform)
else:
MANAGER.register_transform(scoped_nodes.Module, ssl_transform)
|
