From 510beaaede6faf2aa1c18c9ec9c77bcbee6c2f4c Mon Sep 17 00:00:00 2001
From: Richard Belleville <rbellevi@google.com>
Date: Fri, 1 Mar 2019 17:41:56 -0800
Subject: [PATCH] Add a CLI parser to the client

---
 examples/python/multiprocessing/client.py | 15 +++++++++++----
 examples/python/multiprocessing/server.py |  1 -
 2 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/examples/python/multiprocessing/client.py b/examples/python/multiprocessing/client.py
index 788820916ea..920b5285475 100644
--- a/examples/python/multiprocessing/client.py
+++ b/examples/python/multiprocessing/client.py
@@ -17,6 +17,7 @@ from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
+import argparse
 import atexit
 import grpc
 import logging
@@ -29,7 +30,6 @@ import prime_pb2
 import prime_pb2_grpc
 
 _PROCESS_COUNT = 8
-_SERVER_ADDRESS = 'localhost:50051'
 _MAXIMUM_CANDIDATE = 10000
 
 # Each worker process initializes a single channel after forking.
@@ -61,16 +61,23 @@ def _run_worker_query(primality_candidate):
     return _worker_stub_singleton.check(
             prime_pb2.PrimeCandidate(candidate=primality_candidate))
 
-
-def main():
+def _calculate_primes(server_address):
     worker_pool = multiprocessing.Pool(processes=_PROCESS_COUNT,
-                    initializer=_initialize_worker, initargs=(_SERVER_ADDRESS,))
+                    initializer=_initialize_worker, initargs=(server_address,))
     check_range = range(2, _MAXIMUM_CANDIDATE)
     primality = worker_pool.map(_run_worker_query, check_range)
     primes = zip(check_range, map(operator.attrgetter('isPrime'), primality))
     logging.warning(tuple(primes))
 
 
+def main():
+    msg = 'Determine the primality of the first {} integers.'.format(
+            _MAXIMUM_CANDIDATE)
+    parser = argparse.ArgumentParser(description=msg)
+    parser.add_argument('server_address', help='The address of the server (e.g. localhost:50051)')
+    args = parser.parse_args()
+    _calculate_primes(args.server_address)
+
 if __name__ == '__main__':
     logging.basicConfig()
     main()
diff --git a/examples/python/multiprocessing/server.py b/examples/python/multiprocessing/server.py
index f2a2544b827..6801f806126 100644
--- a/examples/python/multiprocessing/server.py
+++ b/examples/python/multiprocessing/server.py
@@ -34,7 +34,6 @@ import prime_pb2_grpc
 _ONE_DAY = datetime.timedelta(days=1)
 _PROCESS_COUNT = 8
 _THREAD_CONCURRENCY = 10
-_BIND_ADDRESS = '[::]:50051'
 
 
 def is_prime(n):