| 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 |