style: improve type handling

Signed-off-by: bossenti <[email protected]>
bossenti · Nov 4, 2023 · 1856b2d · 1856b2d
1 parent ffbede5
commit 1856b2d
Showing 1 changed file with 16 additions and 4 deletions.
diff --git a/vistafetch/model/asset/financial_asset.py b/vistafetch/model/asset/financial_asset.py
@@ -81,17 +81,29 @@ def __query_day_price_data(self, day: datetime.date) -> PriceData:
                 "Please use a valid financial asset class."
             )
 
+        if self.market is None:
+            raise RuntimeError(
+                "No market information available - Unable to query price data."
+            )
+
         response = api_session.get(
-            f"{ONVISTA_API_BASE_URL}instruments/{self.entity_type}/ISIN:{self.isin}/eod_history?idNotation={self.market.id_notation}&range=D1&startDate={day.year}-{day.month}-{day.day}"
-            # type: ignore
+            f"{ONVISTA_API_BASE_URL}instruments/{self.entity_type}/ISIN:{self.isin}/eod_history?idNotation="
+            f"{self.market.id_notation}&range=D1&startDate={day.year}-{day.month}-{day.day}"
         )
 
         price_raw = response.json()
 
         if len(price_raw["datetimeLast"]) == 0:
             return PriceData.model_construct(
-                currency_symbol=None, datetime_high=None, datetime_last=None, datetime_low=None,
-                datetime_open=None, high=None, last=None, low=None, open=None
+                currency_symbol=None,
+                datetime_high=None,
+                datetime_last=None,
+                datetime_low=None,
+                datetime_open=None,
+                high=None,
+                last=None,
+                low=None,
+                open=None,
             )
 
         return PriceData.model_construct(