Skip to content

Commit fe14c17

Browse files
authored
Merge pull request #198 from ethancedwards8/pkgutil-migrate
flask.py: replace pkgutil with importlib
2 parents 3761168 + fb64408 commit fe14c17

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/cs50/flask.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import os
2-
import pkgutil
2+
import importlib.util
33
import sys
44

55

@@ -36,12 +36,12 @@ def _flask_init_after(self, *args, **kwargs):
3636

3737
# If Flask wasn't imported
3838
else:
39-
flask_loader = pkgutil.get_loader("flask")
40-
if flask_loader:
41-
_exec_module_before = flask_loader.exec_module
39+
spec = importlib.util.find_spec("flask")
40+
if spec and spec.loader:
41+
_exec_module_before = spec.loader.exec_module
4242

4343
def _exec_module_after(*args, **kwargs):
4444
_exec_module_before(*args, **kwargs)
4545
_wrap_flask(sys.modules["flask"])
4646

47-
flask_loader.exec_module = _exec_module_after
47+
spec.loader.exec_module = _exec_module_after

0 commit comments

Comments
 (0)