Update src/services/journal_engine.py

src/services/journal_engine.py

@@ -3,6 +3,7 @@ import io
 import os
 import re
 import uuid
+import datetime  # <--- NEW IMPORT
 from google_auth_oauthlib.flow import Flow
 from googleapiclient.discovery import build
 from googleapiclient.http import MediaIoBaseUpload, MediaIoBaseDownload
@@ -39,6 +40,7 @@ class JournalEngine:
         except Exception as e:
             print(f"⚠️ Token Load Error: {e}")
             return None
+
     @staticmethod
     def get_drive_service(creds):
         return build('drive', 'v3', credentials=creds)
@@ -70,7 +72,6 @@ class JournalEngine:
     @staticmethod
     def initialize_journal(service):
         try:
-            # Check if file exists specifically in appDataFolder
             response = service.files().list(
                 q="name='journal.json' and 'appDataFolder' in parents",
                 spaces='appDataFolder',
@@ -96,18 +97,26 @@ class JournalEngine:
 
     @classmethod
     def save_trade(cls, service, file_id, trade_data):
-        """Smart Update: Edits existing ID or Appends new UUID."""
+        """Smart Update: Edits existing ID or Appends new UUID + Auto-Tags Date."""
         journal = cls.load_journal(service, file_id)
         
+        # --- PHASE 1 UPGRADE: Temporal Injection ---
+        if 'trade_date' in trade_data:
+            try:
+                dt = datetime.datetime.strptime(trade_data['trade_date'], "%Y-%m-%d")
+                trade_data['week'] = dt.strftime("%Y-W%W") # e.g. 2026-W05
+                trade_data['month'] = dt.strftime("%Y-%m") # e.g. 2026-02
+            except ValueError:
+                pass # Keep going even if date parse fails
+        # -------------------------------------------
+
         trade_id = trade_data.get('id')
         updated = False
         
-        # Ensure ID exists for new trades
         if not trade_id:
             trade_data['id'] = str(uuid.uuid4())
             journal.append(trade_data)
         else:
-            # Try to find and update
             for i, existing in enumerate(journal):
                 if existing.get('id') == trade_id:
                     journal[i] = trade_data
@@ -119,6 +128,19 @@ class JournalEngine:
         cls.save_to_drive(service, file_id, journal)
         return True
 
+    @classmethod
+    def delete_trade(cls, service, file_id, trade_id):
+        """Removes a trade by UUID and saves the list."""
+        journal = cls.load_journal(service, file_id)
+        
+        # Filter out the trade with matching ID
+        new_journal = [t for t in journal if t.get('id') != trade_id]
+        
+        if len(new_journal) < len(journal):
+            cls.save_to_drive(service, file_id, new_journal)
+            return True
+        return False
+
     @staticmethod
     def parse_pnl(pnl_str):
         try: