Yuvipanda has submitted this change and it was merged.

Change subject: Use argparse to parse parameters for registering stream clients
......................................................................


Use argparse to parse parameters for registering stream clients

Change-Id: I0b056d952bef7dbc16d61dd42b9a2da9e6ff97b9
---
M src/register-stream-client.py
1 file changed, 15 insertions(+), 4 deletions(-)

Approvals:
  Yuvipanda: Verified; Looks good to me, approved



diff --git a/src/register-stream-client.py b/src/register-stream-client.py
index 0de2965..a511c19 100644
--- a/src/register-stream-client.py
+++ b/src/register-stream-client.py
@@ -1,6 +1,7 @@
 import sys
 import os
 import logging
+import argparse
 
 import redis
 import yaml
@@ -25,8 +26,18 @@
 def make_key(*key_parts):
     return PREFIX + "_" + '.'.join(key_parts)
 
-if __name__ == '__main__':
-    new_client_key = sys.argv[1]
-    red.sadd(make_key(CLIENTS_KEY), new_client_key)
+def add_subscription(args):
+    red.sadd(make_key(CLIENTS_KEY), args.key)
     red.save() # We don't want to lose client lists do we
-    logging.info('Added key %s' % new_client_key)
+    logging.info('Added key %s' % args.key)
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser(description="Manage redis gerrit 
subscriptions")
+    subparsers = parser.add_subparsers()
+
+    parser_subscribe = subparsers.add_parser('subscribe', help='Add a 
subscription')
+    parser_subscribe.add_argument('key')
+    parser_subscribe.set_defaults(func=add_subscription)
+
+    args = parser.parse_args()
+    args.func(args)

-- 
To view, visit https://gerrit.wikimedia.org/r/76461
To unsubscribe, visit https://gerrit.wikimedia.org/r/settings

Gerrit-MessageType: merged
Gerrit-Change-Id: I0b056d952bef7dbc16d61dd42b9a2da9e6ff97b9
Gerrit-PatchSet: 1
Gerrit-Project: labs/tools/gerrit-to-redis
Gerrit-Branch: master
Gerrit-Owner: Yuvipanda <[email protected]>
Gerrit-Reviewer: Yuvipanda <[email protected]>

_______________________________________________
MediaWiki-commits mailing list
[email protected]
https://lists.wikimedia.org/mailman/listinfo/mediawiki-commits

Reply via email to