diff --git a/mo/web/acct.py b/mo/web/acct.py index af3f33d3f0688d7ba4d8266217c49c2dad321b23..53d14a07868e1a1208afeeca88b7a311b567ca47 100644 --- a/mo/web/acct.py +++ b/mo/web/acct.py @@ -454,10 +454,11 @@ class Reg2: }, } - def __init__(self, token: str, expected_type: db.RegReqType): + def __init__(self, token: Optional[str], expected_type: db.RegReqType): self.reg_type = expected_type self.status = self._parse_token(token) if self.status == RegStatus.ok: + assert token is not None self.status = self._load_rr(token) def _parse_token(self, token: Optional[str]) -> RegStatus: