diff options
Diffstat (limited to 'jb')
| -rw-r--r-- | jb/managers/assignment.py | 4 | ||||
| -rw-r--r-- | jb/managers/thl.py | 2 |
2 files changed, 4 insertions, 2 deletions
diff --git a/jb/managers/assignment.py b/jb/managers/assignment.py index c0a168e..6104bc5 100644 --- a/jb/managers/assignment.py +++ b/jb/managers/assignment.py @@ -2,7 +2,7 @@ from datetime import datetime, timezone from typing import Optional from psycopg import sql -from pydantic import NonNegativeInt +from pydantic import NonNegativeInt, PositiveInt from jb.managers import PostgresManager from jb.models.assignment import AssignmentStub, Assignment @@ -215,7 +215,7 @@ class AssignmentManager(PostgresManager): return None def missing_tsid_count( - self, amt_worker_id: str, lookback_hrs: int = 24 + self, amt_worker_id: str, lookback_hrs: PositiveInt = 24 ) -> NonNegativeInt: """ Look at this user's previous N hrs of submitted assignments. diff --git a/jb/managers/thl.py b/jb/managers/thl.py index 83f49f6..44b6496 100644 --- a/jb/managers/thl.py +++ b/jb/managers/thl.py @@ -23,6 +23,7 @@ import requests def get_user_profile(amt_worker_id: str) -> UserProfile: url = f"{settings.fsb_host}{settings.product_id}/user/{amt_worker_id}/profile/" res = requests.get(url).json() + if res.get("detail") == "user not found": raise ValueError("user not found") @@ -39,6 +40,7 @@ def get_user_blocked_or_not_exists(amt_worker_id: str) -> Optional[bool]: try: res = get_user_profile(amt_worker_id=amt_worker_id) return res.user.blocked if res.user.blocked is not None else False + except ValueError as e: if e.args[0] == "user not found": return True |