OLD | NEW |
1 from astroid import MANAGER | 1 from astroid import MANAGER, register_module_extender |
2 from astroid.builder import AstroidBuilder | 2 from astroid.builder import AstroidBuilder |
3 | 3 |
4 def mechanize_transform(module): | 4 def mechanize_transform(): |
5 fake = AstroidBuilder(MANAGER).string_build(''' | 5 return AstroidBuilder(MANAGER).string_build(''' |
6 | 6 |
7 class Browser(object): | 7 class Browser(object): |
8 def open(self, url, data=None, timeout=None): | 8 def open(self, url, data=None, timeout=None): |
9 return None | 9 return None |
10 def open_novisit(self, url, data=None, timeout=None): | 10 def open_novisit(self, url, data=None, timeout=None): |
11 return None | 11 return None |
12 def open_local_file(self, filename): | 12 def open_local_file(self, filename): |
13 return None | 13 return None |
14 | 14 |
15 ''') | 15 ''') |
16 module.locals['Browser'] = fake.locals['Browser'] | |
17 | 16 |
18 import py2stdlib | |
19 py2stdlib.MODULE_TRANSFORMS['mechanize'] = mechanize_transform | |
20 | 17 |
| 18 register_module_extender(MANAGER, 'mechanize', mechanize_transform) |
OLD | NEW |