Create main.py

main.py

@@ -0,0 +1,251 @@
+import os
+import json
+import uuid
+import datetime
+import pandas as pd
+from flask import Flask, request, jsonify
+from flask_cors import CORS
+from werkzeug.utils import secure_filename
+from flask_apscheduler import APScheduler
+import firebase_admin
+from firebase_admin import credentials, db, auth as firebase_auth
+from resend import Emails
+
+# === ENV Config ===
+RESEND_API_KEY = os.getenv("RESEND_API_KEY")
+FIREBASE_CRED_JSON = json.loads(os.getenv("FIREBASE"))
+FIREBASE_DB_URL = os.getenv("Firebase_DB")
+ADMIN_EMAILS = ["rairorr@gmail.com", "nharingoshepherd@gmail.com"]
+
+# === Firebase Init ===
+cred = credentials.Certificate(FIREBASE_CRED_JSON)
+firebase_admin.initialize_app(cred, {
+    "databaseURL": FIREBASE_DB_URL
+})
+
+admin_emails = [
+    "rairorr@gmail.com",
+    "nharingoshepherd@gmail.com"
+]
+
+for email in admin_emails:
+    key = email.replace("@", "_").replace(".", "_")
+    db.reference(f"admins/{key}").set({
+        "email": email,
+        "is_admin": True
+    })
+
+print("✅ Admins added.")
+
+# === Flask App Setup ===
+app = Flask(__name__)
+CORS(app)
+
+# === APScheduler Setup ===
+class Config(object):
+    SCHEDULER_API_ENABLED = True
+
+app.config.from_object(Config())
+
+scheduler = APScheduler()
+scheduler.init_app(app)
+scheduler.start()
+
+# === Resend Init ===
+Emails.api_key = RESEND_API_KEY
+
+def send_email(to, subject, html):
+    try:
+        Emails.send({
+            "from": "Admin <admin@resend.dev>",
+            "to": to,
+            "subject": subject,
+            "html": html
+        })
+    except Exception as e:
+        print("Email error:", e)
+
+# === Auth Middleware ===
+def verify_token(req):
+    token = req.headers.get("Authorization")
+    if not token:
+        return None
+    try:
+        decoded = firebase_auth.verify_id_token(token.replace("Bearer ", ""))
+        if decoded["email"] not in ADMIN_EMAILS:
+            return None
+        return decoded
+    except:
+        return None
+
+# === Admin Setup ===
+def setup_admins():
+    ref = db.reference("admins")
+    for email in ADMIN_EMAILS:
+        uid = f"admin_{email.replace('@', '_').replace('.', '_')}"
+        ref.child(uid).set({"email": email, "is_admin": True})
+setup_admins()
+
+# === Core Functions ===
+def assign_roster(job_id):
+    """Assign roster for a job - now a regular function instead of Celery task"""
+    try:
+        # Get job details
+        job_ref = db.reference(f"jobs/{job_id}")
+        job_data = job_ref.get()
+        
+        if not job_data:
+            print(f"Job {job_id} not found")
+            return
+            
+        # Get all members
+        members_ref = db.reference("members")
+        members = members_ref.get()
+        
+        if not members:
+            print("No members found")
+            return
+            
+        # Convert members to list
+        members_list = list(members.values()) if isinstance(members, dict) else members
+        
+        # Simple rotation logic (you can enhance this)
+        current_assignments = job_data.get("assignments", [])
+        last_assigned_index = job_data.get("last_assigned_index", -1)
+        
+        # Find next member to assign
+        next_index = (last_assigned_index + 1) % len(members_list)
+        assigned_member = members_list[next_index]
+        
+        # Create assignment record
+        assignment = {
+            "member": assigned_member,
+            "assigned_at": datetime.datetime.now().isoformat(),
+            "assignment_id": str(uuid.uuid4())
+        }
+        
+        # Update assignments
+        current_assignments.append(assignment)
+        
+        # Update job with new assignment and index
+        job_ref.update({
+            "assignments": current_assignments,
+            "last_assigned_index": next_index,
+            "last_updated": datetime.datetime.now().isoformat()
+        })
+        
+        print(f"Assigned {assigned_member.get('name', 'Unknown')} to job {job_id}")
+        
+        # Send notification email (optional)
+        # send_email(assigned_member.get('email'), "New Assignment", f"You have been assigned to job {job_id}")
+        
+    except Exception as e:
+        print(f"Error in assign_roster for job {job_id}: {e}")
+
+def check_and_rotate_guards():
+    """Periodic function to check jobs and rotate guards"""
+    try:
+        print("🔍 Checking for guard rotations...")
+        
+        # Get all jobs
+        jobs_ref = db.reference("jobs")
+        jobs = jobs_ref.get()
+        
+        if not jobs:
+            print("No jobs found")
+            return
+            
+        # Check each job for rotation needs
+        for job_id, job_data in jobs.items():
+            # Check if job needs rotation (example logic - customize as needed)
+            assignments = job_data.get("assignments", [])
+            
+            if not assignments:
+                # If no assignments yet, start the first one
+                print(f"Starting first assignment for job {job_id}")
+                assign_roster(job_id)
+            else:
+                # Check if current assignment is expired (example: 8 hours)
+                last_assignment = assignments[-1]
+                assigned_at = datetime.datetime.fromisoformat(last_assignment["assigned_at"].replace("Z", "+00:00"))
+                now = datetime.datetime.now(assigned_at.tzinfo)
+                
+                # Rotate every 8 hours (customize as needed)
+                if (now - assigned_at).total_seconds() > 28800:  # 8 hours in seconds
+                    print(f"Rotating guard for job {job_id}")
+                    assign_roster(job_id)
+                    
+    except Exception as e:
+        print(f"Error in check_and_rotate_guards: {e}")
+
+# === Routes ===
+@app.route("/upload_members", methods=["POST"])
+def upload_members():
+    user = verify_token(request)
+    if not user:
+        return jsonify({"error": "Unauthorized"}), 401
+
+    file = request.files.get("file")
+    if not file:
+        return jsonify({"error": "No file uploaded"}), 400
+
+    df = pd.read_csv(file) if file.filename.endswith(".csv") else pd.read_excel(file)
+    members = df.to_dict(orient="records")
+
+    for member in members:
+        member_id = str(uuid.uuid4())
+        db.reference(f"members/{member_id}").set(member)
+
+    return jsonify({"message": "Members uploaded successfully"}), 200
+
+@app.route("/add_member", methods=["POST"])
+def add_member():
+    user = verify_token(request)
+    if not user:
+        return jsonify({"error": "Unauthorized"}), 401
+
+    data = request.json
+    member_id = str(uuid.uuid4())
+    db.reference(f"members/{member_id}").set(data)
+    return jsonify({"message": "Member added"}), 200
+
+@app.route("/create_job", methods=["POST"])
+def create_job():
+    user = verify_token(request)
+    if not user:
+        return jsonify({"error": "Unauthorized"}), 401
+
+    data = request.json
+    job_id = str(uuid.uuid4())
+    db.reference(f"jobs/{job_id}").set(data)
+    return jsonify({"message": "Job created", "job_id": job_id}), 200
+
+@app.route("/start_job/<job_id>", methods=["POST"])
+def start_job(job_id):
+    user = verify_token(request)
+    if not user:
+        return jsonify({"error": "Unauthorized"}), 401
+
+    # Schedule the first assignment immediately
+    scheduler.add_job(
+        func=assign_roster,
+        trigger="date",
+        run_date=datetime.datetime.now() + datetime.timedelta(seconds=5),
+        args=[job_id],
+        id=f"start_{job_id}_{uuid.uuid4().hex[:8]}"
+    )
+    
+    return jsonify({"message": f"Roster assignment started for job {job_id}"}), 202
+
+# === Schedule periodic guard rotation ===
+# Run every 5 minutes to check for rotations
+scheduler.add_job(
+    func=check_and_rotate_guards,
+    trigger="interval",
+    minutes=5,
+    id="guard_rotation_job"
+)
+
+# === Run Server ===
+if __name__ == "__main__":
+    app.run(debug=True, port=int(os.getenv("PORT", 7860)))