feat: add fetch_price_info() method on subscription

Author: ericbstie

tibber/types/subscription.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 """A class representing the Subscription type from the GraphQL Tibber API."""
-from typing import TYPE_CHECKING, Optional
+from typing import TYPE_CHECKING, Optional, Literal
 
 from tibber.networking.query_builder import QueryBuilder
 from tibber.types.legal_entity import LegalEntity
@@ -50,6 +50,31 @@ def price_info(self) -> PriceInfo:
         """Price information related to the subscription"""
         return PriceInfo(self.cache.get("priceInfo"), self.tibber_client)
 
+    def fetch_price_info(self,
+        resolution: Literal["HOURLY", "QUARTER_HOURLY"],
+        home_id: Optional[str] = None
+    ) -> PriceInfo:
+        price_info_query_dict = QueryBuilder.price_info_query(resolution)
+
+        price_info_query = QueryBuilder.create_query(
+            "viewer", "homes", "currentSubscription", price_info_query_dict
+        )
+
+        full_data = self.tibber_client.execute_query(
+            self.tibber_client.token, price_info_query
+        )
+
+        home = full_data["viewer"]["homes"][0]
+        if home_id:
+            home_of_id = [
+                home for home in full_data["viewer"]["homes"] if home["id"] == home_id
+            ][0]
+
+            if home_of_id:
+                home = home_of_id
+
+        return PriceInfo(home["currentSubscription"]["priceInfo"], self.tibber_client)
+
     def fetch_price_info_range(
         self,
         resolution: str,