From 205449a8e59c78f3b1032a3b996463d22588f05f Mon Sep 17 00:00:00 2001 From: Max Nanis Date: Fri, 27 Feb 2026 02:35:47 -0500 Subject: tests.flow all pass. Now everything is GREEN ✅ --- jb/managers/assignment.py | 4 ++-- jb/managers/thl.py | 2 ++ 2 files changed, 4 insertions(+), 2 deletions(-) (limited to 'jb') 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 -- cgit v1.2.3