pidatametrics1 0.3.4__tar.gz → 0.3.6__tar.gz

{pidatametrics1-0.3.4 → pidatametrics1-0.3.6}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: pidatametrics1
-Version: 0.3.4
-Summary: A wrapper for Pi Datametrics API with CSV and BigQuery support.
+Version: 0.3.6
+Summary: A test wrapper for Pi Datametrics API with CSV and BigQuery support.
 Requires-Dist: google-auth
 Requires-Dist: google-cloud-bigquery
 Requires-Dist: gspread

{pidatametrics1-0.3.4 → pidatametrics1-0.3.6}/pyproject.toml

@@ -4,8 +4,8 @@ build-backend = "hatchling.build"
 
 [project]
 name = "pidatametrics1"
-version = "0.3.4"
-description = "A wrapper for Pi Datametrics API with CSV and BigQuery support."
+version = "0.3.6"
+description = "A test wrapper for Pi Datametrics API with CSV and BigQuery support."
 dependencies = [
     "requests",
     "google-cloud-bigquery",

{pidatametrics1-0.3.4 → pidatametrics1-0.3.6}/src/pidatametrics/manager.py

@@ -5,6 +5,24 @@ import datetime
 from dateutil.relativedelta import relativedelta
 
 class PiReportManager(PiDataMetrics):
+    
+    # --- HELPER: Generate Unique Name ---
+    def _generate_unique_name(self, base_name, workspace_ref=None):
+        """
+        Creates a unique name like: Hist_51780_0502_1430
+        (Base_WorkspaceID_Date_Time)
+        """
+        now = datetime.datetime.now()
+        timestamp = now.strftime("%d%m_%H%M") # e.g., 0502_1430
+        
+        ws_part = f"_{workspace_ref}" if workspace_ref else ""
+        
+        # Combine parts
+        full_name = f"{base_name}{ws_part}_{timestamp}"
+        
+        # Truncate to 31 chars (Google Sheets limit) just in case
+        return full_name[:31]
+
     def _resolve_workspaces(self, ids_str=None, name_pattern=None):
         all_ws = self.get_workspaces()
         targets = {}
@@ -43,21 +61,34 @@ class PiReportManager(PiDataMetrics):
                 current_date -= relativedelta(months=1)
         return dates
 
-    # --- HELPER METHOD FOR EXPORTING ---
-    def _export_data(self, data, output_mode, filename, bq_config, spreadsheet_name, tab_name="Sheet1"):
+    # --- UPDATED EXPORT LOGIC ---
+    def _export_data(self, data, output_mode, bq_config, spreadsheet_name, unique_name):
+        """
+        Handles export routing.
+        unique_name is used for:
+        - CSV Filename
+        - Excel Filename
+        - Google Sheet Tab Name
+        """
         if not data:
             print("No data to export.")
             return
 
         if output_mode == 'bigquery' and bq_config:
+            # BigQuery doesn't use filenames, it uses the Table ID in config
             PiExporter.to_bigquery(data, bq_config['project'], bq_config['dataset'], bq_config['table'])
+        
         elif output_mode == 'excel':
-            PiExporter.to_excel(data, filename)
+            # Use unique_name as filename
+            PiExporter.to_excel(data, unique_name)
+            
         elif output_mode == 'gsheet' and spreadsheet_name:
-            PiExporter.to_google_sheet(data, spreadsheet_name, tab_name)
+            # Use unique_name as Tab Name
+            PiExporter.to_google_sheet(data, spreadsheet_name, tab_name=unique_name)
+            
         else:
-            # Default to CSV
-            PiExporter.to_csv(data, filename)
+            # Default to CSV, use unique_name as filename
+            PiExporter.to_csv(data, unique_name)
 
     def run_volume_report(self, filename, workspace_ids=None, workspace_name=None, output_mode='csv', bq_config=None, spreadsheet_name=None):
         targets = self._resolve_workspaces(workspace_ids, workspace_name)
@@ -71,7 +102,11 @@ class PiReportManager(PiDataMetrics):
                 rows = PiParsers.parse_volume_data(vol_data, stg['name'], terms, ws_name)
                 all_rows.extend(rows)
         
-        self._export_data(all_rows, output_mode, filename, bq_config, spreadsheet_name, tab_name="Volume")
+        # Generate Unique Name
+        ws_ref = list(targets.keys())[0] if targets else "Multi"
+        unique_name = self._generate_unique_name("Vol", ws_ref)
+
+        self._export_data(all_rows, output_mode, bq_config, spreadsheet_name, unique_name)
 
     def run_serp_report(self, data_sources, output_mode='csv', bq_config=None, filename=None, manual_duplication=None, spreadsheet_name=None):
         yesterday = (datetime.datetime.now() - datetime.timedelta(days=1)).strftime("%Y-%m-%d")
@@ -83,7 +118,11 @@ class PiReportManager(PiDataMetrics):
             rows = PiParsers.parse_serp_response(raw_data, market, w_name, se_name, yesterday, cat_map, manual_duplication)
             all_rows.extend(rows)
         
-        self._export_data(all_rows, output_mode, filename or "serp_output", bq_config, spreadsheet_name, tab_name="SERP")
+        # Generate Unique Name
+        ws_ref = data_sources[0][1] if data_sources else "All"
+        unique_name = self._generate_unique_name("SERP", ws_ref)
+
+        self._export_data(all_rows, output_mode, bq_config, spreadsheet_name, unique_name)
 
     def run_historical_serp_report(self, data_sources, duration, frequency, start_date=None, features=None, num_results=25, output_mode='csv', bq_config=None, filename="historical_data", spreadsheet_name=None):
         if features is None:
@@ -96,12 +135,11 @@ class PiReportManager(PiDataMetrics):
         
         print(f"Starting Historical Report ({frequency}) for last {duration} periods...")
         
-        all_file_rows = [] # Used for CSV, Excel, and GSheet
+        all_file_rows = [] 
 
         for i, date in enumerate(target_dates):
             print(f"[{i+1}/{len(target_dates)}] Processing Date: {date}...")
-            
-            daily_rows = [] # Reset container for this specific date
+            daily_rows = [] 
 
             for source in data_sources:
                 market, w_id, w_name, se_id, se_name = source
@@ -111,31 +149,26 @@ class PiReportManager(PiDataMetrics):
                         'serp-feature[]': features
                     }
                     raw_data = self.get_bulk_serp_data(w_id, se_id, date, **params)
-                    
-                    rows = PiParsers.parse_serp_response(
-                        raw_data, market, w_name, se_name, date, category_map=None
-                    )
-                    
+                    rows = PiParsers.parse_serp_response(raw_data, market, w_name, se_name, date, category_map=None)
                     daily_rows.extend(rows)
-
                 except Exception as e:
                     print(f"Failed to fetch {w_name} on {date}: {e}")
 
-            # --- UPLOAD LOGIC: PER DATE (BigQuery Only) ---
+            # BigQuery uploads immediately per day
             if output_mode == 'bigquery' and bq_config:
                 if daily_rows:
                     print(f"Uploading {len(daily_rows)} rows for {date} to BigQuery...")
                     PiExporter.to_bigquery(daily_rows, bq_config['project'], bq_config['dataset'], bq_config['table'])
-                else:
-                    print(f"No data found for {date}, skipping upload.")
-            
-            # --- FILE LOGIC: ACCUMULATE (CSV, Excel, GSheet) ---
+            # Others accumulate
             elif output_mode in ['csv', 'excel', 'gsheet']:
                 all_file_rows.extend(daily_rows)
 
-        # Final Save for File-based outputs
+        # Final Export for Files
         if output_mode in ['csv', 'excel', 'gsheet']:
-            self._export_data(all_file_rows, output_mode, filename, bq_config, spreadsheet_name, tab_name="Historical")
+            ws_ref = data_sources[0][1] if data_sources else "All"
+            unique_name = self._generate_unique_name("Hist", ws_ref)
+            
+            self._export_data(all_file_rows, output_mode, bq_config, spreadsheet_name, unique_name)
 
     def run_llm_report(self, data_sources, start_period, end_period, stg_ids=None, output_mode='csv', bq_config=None, filename="llm_output", spreadsheet_name=None):
         all_rows = []
@@ -146,11 +179,14 @@ class PiReportManager(PiDataMetrics):
             try:
                 print(f"Fetching LLM data for {w_name} ({se_name})...")
                 raw_data = self.get_llm_mentions(w_id, se_id, start_period, end_period, stg_ids)
-                
                 rows = PiParsers.parse_llm_response(raw_data, market, w_name, se_name)
                 all_rows.extend(rows)
                 print(f"Found {len(rows)} mentions/queries.")
             except Exception as e:
                 print(f"Failed to fetch LLM data for {w_name}: {e}")
 
-        self._export_data(all_rows, output_mode, filename, bq_config, spreadsheet_name, tab_name="LLM_Mentions")
+        # Generate Unique Name
+        ws_ref = data_sources[0][1] if data_sources else "All"
+        unique_name = self._generate_unique_name("LLM", ws_ref)
+
+        self._export_data(all_rows, output_mode, bq_config, spreadsheet_name, unique_name)