From 66a27d435f2bb8aea97ea2875bf6ded190f267c3 Mon Sep 17 00:00:00 2001 From: PCoder Date: Tue, 13 Aug 2019 13:03:59 +0530 Subject: [PATCH] Allow ungleich_pay_server parameter to be passed by the user --- ungleich_pay.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/ungleich_pay.py b/ungleich_pay.py index 623880e..64ee68c 100644 --- a/ungleich_pay.py +++ b/ungleich_pay.py @@ -18,11 +18,18 @@ class ungleichPay(object): '--token', required=True, help="A valid token obtained from ungleich" ) + self.parser['ungleich_pay_server'].add_argument( + '--ungleich-pay-server', required=False, + help="A valid ungleich pay server url" + ) self.parser['customers-list'].set_defaults(func=self.customers_list) def customers_list(self, args): customers_list_endpoint = "/customers-list" - request_url = UNGLEICH_PAY_SERVER_URL + customers_list_endpoint + if args.ungleich_pay_server: + request_url = args.ungleich_pay_server + customers_list_endpoint + else: + request_url = UNGLEICH_PAY_SERVER_URL + customers_list_endpoint req = urllib.request.Request(url=request_url, method='GET', headers={