fixed TOTP for #3

henrydatei · Oct 1, 2022 · 43dd1ab · 43dd1ab
1 parent 2ea49fd
commit 43dd1ab
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/wikifolio.py b/wikifolio.py
@@ -601,7 +601,8 @@ def buy_quote(self, amount: int, isin: str) -> OrderResponse:
                 ws.close()
 
                 if self.twoFA_key != None:
-                    auth = requests.post('https://www.wikifolio.com/api/totp/verify', data = self.twoFA_key, cookies = self.cookie)
+                    totp = TOTP(self.twoFA_key)
+                    auth = requests.post('https://www.wikifolio.com/api/totp/verify', data = totp.now(), cookies = self.cookie)
                     cookies = auth.cookies
                 else: 
                     cookies = self.cookie
@@ -672,7 +673,8 @@ def sell_quote(self, amount: int, isin: str) -> OrderResponse:
                 ws.close()
 
                 if self.twoFA_key != None:
-                    auth = requests.post('https://www.wikifolio.com/api/totp/verify', data = self.twoFA_key, cookies = self.cookie)
+                    totp = TOTP(self.twoFA_key)
+                    auth = requests.post('https://www.wikifolio.com/api/totp/verify', data = totp.now(), cookies = self.cookie)
                     cookies = auth.cookies
                 else: 
                     cookies = self.cookie