Try not to assume a module's file extension is .py

This commit is contained in:
Kodi Arfer
2017-04-19 12:26:56 -07:00
parent 581b463b60
commit dcefb287fc
4 changed files with 36 additions and 2 deletions

View File

@@ -3,6 +3,7 @@
import fnmatch
import inspect
import sys
import os
import collections
import math
from itertools import count
@@ -235,8 +236,7 @@ class PyobjMixin(PyobjContext):
continue
name = node.name
if isinstance(node, Module):
assert name.endswith(".py")
name = name[:-3]
name = os.path.splitext(name)[0]
if stopatmodule:
if includemodule:
parts.append(name)