Use imp to get the suffix that should be dropped (Closes: bpython#204).

Sebastian Ramacher · Sebastian Ramacher · commit 6e75552debf3 · 2011-11-16T14:28:34.000+01:00
Don't just drop the extension since Python 3 supports modules named like
foo.cpython-XY.so. Use the suffixes from imp.get_suffixes() instead.
diff --git a/bpython/importcompletion.py b/bpython/importcompletion.py
@@ -123,7 +123,10 @@ def find_modules(path):
             # Unfortunately, CPython just crashes if there is a directory
             # which ends with a python extension, so work around.
             continue
-        name = os.path.splitext(name)[0]
+        for suffix in imp.get_suffixes():
+            if name.endswith(suffix[0]):
+                name = name[:-len(suffix[0])]
+                break
         if py3 and name == "badsyntax_pep3120":
             # Workaround for issue #166
             continue
