diff --git a/wxPython/scripts/pywrap b/wxPython/scripts/pywrap index d2ec252016..3bd7ef1f84 100755 --- a/wxPython/scripts/pywrap +++ b/wxPython/scripts/pywrap @@ -1,7 +1,4 @@ #!/usr/bin/env python from wx.py.PyWrap import main -import sys, os - -sys.path.insert(0, os.curdir) -main(sys.argv) +main() diff --git a/wxPython/wxPython/py/PyAlaCarte.py b/wxPython/wxPython/py/PyAlaCarte.py index f53452429a..578ba12247 100755 --- a/wxPython/wxPython/py/PyAlaCarte.py +++ b/wxPython/wxPython/py/PyAlaCarte.py @@ -32,11 +32,12 @@ class App(wx.wxApp): return True def main(filename=None): + if not filename and len(sys.argv) > 1: + filename = sys.argv[1] + if filename: + filename = os.path.realpath(filename) app = App(filename) app.MainLoop() if __name__ == '__main__': - filename = None - if len(sys.argv) > 1: - filename = os.path.realpath(sys.argv[1]) - main(filename) + main() diff --git a/wxPython/wxPython/py/PyAlaMode.py b/wxPython/wxPython/py/PyAlaMode.py index 65a478f01d..24eaccedf4 100755 --- a/wxPython/wxPython/py/PyAlaMode.py +++ b/wxPython/wxPython/py/PyAlaMode.py @@ -32,11 +32,12 @@ class App(wx.wxApp): return True def main(filename=None): + if not filename and len(sys.argv) > 1: + filename = sys.argv[1] + if filename: + filename = os.path.realpath(filename) app = App(filename) app.MainLoop() if __name__ == '__main__': - filename = None - if len(sys.argv) > 1: - filename = os.path.realpath(sys.argv[1]) - main(filename) + main() diff --git a/wxPython/wxPython/py/PyWrap.py b/wxPython/wxPython/py/PyWrap.py index 4808cb35d1..381bbe97f9 100755 --- a/wxPython/wxPython/py/PyWrap.py +++ b/wxPython/wxPython/py/PyWrap.py @@ -26,14 +26,16 @@ def wrap(app): app.MainLoop() -def main(argv): - if len(argv) < 2: - print "Please specify a module name." - raise SystemExit - name = argv[1] - if name[-3:] == '.py': - name = name[:-3] - module = __import__(name) +def main(modulename=None): + sys.path.insert(0, os.curdir) + if not modulename: + if len(sys.argv) < 2: + print "Please specify a module name." + raise SystemExit + modulename = sys.argv[1] + if modulename.endswith('.py'): + modulename = modulename[:-3] + module = __import__(modulename) # Find the App class. App = None d = module.__dict__ @@ -51,5 +53,4 @@ def main(argv): if __name__ == '__main__': - sys.path.insert(0, os.curdir) main(sys.argv)