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

Sebastian Ramacher · Sebastian Ramacher · commit 53ba9f10ea30 · 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.

--HG--
branch : 0.10-bugfix
extra : transplant_source : %D3%D9Jn%3C%F92%F5%AA%02%1E%EB%84%17%89%83%2A%ED%B5y
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
