import os, sys, django
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'artaghadir.settings')
django.setup()

from apps.projects.models import Project, ProjectTimeline
import datetime

# مراحل نمونه برای هر پروژه
timelines = {
    'saghand-iron-ore': [
        {'title_fa': 'اخذ پروانه اکتشاف', 'date': '1399-03-01', 'status': 'done',
         'description_fa': 'دریافت مجوز اکتشاف از وزارت صنعت، معدن و تجارت برای محدوده معدن ساغند'},
        {'title_fa': 'عملیات حفاری اکتشافی', 'date': '1399-09-01', 'status': 'done',
         'description_fa': 'اجرای ۴۵ گمانه اکتشافی و برداشت نمونه‌های معدنی برای تخمین ذخایر'},
        {'title_fa': 'انعقاد قرارداد استخراج', 'date': '1400-06-01', 'status': 'done',
         'description_fa': 'امضای قرارداد استخراج بالای ۴ میلیون تن با مجتمع عصر نوین ساغند'},
        {'title_fa': 'شروع عملیات استخراج', 'date': '1401-01-01', 'status': 'done',
         'description_fa': 'آغاز رسمی استخراج سنگ آهن با ظرفیت اولیه ۵۰۰ هزار تن در سال'},
        {'title_fa': 'افزایش ظرفیت استخراج', 'date': '1402-04-01', 'status': 'done',
         'description_fa': 'ارتقای تجهیزات و افزایش ظرفیت به بیش از ۱ میلیون تن در سال'},
        {'title_fa': 'راه‌اندازی کارخانه فرآوری', 'date': '1403-07-01', 'status': 'in_progress',
         'description_fa': 'احداث و راه‌اندازی کارخانه تولید کنسانتره سنگ آهن با ظرفیت ۸۰۰ هزار تن'},
        {'title_fa': 'صادرات کنسانتره', 'date': '1404-01-01', 'status': 'planned',
         'description_fa': 'برنامه‌ریزی برای صادرات کنسانتره به بازارهای منطقه‌ای'},
    ],
    'kerman-copper': [
        {'title_fa': 'مطالعات ژئوفیزیکی', 'date': '1401-02-01', 'status': 'done',
         'description_fa': 'برداشت اطلاعات مغناطیس‌سنجی و IP در محدوده ۵۶۰۰ هکتاری'},
        {'title_fa': 'حفاری اکتشافی فاز اول', 'date': '1401-09-01', 'status': 'done',
         'description_fa': 'اجرای ۱۸ گمانه اکتشافی و تأیید وجود کانسار مس'},
        {'title_fa': 'برآورد ذخایر معدنی', 'date': '1402-05-01', 'status': 'in_progress',
         'description_fa': 'مطالعات تخمین ذخایر و ارزیابی اقتصادی پروژه توسط مشاوران خارجی'},
        {'title_fa': 'طراحی معدن و تأمین مالی', 'date': '1403-03-01', 'status': 'planned',
         'description_fa': 'تهیه طرح جامع معدن و جذب سرمایه‌گذار برای فاز استخراج'},
    ],
    'zanjan-lead-zinc': [
        {'title_fa': 'اخذ پروانه بهره‌برداری', 'date': '1398-06-01', 'status': 'done',
         'description_fa': 'دریافت مجوز بهره‌برداری معدن سرب و روی زنجان'},
        {'title_fa': 'نصب تجهیزات استخراج', 'date': '1399-03-01', 'status': 'done',
         'description_fa': 'خرید و نصب ماشین‌آلات معدنی و سیستم تهویه تونل‌ها'},
        {'title_fa': 'شروع استخراج زیرزمینی', 'date': '1400-01-01', 'status': 'done',
         'description_fa': 'آغاز استخراج از تونل‌های زیرزمینی با روش Room and Pillar'},
        {'title_fa': 'احداث کارخانه فلوتاسیون', 'date': '1401-08-01', 'status': 'done',
         'description_fa': 'راه‌اندازی کارخانه فلوتاسیون با ظرفیت ۳۰۰ هزار تن در سال'},
        {'title_fa': 'گسترش محدوده استخراج', 'date': '1403-01-01', 'status': 'in_progress',
         'description_fa': 'توسعه تونل‌های جدید و افزایش ظرفیت استخراج'},
        {'title_fa': 'ارتقاء کارخانه فرآوری', 'date': '1404-06-01', 'status': 'planned',
         'description_fa': 'بروزرسانی تجهیزات و بهینه‌سازی بازیابی فلزات'},
    ],
}

total = 0
for slug, steps in timelines.items():
    project = Project.objects.filter(slug=slug).first()
    if not project:
        print(f"  پروژه پیدا نشد: {slug}")
        continue
    # Delete old timeline
    project.timeline.all().delete()
    for i, step in enumerate(steps):
        date_parts = step['date'].split('-')
        date = datetime.date(int(date_parts[0])+621, int(date_parts[1]), int(date_parts[2]))
        # Convert Jalali to approximate Gregorian
        y = int(date_parts[0])
        m = int(date_parts[1])
        d = int(date_parts[2])
        # Simple conversion: add 621 years
        gy = y + 621
        date = datetime.date(gy, m, d)
        ProjectTimeline.objects.create(
            project=project,
            title_fa=step['title_fa'],
            description_fa=step['description_fa'],
            date=date,
            status=step['status'],
            order=i
        )
        total += 1
    print(f"  ✓ {project.title_fa}: {len(steps)} مرحله")

print(f"\n✅ {total} مرحله تایم‌لاین برای {len(timelines)} پروژه اضافه شد")
print("حالا برید به صفحه جزئیات پروژه‌ها — تایم‌لاین نمایش داده می‌شه")