More pythonic "main" structure

warden3/warden_client/warden_client_test.py

@@ -126,36 +126,41 @@ def gen_random_idea():
     return event
 
 
-wclient = Client(**read_cfg("warden_client.cfg"))
-# Also inline arguments are possible:
-# wclient = Client(
-#     url  = 'https://warden.example.com/warden3',
-#     keyfile  = '/opt/warden3/etc/key.pem',
-#     certfile = '/opt/warden3/etc/cert.pem',
-#     cafile = '/opt/warden3/etc/tcs-ca-bundle.pem',
-#     timeout=10,
-#     errlog={"level": "debug"},
-#     filelog={"level": "debug"},
-#     idstore="MyClient.id",
-#     name="MyClient")
-
-print "=== Getting 10 events ==="
-start = time()
-ret = wclient.getEvents(count=10)
-print "Time: %f" % (time()-start)
-for e in ret:
-    print e
-if ret:
-    print len(ret)
-
-print "=== Sending 500 events ==="
-start = time()
-ret = wclient.sendEvents([gen_random_idea() for i in range(500)])
-if ret:
-    print ret
-print "Time: %f" % (time()-start)
-
-print "=== Server info ==="
-info = wclient.getInfo()
-if not isinstance(info, Error):
-    pprint(info)
+def main():
+    wclient = Client(**read_cfg("warden_client.cfg"))
+    # Also inline arguments are possible:
+    # wclient = Client(
+    #     url  = 'https://warden.example.com/warden3',
+    #     keyfile  = '/opt/warden3/etc/key.pem',
+    #     certfile = '/opt/warden3/etc/cert.pem',
+    #     cafile = '/opt/warden3/etc/tcs-ca-bundle.pem',
+    #     timeout=10,
+    #     errlog={"level": "debug"},
+    #     filelog={"level": "debug"},
+    #     idstore="MyClient.id",
+    #     name="MyClient")
+
+    print "=== Getting 10 events ==="
+    start = time()
+    ret = wclient.getEvents(count=10)
+    print "Time: %f" % (time()-start)
+    for e in ret:
+        print e
+    if ret:
+        print len(ret)
+
+    print "=== Sending 500 events ==="
+    start = time()
+    ret = wclient.sendEvents([gen_random_idea() for i in range(500)])
+    if ret:
+        print ret
+    print "Time: %f" % (time()-start)
+
+    print "=== Server info ==="
+    info = wclient.getInfo()
+    if not isinstance(info, Error):
+        pprint(info)
+
+
+if __name__ == "__main__":
+    main()