diff --git a/src/asquare.py b/src/asquare.py index c172348..1059b98 100644 --- a/src/asquare.py +++ b/src/asquare.py @@ -103,7 +103,11 @@ async def acat2id(cat_id: str): { "id": obj["id"], "name": obj.get("item_data", {}).get("name", "Unknown"), - "number": get_set_number(obj.get("item_data", {}).get("name", "")) + "number": None if not obj.get("item_data", {}).get("name", "") else ( + get_set_number(obj.get("item_data", {}).get("name", "")) + if re.findall(r"\d{5}", obj.get("item_data", {}).get("name", "")) + else None + ) } for obj in result_json.get("objects", []) ]