[tracker] Massage tracker_service.py a bit, so it can be import'ed in ipython

git-svn-id: svn+ssh://svn.debian.org/svn/secure-testing@39342 e39458fd-73e7-0310-bf30-c45bca0a0e42

1 files changed, 18 insertions, 14 deletions

diff --git a/bin/tracker_service.py b/bin/tracker_service.py
index 20e6c8e7a3..2f4f1332d5 100644
--- a/bin/tracker_service.py
+++ b/bin/tracker_service.py
@@ -8,20 +8,23 @@ import security_db
 from web_support import *
 import json
 
-if len(sys.argv) not in (3, 5):
-    print "usage: python tracker_service.py SOCKET-PATH DATABASE-PATH"
-    print "       python tracker_service.py URL HOST PORT DATABASE-PATH"
-    sys.exit(1)
-if len(sys.argv) == 3:
-    socket_name = sys.argv[1]
-    db_name = sys.argv[2]
-    webservice_base_class = WebService
+if __name__ == "__main__":
+    if len(sys.argv) not in (3, 5):
+        print "usage: python tracker_service.py SOCKET-PATH DATABASE-PATH"
+        print "       python tracker_service.py URL HOST PORT DATABASE-PATH"
+        sys.exit(1)
+    if len(sys.argv) == 3:
+        socket_name = sys.argv[1]
+        db_name = sys.argv[2]
+        webservice_base_class = WebService
+    else:
+        server_base_url = sys.argv[1]
+        server_address = sys.argv[2]
+        server_port = int(sys.argv[3])
+        socket_name = (server_base_url, server_address, server_port)
+        db_name = sys.argv[4]
+        webservice_base_class = WebServiceHTTP
 else:
-    server_base_url = sys.argv[1]
-    server_address = sys.argv[2]
-    server_port = int(sys.argv[3])
-    socket_name = (server_base_url, server_address, server_port)
-    db_name = sys.argv[4]
     webservice_base_class = WebServiceHTTP
 
 def clean_dict(d):
@@ -1622,4 +1625,5 @@ Debian bug number.'''),
     def pre_dispatch(self):
         pass
 
-TrackerService(socket_name, db_name).run()
+if __name__ == "__main__":
+    TrackerService(socket_name, db_name).run()