Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import json
 import logging
+import datetime
 from datetime import datetime, timedelta, timezone
 from typing import Optional, List
 from contextlib import asynccontextmanager
@@ -18,7 +19,7 @@ from google import genai
 import bcrypt
 
 # ==========================================
-# بخش اول: سیستم ثبت لاگ خطاهای پشت‌صحنه سرور
+# بخش اول: سیستم ثبت لاگ خطارهای سرور
 # ==========================================
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger("production_backend")
@@ -27,7 +28,7 @@ logger = logging.getLogger("production_backend")
 AI_CORE_KEY = os.getenv("GEMINI_API_KEY") or os.getenv("SECRET")
 
 # ==========================================
-# بخش دوم: اتصال پایگاه داده Neon.tech و سوئیچ به SQLite
+# بخش دوم: اتصال پایگاه داده Neon.tech و ساختار پایه
 # ==========================================
 DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///./test.db")
 if DATABASE_URL.startswith("postgres://"):
@@ -51,15 +52,12 @@ except Exception as db_init_err:
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 Base = declarative_base()
 
-# ==========================================
-# بخش سوم: توابع همزمانی پایه (حل قطعی NameError)
-# ==========================================
+# تعریف تابع زمان بومی سرور
 def get_now():
-    """تولید زمان بومی و دقیق کنونی جهت هماهنگی با فرآیندهای دیتابیس"""
     return datetime.now(timezone.utc).replace(tzinfo=None)
 
 # ==========================================
-# بخش چهارم: مدل‌های دیتابیس (SQLAlchemy Models)
+# بخش سوم: مدل‌های دیتابیس (SQLAlchemy Models)
 # ==========================================
 
 class User(Base):
@@ -214,7 +212,7 @@ class Review(Base):
     comment = Column(Text, nullable=True)
 
 # ==========================================
-# بخش پنجم: توابع کمکی امنیتی و رمزنگاری
+# بخش چهارم: توابع کمکی امنیتی و رمزنگاری
 # ==========================================
 
 def get_password_hash(password: str) -> str:
@@ -232,14 +230,8 @@ SECRET_KEY = os.getenv("JWT_SECRET", "8c12fa4870a23049102c89eeffb8a719280abdc029
 ALGORITHM = "HS256"
 security = HTTPBearer()
 
-def create_access_token(data: dict, expires_delta: Optional[timedelta] = None):
-    to_encode = data.copy()
-    expire = get_now() + (expires_delta or timedelta(minutes=1440))
-    to_encode.update({"exp": expire})
-    return jwt.encode(to_encode, SECRET_KEY, algorithm=ALGORITHM)
-
 # ==========================================
-# بخش ششم: متدهای پاک‌سازی دوره‌ای و تزریق وابستگی دیتابیس
+# بخش پنجم: توابع وابستگی تزریق پایگاه داده و احراز هویت
 # ==========================================
 
 def clean_expired_bookings(db: Session):
@@ -264,7 +256,7 @@ def clean_expired_bookings(db: Session):
 
 def get_db():
     db = SessionLocal()
-    clean_expired_bookings(db)  # اجرای آزادسازی خودکار نوبت‌ها
+    clean_expired_bookings(db)  # اجرای مکانیزم آزادسازی نوبت‌ها در لحظه لود دیتابیس
     try:
         yield db
     finally:
@@ -292,7 +284,7 @@ def get_current_user(db: Session = Depends(get_db), credentials: HTTPAuthorizati
     return user
 
 # ==========================================
-# بخش هفتم: توابع پردازش ابری هوشمند بومی
+# بخش ششم: توابع پردازش ابری هوشمند بومی تاپ دانش
 # ==========================================
 
 def _execute_native_matrix_processing(prompt: str, model_type: str = "fast") -> str:
@@ -376,7 +368,7 @@ def core_bio_extraction(resume_text: str) -> str:
     return result
 
 # ==========================================
-# بخش هشتم: کلاس‌های ورودی Pydantic
+# بخش هفتم: کلاس‌های ورودی Pydantic (صحت‌سنجی)
 # ==========================================
 
 class RegisterRequest(BaseModel):
@@ -476,68 +468,15 @@ class UpdateProfileRequest(BaseModel):
     city: Optional[str] = None
 
 # ==========================================
-# بخش نهم: ساختار فرآیند بارگذاری Lifespan و Seeding
+# بخش هشتم: راه‌اندازی و اجرای وب‌سرور FastAPI
 # ==========================================
 
-def seed_initial_data(db: Session):
-    try:
-        test_pwd = get_password_hash("demo1234")
-        
-        # دانشجویان پیش‌فرض
-        student = User(email="student@demo.com", hashed_password=test_pwd, role="student", is_verified=True, balance=500000.0, phone_number="09121111111", city="تهران")
-        db.add(student)
-        
-        # مشاوران پیش‌فرض
-        counselor1 = User(email="counselor1@demo.com", hashed_password=test_pwd, role="counselor", is_verified=True, rating=4.8, city="تهران")
-        counselor2 = User(email="counselor2@demo.com", hashed_password=test_pwd, role="counselor", is_verified=True, rating=4.5, city="شیراز")
-        db.add(counselor1)
-        db.add(counselor2)
-        db.flush()
-        
-        p1 = CounselorProfile(user_id=counselor1.id, specialty="برنامه‌ریزی کنکور تجربی", resume="رتبه دو رقمی سراسری با سابقه تدریس تخصصی", bio="مشاور ارشد با تخصص رتبه‌پروری تجربی", hourly_rate=120000.0)
-        p2 = CounselorProfile(user_id=counselor2.id, specialty="هدایت تحصیلی ریاضی و فیزیک", resume="عضو بنیاد نخبگان ریاضی", bio="متخصص بهینه‌سازی مسیر قبولی رشته ریاضی", hourly_rate=100000.0)
-        db.add(p1)
-        db.add(p2)
-        
-        # آموزشگاه‌های پیش‌فرض
-        school_user = User(email="school@demo.com", hashed_password=test_pwd, role="school_representative", is_verified=True, city="اصفهان")
-        db.add(school_user)
-        db.flush()
-        
-        s_profile = SchoolProfile(
-            user_id=school_user.id, 
-            name="موسسه کنکور اندیشه", 
-            subscription_active_until=get_now() + timedelta(days=60), 
-            is_sponsored=True,
-            bio="برترین آکادمی تستی کنکور اصفهان",
-            address="اصفهان، خیابان دانشگاه",
-            phone_number="03133333333",
-            founding_year=1395
-        )
-        db.add(s_profile)
-        db.flush()
-        
-        course = SchoolCourse(school_id=s_profile.id, course_name="ریاضی تخصصی کنکور", teacher_name="استاد علوی", teaching_style="روش‌های سرعتی و محاسباتی")
-        db.add(course)
-        
-        # اسلات آزاد مشاور
-        now = get_now()
-        slot1 = TimeSlot(counselor_id=counselor1.id, start_time=now + timedelta(days=1, hours=10), end_time=now + timedelta(days=1, hours=11), is_booked=False)
-        db.add(slot1)
-        
-        db.commit()
-    except Exception as e:
-        logger.error(f"Seeding process crashed: {e}", exc_info=True)
-        db.rollback()
-
-
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     Base.metadata.create_all(bind=engine)
     db = SessionLocal()
     try:
-        if db.query(User).count() == 0:
-            seed_initial_data(db)
+        seed_initial_data(db)
     except Exception as seed_err:
         logger.error(f"Error seeding database during startup: {seed_err}", exc_info=True)
     finally:
@@ -559,6 +498,89 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+# ==========================================
+# بخش نهم: فرآیند ثبت و همگام‌سازی داده‌های تستی و دمو (Check-Seeding)
+# ==========================================
+
+def seed_initial_data(db: Session):
+    try:
+        # لیست جامع اکانت‌های پیش‌فرض قدیمی و جدید با کلمات عبور هماهنگ با کلاینت اندروید
+        seed_users = [
+            {"email": "student@demo.com", "password": "demo1234", "role": "student", "balance": 500000.0, "phone": "09121111111", "city": "تهران"},
+            {"email": "student@test.com", "password": "student123", "role": "student", "balance": 300000.0, "phone": "09122222222", "city": "تهران"},
+            {"email": "counselor1@demo.com", "password": "demo1234", "role": "counselor", "rating": 4.8, "city": "تهران"},
+            {"email": "counselor2@demo.com", "password": "demo1234", "role": "counselor", "rating": 4.5, "city": "شیراز"},
+            {"email": "counselor@test.com", "password": "counselor123", "role": "counselor", "rating": 4.7, "city": "تهران"},
+            {"email": "school@demo.com", "password": "demo1234", "role": "school_representative", "city": "اصفهان"},
+            {"email": "school@test.com", "password": "school123", "role": "school_representative", "city": "تهران"}
+        ]
+        
+        for u_data in seed_users:
+            # بررسی تک تک کاندیداها به صورت پویا (سازگار با داده‌های قبلی Neon)
+            existing = db.query(User).filter(User.email == u_data["email"]).first()
+            if not existing:
+                new_user = User(
+                    email=u_data["email"],
+                    hashed_password=get_password_hash(u_data["password"]),
+                    role=u_data["role"],
+                    is_verified=True,
+                    balance=u_data.get("balance", 0.0),
+                    rating=u_data.get("rating", 5.0),
+                    phone_number=u_data.get("phone"),
+                    city=u_data.get("city")
+                )
+                db.add(new_user)
+                db.flush()
+                
+                # ثبت دقیق پروفایل‌های همبسته
+                if u_data["role"] == "counselor":
+                    p = CounselorProfile(
+                        user_id=new_user.id,
+                        specialty="برنامه‌ریزی کنکور و هدایت تحصیلی",
+                        resume="رتبه برتر کنکور و کارشناس باسابقه پلتفرم تراز تحصیلی بومی تاپ دانش",
+                        bio="متخصص افزایش راندمان مطالعه داوطلبان با رویکردهای نوین مشاوره کلاسی",
+                        hourly_rate=120000.0
+                    )
+                    db.add(p)
+                    
+                    # ایجاد اسلات زمانی آزاد عمومی پیش‌فرض
+                    now = get_now()
+                    slot = TimeSlot(
+                        counselor_id=new_user.id,
+                        start_time=now + timedelta(days=1, hours=10),
+                        end_time=now + timedelta(days=1, hours=11),
+                        is_booked=False
+                    )
+                    db.add(slot)
+                    
+                elif u_data["role"] == "school_representative":
+                    s = SchoolProfile(
+                        user_id=new_user.id,
+                        name="موسسه اندیشه نوین",
+                        subscription_active_until=get_now() + timedelta(days=60),
+                        is_sponsored=True,
+                        bio="برترین آکادمی تستی کنکور با کادر اساتید برجسته بومی",
+                        address="خیابان ولیعصر، پلاک ۱۲",
+                        phone_number="02122222222",
+                        founding_year=1395
+                    )
+                    db.add(s)
+                    db.flush()
+                    
+                    course = SchoolCourse(
+                        school_id=s.id,
+                        course_name="شیمی تخصصی جامع",
+                        teacher_name="استاد احمدی",
+                        teaching_style="روش‌های تستی سرعتی"
+                    )
+                    db.add(course)
+                    
+        db.commit()
+        logger.info("Check-Seeding process successfully verified and updated!")
+    except Exception as e:
+        logger.error(f"Seeding process crashed: {e}", exc_info=True)
+        db.rollback()
+
 
 @app.exception_handler(Exception)
 async def global_exception_handler(request: Request, exc: Exception):
@@ -569,7 +591,7 @@ async def global_exception_handler(request: Request, exc: Exception):
     )
 
 # ==========================================
-# بخش دهم: آدرس‌های API
+# بخش دهم: کلیه آدرس‌های APIها (مراحل ۱ تا ۴)
 # ==========================================
 
 @app.post("/api/auth/register")