|
31 | 31 | except ImportError: |
32 | 32 | pass |
33 | 33 | else: |
34 | | - _before = SQL.execute |
35 | | - def _after(*args, **kwargs): |
| 34 | + _execute_before = SQL.execute |
| 35 | + def _execute_after(*args, **kwargs): |
36 | 36 | disabled = logging.getLogger("cs50").disabled |
37 | 37 | if flask.current_app: |
38 | 38 | logging.getLogger("cs50").disabled = False |
39 | 39 | try: |
40 | | - return _before(*args, **kwargs) |
| 40 | + return _execute_before(*args, **kwargs) |
41 | 41 | finally: |
42 | 42 | logging.getLogger("cs50").disabled = disabled |
43 | | - SQL.execute = _after |
| 43 | + SQL.execute = _execute_after |
44 | 44 |
|
45 | 45 | # When behind CS50 IDE's proxy, ensure that flask.redirect doesn't redirect from HTTPS to HTTP |
46 | 46 | # https://werkzeug.palletsprojects.com/en/0.15.x/middleware/proxy_fix/#module-werkzeug.middleware.proxy_fix |
47 | 47 | if getenv("C9_HOSTNAME") and not getenv("IDE_OFFLINE"): |
48 | 48 | try: |
49 | 49 | import flask |
50 | 50 | from werkzeug.middleware.proxy_fix import ProxyFix |
51 | | - _before = flask.Flask.__init__ |
52 | | - def _after(self, *args, **kwargs): |
53 | | - _before(self, *args, **kwargs) |
| 51 | + _flask_init_before = flask.Flask.__init__ |
| 52 | + def _flask_init_after(self, *args, **kwargs): |
| 53 | + _flask_init_before(self, *args, **kwargs) |
54 | 54 | self.wsgi_app = ProxyFix(self.wsgi_app, x_proto=1) |
55 | | - flask.Flask.__init__ = _after |
| 55 | + flask.Flask.__init__ = _flask_init_after |
56 | 56 | except: |
57 | 57 | pass |
58 | 58 |
|
|
0 commit comments