hawk-sdk 0.0.10__tar.gz → 0.0.11__tar.gz

{hawk_sdk-0.0.10 → hawk_sdk-0.0.11}/PKG-INFO

@@ -1,5 +1,5 @@
 Metadata-Version: 2.1
 Name: hawk-sdk
-Version: 0.0.10
+Version: 0.0.11
 Requires-Dist: google-cloud-bigquery
 Requires-Dist: pandas

{hawk_sdk-0.0.10 → hawk_sdk-0.0.11}/hawk_sdk/api/equities/repository.py

@@ -23,15 +23,20 @@ class EquitiesRepository:
 
     def fetch_adjusted_ohlc(self, start_date: str, end_date: str, interval: str, hawk_ids: List[int]) -> Iterator[dict]:
         """Fetches raw adjusted OHLC data from BigQuery for the given date range and hawk_ids using query parameters."""
+        open_field = f"adjusted_open_{interval}"
+        high_field = f"adjusted_high_{interval}"
+        low_field = f"adjusted_low_{interval}"
+        close_field = f"adjusted_close_{interval}"
+
         query = f"""
         WITH records_data AS (
           SELECT 
             r.record_timestamp AS date,
             hi.value AS ticker,
-            MAX(CASE WHEN f.field_name = @open_field THEN r.double_value END) AS @open_field,
-            MAX(CASE WHEN f.field_name = @high_field THEN r.double_value END) AS @high_field,
-            MAX(CASE WHEN f.field_name = @low_field THEN r.double_value END) AS @low_field,
-            MAX(CASE WHEN f.field_name = @close_field THEN r.double_value END) AS @close_field
+            MAX(CASE WHEN f.field_name = @open_field THEN r.double_value END) AS {open_field},
+            MAX(CASE WHEN f.field_name = @high_field THEN r.double_value END) AS {high_field},
+            MAX(CASE WHEN f.field_name = @low_field THEN r.double_value END) AS {low_field},
+            MAX(CASE WHEN f.field_name = @close_field THEN r.double_value END) AS {close_field}
           FROM 
             `wsb-hc-qasap-ae2e.{self.environment}.records` AS r
           JOIN 
@@ -50,24 +55,26 @@ class EquitiesRepository:
         SELECT DISTINCT
           date,
           ticker,
-          @open_field,
-          @high_field,
-          @low_field,
-          @close_field,
+          {open_field},
+          {high_field},
+          {low_field},
+          {close_field},
         FROM 
           records_data
         ORDER BY 
           date;
         """
 
+
+
         query_params = [
             bigquery.ArrayQueryParameter("hawk_ids", "INT64", hawk_ids),
             bigquery.ScalarQueryParameter("start_date", "STRING", start_date),
             bigquery.ScalarQueryParameter("end_date", "STRING", end_date),
-            bigquery.ScalarQueryParameter("open_field", "STRING", f"adjusted_open_{interval}"),
-            bigquery.ScalarQueryParameter("high_field", "STRING", f"adjusted_high_{interval}"),
-            bigquery.ScalarQueryParameter("low_field", "STRING", f"adjusted_low_{interval}"),
-            bigquery.ScalarQueryParameter("close_field", "STRING", f"adjusted_close_{interval}")
+            bigquery.ScalarQueryParameter("open_field", "STRING", open_field),
+            bigquery.ScalarQueryParameter("high_field", "STRING", high_field),
+            bigquery.ScalarQueryParameter("low_field", "STRING", low_field),
+            bigquery.ScalarQueryParameter("close_field", "STRING", close_field)
         ]
 
         job_config = bigquery.QueryJobConfig(query_parameters=query_params)

{hawk_sdk-0.0.10 → hawk_sdk-0.0.11}/hawk_sdk.egg-info/PKG-INFO

@@ -1,5 +1,5 @@
 Metadata-Version: 2.1
 Name: hawk-sdk
-Version: 0.0.10
+Version: 0.0.11
 Requires-Dist: google-cloud-bigquery
 Requires-Dist: pandas

{hawk_sdk-0.0.10 → hawk_sdk-0.0.11}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='hawk-sdk',
-    version='0.0.10',
+    version='0.0.11',
     packages=find_packages(),
     install_requires=[
         'google-cloud-bigquery',