fixed for virtualenvs

dmalan · dmalan · commit df491f9867a0 · 2017-10-29T19:20:54.000-04:00
diff --git a/src/cs50/cs50.py b/src/cs50/cs50.py
@@ -6,7 +6,6 @@
 
 from distutils.sysconfig import get_python_lib
 from os.path import abspath, join
-from site import getusersitepackages
 from termcolor import cprint
 from traceback import extract_tb, format_list, format_exception_only
 
@@ -16,10 +15,9 @@ def excepthook(type, value, tb):
     Format traceback, darkening entries from global site-packages directories
     and user-specific site-packages directory.
 
-    https://stackoverflow.com/a/33042323/5156190
-    https://stackoverflow.com/a/122340/5156190
+    https://stackoverflow.com/a/46071447/5156190
     """
-    packages = tuple(join(abspath(p), "") for p in [get_python_lib(), getusersitepackages()])
+    packages = tuple(join(abspath(p), "") for p in sys.path[1:])
     for entry in extract_tb(tb):
         fmt = format_list((entry,))
         if (entry[0].startswith(packages)):
