Implementing module name parsing with os.path.

1 files changed, 2 insertions, 8 deletions

diff --git a/bin/cython_freeze b/bin/cython_freeze
index 2702ce7e0..74f365689 100755
--- a/bin/cython_freeze
+++ b/bin/cython_freeze
@@ -7,6 +7,7 @@ See Demos/freeze/README.txt for more details.
 """
 
 import optparse
+from os.path import splitext, basename
 
 usage= '%prog [-o outfile] [-p] module [module ...]'
 description = 'Create a C file for embedding Cython modules.'
@@ -27,14 +28,7 @@ if options.output:
     old_stdout = sys.stdout
     sys.stdout = open(options.output, 'w')
 
-def format_modname(name):
-    if name.endswith('.pyx'):
-        name = name[:-4]
-    elif name.endswith('.py'):
-        name = name[:-3]
-    return name.replace('.','_')
-
-modules = [format_modname(x) for x in args]
+modules = [basename(splitext(x)[0]).replace('.', '_') for x in args]
 
 print """\
 #include <Python.h>