From 725b211b5f3b04c21649629b15cfa8065eac2ae6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Pavel=20K=C3=A1cha?= <ph@cesnet.cz>
Date: Thu, 16 Oct 2014 11:26:30 +0200
Subject: [PATCH] More pythonic "main" structure
---
warden3/warden_client/warden_client_test.py | 71 +++++++++++----------
1 file changed, 38 insertions(+), 33 deletions(-)
diff --git a/warden3/warden_client/warden_client_test.py b/warden3/warden_client/warden_client_test.py
index 568b082..93297ee 100755
--- a/warden3/warden_client/warden_client_test.py
+++ b/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()
--
GitLab