Add type-hints for core.auto_process.common.ProcessResult

This commit is contained in:
Labrys of Knossos 2022-12-03 17:16:22 -05:00
commit aeb5698e3f

View file

@ -1,26 +1,28 @@
import typing
import requests import requests
from core import logger from core import logger
class ProcessResult: class ProcessResult:
def __init__(self, message, status_code): def __init__(self, message: str, status_code: int):
self.message = message self.message = message
self.status_code = status_code self.status_code = status_code
def __iter__(self): def __iter__(self) -> typing.Tuple[int, str]:
return self.status_code, self.message return self.status_code, self.message
def __bool__(self): def __bool__(self) -> bool:
return not bool(self.status_code) return not bool(self.status_code)
def __str__(self): def __str__(self) -> str:
return 'Processing {0}: {1}'.format( return 'Processing {0}: {1}'.format(
'succeeded' if bool(self) else 'failed', 'succeeded' if bool(self) else 'failed',
self.message, self.message,
) )
def __repr__(self): def __repr__(self) -> str:
return '<ProcessResult {0}: {1}>'.format( return '<ProcessResult {0}: {1}>'.format(
self.status_code, self.status_code,
self.message, self.message,