Created Cally

backend/servicem8.py

@@ -0,0 +1,112 @@
+import requests
+import os
+from datetime import datetime, date, timedelta
+from sqlalchemy.orm import Session
+from . import cally_models
+from dotenv import load_dotenv
+
+load_dotenv()
+
+SM8_TOKEN = os.getenv("SERVICEM8_ACCESS_TOKEN")
+BASE_URL = "https://api.servicem8.com/api_1.0"
+
+def get_sm8_headers():
+    return {
+        "X-Api-Key": SM8_TOKEN,
+        "Accept": "application/json",
+        "Content-Type": "application/json"
+    }
+
+def sync_from_servicem8(db: Session):
+    headers = get_sm8_headers()
+    
+    # 1. Sync Staff
+    staff_resp = requests.get(f"{BASE_URL}/staff.json", headers=headers)
+    if staff_resp.status_code == 200:
+        staff_list = staff_resp.json()
+        for s in staff_list:
+            # We only care about active staff who are shown on the schedule
+            if s.get('active') == 1 and s.get('hide_from_schedule') == 0:
+                first = s.get('first', '')
+                last = s.get('last', '')
+                db_staff = cally_models.Sm8Staff(
+                    uuid=s.get('uuid'),
+                    name=f"{first} {last}".strip() or "Unknown Staff",
+                    active=True
+                )
+                db.merge(db_staff)
+        db.commit()
+
+    # 2. Sync Schedule (Fetch last 30 days and next 60 days)
+    start_date = (date.today() - timedelta(days=31)).isoformat()
+    # ServiceM8 API v1.0 OData does not support 'ge', but supports 'gt'
+    # We use 'and' to combine filters
+    params = {
+        "$filter": f"activity_was_scheduled eq 1 and start_date gt '{start_date} 00:00:00'"
+    }
+    
+    activity_resp = requests.get(f"{BASE_URL}/jobactivity.json", headers=headers, params=params)
+    if activity_resp.status_code == 200:
+        activities = activity_resp.json()
+        
+        # Clear existing cached schedule in the range to avoid duplicates
+        db.query(cally_models.ScheduleDay).delete()
+        
+        # Track busy days: {staff_uuid: {date_str: [job_uuids]}}
+        busy_map = {}
+        unique_job_uuids = set()
+        
+        for a in activities:
+            staff_uuid = a.get('staff_uuid')
+            start_dt_str = a.get('start_date')
+            job_uuid = a.get('job_uuid')
+            if not staff_uuid or not start_dt_str or not job_uuid:
+                continue
+                
+            # Extract date from 'YYYY-MM-DD HH:MM:SS'
+            activity_date_str = start_dt_str.split(' ')[0]
+            
+            if staff_uuid not in busy_map:
+                busy_map[staff_uuid] = {}
+            
+            if activity_date_str not in busy_map[staff_uuid]:
+                busy_map[staff_uuid][activity_date_str] = []
+            
+            if job_uuid not in busy_map[staff_uuid][activity_date_str]:
+                busy_map[staff_uuid][activity_date_str].append(job_uuid)
+                unique_job_uuids.add(job_uuid)
+            
+        # 3. Sync Job Details for found activities
+        # We need human-readable IDs (#123) from the 'job' table
+        for job_uuid in unique_job_uuids:
+            # Check if we already have this job detail
+            existing_job = db.query(cally_models.Sm8Job).filter(cally_models.Sm8Job.uuid == job_uuid).first()
+            if not existing_job:
+                job_resp = requests.get(f"{BASE_URL}/job/{job_uuid}.json", headers=headers)
+                if job_resp.status_code == 200:
+                    j = job_resp.json()
+                    db_job = cally_models.Sm8Job(
+                        uuid=job_uuid,
+                        job_id=j.get('generated_job_id', 'Unknown'),
+                        address=j.get('job_address', '')
+                    )
+                    db.add(db_job)
+        db.commit()
+
+        # 4. Save Schedule to DB
+        for staff_uuid, dates_dict in busy_map.items():
+            for d_str, job_list in dates_dict.items():
+                d_obj = date.fromisoformat(d_str)
+                db_day = cally_models.ScheduleDay(
+                    staff_uuid=staff_uuid,
+                    date=d_obj,
+                    is_busy=True,
+                    job_count=len(job_list),
+                    job_uuids=",".join(job_list)
+                )
+                db.add(db_day)
+        
+        db.commit()
+        return True
+    
+    return False