From ae1e292990a0b5c57b033bfd278c9b1d618673e9 Mon Sep 17 00:00:00 2001 From: Nelson Dane <47427072+NelsonDane@users.noreply.github.com> Date: Wed, 10 Apr 2024 15:21:10 -0400 Subject: [PATCH] deepsource fixes --- fennel_invest_api/__init__.py | 2 +- fennel_invest_api/endpoints.py | 4 +++- fennel_invest_api/fennel.py | 10 ++++------ setup.py | 2 +- 4 files changed, 9 insertions(+), 9 deletions(-) diff --git a/fennel_invest_api/__init__.py b/fennel_invest_api/__init__.py index 60b8911..2a82a22 100644 --- a/fennel_invest_api/__init__.py +++ b/fennel_invest_api/__init__.py @@ -1,4 +1,4 @@ from fennel_invest_api.endpoints import Endpoints from fennel_invest_api.fennel import Fennel -__all__ = ["Endpoints", "Fennel"] \ No newline at end of file +__all__ = ["Endpoints", "Fennel"] diff --git a/fennel_invest_api/endpoints.py b/fennel_invest_api/endpoints.py index 7a610d3..88936ae 100644 --- a/fennel_invest_api/endpoints.py +++ b/fennel_invest_api/endpoints.py @@ -13,7 +13,9 @@ def retrieve_bearer_url(self): def oauth_url(self): return f"{self.accounts}/oauth/token" - def build_graphql_payload(self, query, variables={}): + def build_graphql_payload(self, query, variables=None): + if variables is None: + variables = {} return { "operationName": None, "variables": variables, diff --git a/fennel_invest_api/fennel.py b/fennel_invest_api/fennel.py index 0145acb..268ca26 100644 --- a/fennel_invest_api/fennel.py +++ b/fennel_invest_api/fennel.py @@ -69,8 +69,7 @@ def login(self, email, wait_for_code=True, code=None): if self.Bearer is not None: if self._verify_login(): return True - else: - self._clear_credentials() + self._clear_credentials() if code is None: url = self.endpoints.retrieve_bearer_url() payload = { @@ -103,7 +102,6 @@ def login(self, email, wait_for_code=True, code=None): self.Bearer = response['access_token'] self.Refresh = response['refresh_token'] self.ID_Token = response['id_token'] - # refresh_token() # Refresh token after login? self._save_credentials() return True @@ -129,12 +127,12 @@ def _verify_login(self): try: self.get_portfolio_summary() return True - except Exception as e: + except Exception: self.refresh_token() try: self.get_portfolio_summary() return True - except Exception as e: + except Exception: return False @check_login @@ -186,4 +184,4 @@ def place_order(self, ticker, quantity, side, price="market"): order_response = self.session.post(self.endpoints.graphql, headers=headers, data=query) if order_response.status_code != 200: raise Exception(f"Order Request failed with status code {order_response.status_code}: {order_response.text}") - return order_response.json() \ No newline at end of file + return order_response.json() diff --git a/setup.py b/setup.py index 145db07..4c55783 100644 --- a/setup.py +++ b/setup.py @@ -10,4 +10,4 @@ author="Nelson Dane", packages=["fennel_invest_api"], install_requires=["requests", "python-dotenv"], -) \ No newline at end of file +)