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={