From a21283bd2ed5e077a1c4d202b7db2e8db49d79a9 Mon Sep 17 00:00:00 2001 From: Labrys of Knossos Date: Sat, 3 Dec 2022 18:31:11 -0500 Subject: [PATCH] Add type-hints for `SickBeard.api_call` and `SickBeard.process_response` --- core/auto_process/managers/sickbeard.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/core/auto_process/managers/sickbeard.py b/core/auto_process/managers/sickbeard.py index e7403331..6c0a85b8 100644 --- a/core/auto_process/managers/sickbeard.py +++ b/core/auto_process/managers/sickbeard.py @@ -432,7 +432,7 @@ class SickBeard: # delete any unused params so we don't pass them to SB by mistake [fork_params.pop(k) for k, v in list(fork_params.items()) if v is None] - def api_call(self): + def api_call(self) -> ProcessResult: """Perform a base sickbeard api call.""" self._process_fork_prarams() logger.debug(f'Opening URL: {self.url} with params: {self.sb_init.fork_params}', self.sb_init.section) @@ -468,7 +468,7 @@ class SickBeard: return self.process_response(response) - def process_response(self, response): + def process_response(self, response: requests.Response) -> ProcessResult: """Iterate over the lines returned, and log. :param response: Streamed Requests response object.