From 42babb76a21a9c3cfed7c85e34e090b1bf8e4be9 Mon Sep 17 00:00:00 2001 From: Dan Miller Date: Fri, 16 Nov 2012 20:54:55 -0800 Subject: [PATCH] Implementing module name parsing with os.path. --- bin/cython_freeze | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/bin/cython_freeze b/bin/cython_freeze index 2702ce7..74f3656 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 -- 2.7.4