From 3e6d535a74654e7842d1189792e4c96395d1b99c Mon Sep 17 00:00:00 2001 From: PCoder Date: Tue, 13 Aug 2019 14:48:50 +0530 Subject: [PATCH] Replace token with seed and obtain token from seed + fix bug Bug = getting data from dict --- ungleich_pay.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/ungleich_pay.py b/ungleich_pay.py index d800433..b5fb711 100644 --- a/ungleich_pay.py +++ b/ungleich_pay.py @@ -1,5 +1,6 @@ import urllib.request import json +from pyotp import TOTP UNGLEICH_PAY_SERVER_URL = "https://pay.ungleich.ch" @@ -15,8 +16,12 @@ class ungleichPay(object): self.parser['customers-list'].add_argument('--username', required=True) self.parser['customers-list'].add_argument('--realm', required=True) self.parser['customers-list'].add_argument( - '--token', required=True, - help="A valid token obtained from ungleich" + '--seed', required=True, + help="A valid seed obtained from ungleich" + ) + self.parser['customers-list'].add_argument( + '--email', required=True, + help="Email associated with the username" ) self.parser['customers-list'].add_argument( '--ungleich-pay-server', required=False, @@ -30,14 +35,16 @@ class ungleichPay(object): request_url = args.ungleich_pay_server + customers_list_endpoint else: request_url = UNGLEICH_PAY_SERVER_URL + customers_list_endpoint + print(f"request_url={request_url}") req = urllib.request.Request(url=request_url, method='GET', headers={ + "email": args.email, "username": args.username, "realm": args.realm, - "token": args.token, + "token": TOTP(args.seed).now(), "Accept": "application/json" }) response = urllib.request.urlopen(req) response_json = json.loads(response.read().decode('utf-8')) - print(response_json.data) + print(response_json["data"])