Исправление ошибки неизвестного enum

Author: p141592

avito/client.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from collections.abc import Iterable
+from collections.abc import Callable, Iterable
 from datetime import date, datetime
 from pathlib import Path
 from types import TracebackType
@@ -82,6 +82,16 @@ def _summary_unavailable_section(section: str, error: AvitoError) -> SummaryUnav
     )
 
 
+def _safe_summary[SummaryT](
+    section: str,
+    factory: Callable[[], SummaryT],
+) -> tuple[SummaryT | None, list[SummaryUnavailableSection]]:
+    try:
+        return factory(), []
+    except AvitoError as error:
+        return None, [_summary_unavailable_section(section, error)]
+
+
 class AvitoClient:
     """Единственная публичная точка входа SDK с фабриками доменных объектов.
 
@@ -189,18 +199,43 @@ def account_health(
             date_from=date_from,
             date_to=date_to,
         )
+        item_ids = [item.item_id for item in listings.items if item.item_id is not None]
+        chats, chats_unavailable = _safe_summary(
+            "chats",
+            lambda: self.chat_summary(user_id=resolved_user_id),
+        )
+        orders, orders_unavailable = _safe_summary("orders", self.order_summary)
+        reviews, reviews_unavailable = _safe_summary("reviews", self.review_summary)
+        promotion, promotion_unavailable = _safe_summary(
+            "promotion",
+            lambda: self.promotion_summary(item_ids=item_ids),
+        )
+        unavailable_sections = [
+            *listings.unavailable_sections,
+            *chats_unavailable,
+            *orders_unavailable,
+            *reviews_unavailable,
+            *promotion_unavailable,
+        ]
+        if chats is not None:
+            unavailable_sections.extend(chats.unavailable_sections)
+        if orders is not None:
+            unavailable_sections.extend(orders.unavailable_sections)
+        if reviews is not None:
+            unavailable_sections.extend(reviews.unavailable_sections)
+        if promotion is not None:
+            unavailable_sections.extend(promotion.unavailable_sections)
         return AccountHealthSummary(
             user_id=resolved_user_id,
             balance_total=balance.total,
             balance_real=balance.real,
             balance_bonus=balance.bonus,
             listings=listings,
-            chats=self.chat_summary(user_id=resolved_user_id),
-            orders=self.order_summary(),
-            reviews=self.review_summary(),
-            promotion=self.promotion_summary(
-                item_ids=[item.item_id for item in listings.items if item.item_id is not None]
-            ),
+            chats=chats,
+            orders=orders,
+            reviews=reviews,
+            promotion=promotion,
+            unavailable_sections=unavailable_sections,
         )
 
     def listing_health(
@@ -322,15 +357,36 @@ def order_summary(self) -> OrderSummary:
     def review_summary(self) -> ReviewSummary:
         """Возвращает итоговую read-only сводку по отзывам."""
 
-        reviews = self.review().list()
+        reviews_error: AvitoError | None = None
+        try:
+            reviews = self.review().list()
+        except AvitoError as error:
+            reviews = None
+            reviews_error = error
         rating = self.rating_profile().get()
-        scores = [item.score for item in reviews.items if item.score is not None]
+        scores = [item.score for item in reviews.items if item.score is not None] if reviews else []
         average_score = sum(scores) / len(scores) if scores else None
+        unavailable_sections = (
+            [_summary_unavailable_section("reviews", reviews_error)]
+            if reviews_error is not None
+            else []
+        )
         return ReviewSummary(
-            total_reviews=reviews.total if reviews.total is not None else len(reviews.items),
-            average_score=average_score,
-            unanswered_reviews=sum(1 for item in reviews.items if item.can_answer is True),
+            total_reviews=(
+                reviews.total
+                if reviews is not None and reviews.total is not None
+                else rating.reviews_count
+                if reviews is None
+                else len(reviews.items)
+            ),
+            average_score=average_score if reviews is not None else rating.score,
+            unanswered_reviews=(
+                sum(1 for item in reviews.items if item.can_answer is True)
+                if reviews is not None
+                else None
+            ),
             rating_score=rating.score,
+            unavailable_sections=unavailable_sections,
         )
 
     def promotion_summary(self, *, item_ids: list[int] | None = None) -> PromotionSummary:

avito/summary/models.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 
 from avito.ads.enums import ListingStatus
 from avito.core.serialization import SerializableModel
@@ -50,7 +50,7 @@ class ListingHealthSummary(SerializableModel):
     total_favorites: int | None
     total_calls: int | None
     total_spendings: float | None
-    unavailable_sections: list[SummaryUnavailableSection]
+    unavailable_sections: list[SummaryUnavailableSection] = field(default_factory=list)
 
 
 @dataclass(slots=True, frozen=True)
@@ -61,6 +61,7 @@ class ChatSummary(SerializableModel):
     total_chats: int
     unread_chats: int
     unread_messages: int
+    unavailable_sections: list[SummaryUnavailableSection] = field(default_factory=list)
 
 
 @dataclass(slots=True, frozen=True)
@@ -69,16 +70,18 @@ class OrderSummary(SerializableModel):
 
     total_orders: int
     active_orders: int
+    unavailable_sections: list[SummaryUnavailableSection] = field(default_factory=list)
 
 
 @dataclass(slots=True, frozen=True)
 class ReviewSummary(SerializableModel):
     """Итоговая read-only сводка по отзывам."""
 
-    total_reviews: int
+    total_reviews: int | None
     average_score: float | None
-    unanswered_reviews: int
+    unanswered_reviews: int | None
     rating_score: float | None
+    unavailable_sections: list[SummaryUnavailableSection] = field(default_factory=list)
 
 
 @dataclass(slots=True, frozen=True)
@@ -89,6 +92,7 @@ class PromotionSummary(SerializableModel):
     active_orders: int
     total_services: int
     available_services: int
+    unavailable_sections: list[SummaryUnavailableSection] = field(default_factory=list)
 
 
 @dataclass(slots=True, frozen=True)
@@ -104,6 +108,7 @@ class AccountHealthSummary(SerializableModel):
     orders: OrderSummary | None = None
     reviews: ReviewSummary | None = None
     promotion: PromotionSummary | None = None
+    unavailable_sections: list[SummaryUnavailableSection] = field(default_factory=list)
 
 
 @dataclass(slots=True, frozen=True)