Trying to get launch script working for windows

Anthony Oliver · Anthony Oliver · commit aed98b8231b6 · 2012-07-30T11:40:31.000-04:00
diff --git a/SimpleCV/Shell/Shell.py b/SimpleCV/Shell/Shell.py
@@ -25,6 +25,7 @@
 import subprocess
 import time
 import webbrowser
+import sys
 
 #Load simpleCV libraries
 from SimpleCV.Shell.Tutorial import *
@@ -180,23 +181,28 @@ def run_notebook():
 
 def main(*args):
 
+    print "args:",sys.argv
+
+    #~ args = sys.argv
+    #~ import pdb;pdb.set_trace()
+    #~ import pdb;pdb.set_trace()
     log_level = logging.WARNING
-    if len(args) and len(args[0]) > 1:
-      for flag in args[0]:
-        if flag == "notebook" and IPVER > 10:
-            run_notebook()
-            sys.exit()
-
-        if flag in ["--headless","headless"]:
-          # set SDL to use the dummy NULL video driver,
-          #   so it doesn't need a windowing system.
-          os.environ["SDL_VIDEODRIVER"] = "dummy"
-
-        elif flag in ['--nowarnings','nowarnings']:
-          log_level = logging.INFO
-
-        elif flag in ['--debug','debug']:
-          log_level = logging.DEBUG
+    if len(sys.argv) > 1 and len(sys.argv[1]) > 1:
+      flag = sys.argv[1]
+      if flag == "notebook" and IPVER > 10:
+          run_notebook()
+          sys.exit()
+
+      if flag in ["--headless","headless"]:
+        # set SDL to use the dummy NULL video driver,
+        #   so it doesn't need a windowing system.
+        os.environ["SDL_VIDEODRIVER"] = "dummy"
+
+      elif flag in ['--nowarnings','nowarnings']:
+        log_level = logging.INFO
+
+      elif flag in ['--debug','debug']:
+        log_level = logging.DEBUG
 
     init_logging(log_level)
     shellclear()
diff --git a/setup.py b/setup.py
@@ -40,6 +40,11 @@
                         'examples/tracking/*'
                         ],
   },
-  scripts=['scripts/simplecv']
+  #~ scripts=['scripts/simplecv'],
+  entry_points={
+    'console_scripts': [
+      'simplecv = SimpleCV.Shell:main',
+    ],
+  },
 
   )
