diff --git a/companion.py b/companion.py index 8f0633c1..c96b84d4 100644 --- a/companion.py +++ b/companion.py @@ -476,6 +476,14 @@ class Session(object): self.auth: Optional[Auth] = None self.retrying = False # Avoid infinite loop when successful auth / unsuccessful query + def start_frontier_auth(self, access_token: str) -> None: + """Start an oAuth2 session.""" + logger.debug('Starting session') + self.session = requests.Session() + self.session.headers['Authorization'] = f'Bearer {access_token}' + self.session.headers['User-Agent'] = USER_AGENT + self.state = Session.STATE_OK + def login(self, cmdr: str = None, is_beta: Union[None, bool] = None) -> bool: """ Attempt oAuth2 login. @@ -545,14 +553,6 @@ class Session(object): self.auth = None raise # Bad thing happened - def start_frontier_auth(self, access_token: str) -> None: - """Start an oAuth2 session.""" - logger.debug('Starting session') - self.session = requests.Session() - self.session.headers['Authorization'] = f'Bearer {access_token}' - self.session.headers['User-Agent'] = USER_AGENT - self.state = Session.STATE_OK - def query(self, endpoint: str) -> CAPIData: # noqa: CCR001, C901 """Perform a query against the specified CAPI endpoint.""" logger.trace_if('capi.query', f'Performing query for endpoint "{endpoint}"')