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

from PIL import Image, ImageDraw, ImageFont
from io import BytesIO
from django.core.files.base import ContentFile
from apps.core.models import SiteSettings, TeamMember, Service, Award, ExpertiseArea, Slider, Stat, AboutSection
from apps.projects.models import Project
from apps.news.models import Article

def make_image(text, bg_color, size=(400,400), text_color=(255,255,255)):
    img = Image.new('RGB', size, bg_color)
    draw = ImageDraw.Draw(img)
    # Draw text in center
    bbox = draw.textbbox((0,0), text)
    tw, th = bbox[2]-bbox[0], bbox[3]-bbox[1]
    x = (size[0]-tw)//2
    y = (size[1]-th)//2
    draw.text((x,y), text, fill=text_color)
    buf = BytesIO()
    img.save(buf, 'JPEG', quality=85)
    buf.seek(0)
    return ContentFile(buf.read())

def make_avatar(initials, color):
    img = Image.new('RGB', (300,300), color)
    draw = ImageDraw.Draw(img)
    # White circle
    draw.ellipse([20,20,280,280], fill=color)
    # Initial letters
    bbox = draw.textbbox((0,0), initials)
    tw, th = bbox[2]-bbox[0], bbox[3]-bbox[1]
    draw.text(((300-tw)//2,(300-th)//2), initials, fill=(255,255,255))
    buf = BytesIO()
    img.save(buf, 'JPEG', quality=85)
    buf.seek(0)
    return ContentFile(buf.read())

COLORS = {
    'navy': (27,58,107), 'gold': (200,150,42),
    'green': (16,185,129), 'teal': (13,148,136),
    'purple': (139,92,246), 'red': (239,68,68),
    'blue': (59,130,246), 'orange': (249,115,22),
}

print("ساخت داده‌های نمونه...")

# ── SiteSettings ─────────────────────────────────────────
site = SiteSettings.get()
site.instagram = 'https://instagram.com/artaghadir'
site.linkedin  = 'https://linkedin.com/company/artaghadir'
site.telegram  = 'https://t.me/artaghadir'
site.youtube   = 'https://youtube.com/@artaghadir'
site.phone     = '021-88765432'
site.email     = 'info@artaghadir.com'
site.address_fa= 'تهران، خیابان ملاصدرا، نرسیده به اتوبان چمران، روبروی خیابان خوارزمی، پلاک ۲۳۵'
site.about_short_fa = 'شرکت کانی کاوش آرتا غدیر بازوی فعال هلدینگ صنایع و معادن غدیر در زمینه اکتشاف، استخراج و فرآوری مواد معدنی در بیش از ۱۱ استان کشور است.'
site.map_lat   = 35.756610355525446
site.map_lng   = 51.39176235347857
site.save()
print("  ✓ تنظیمات سایت")

# ── Team Members ──────────────────────────────────────────
team_data = [
    ('مهندس علی رضایی',   'Ali Rezaei',   'مدیرعامل',         'CEO',              'AR', COLORS['navy']),
    ('دکتر محمد کریمی',   'Dr. M. Karimi', 'رئیس هیئت مدیره',  'Chairman',         'MK', COLORS['gold']),
    ('مهندس سارا احمدی',  'Sara Ahmadi',   'مدیر فنی',         'Technical Director','SA', COLORS['teal']),
    ('مهندس رضا موسوی',   'Reza Mousavi',  'مدیر مالی',        'CFO',              'RM', COLORS['green']),
    ('مهندس نیلوفر حسینی','N. Hosseini',   'مدیر اکتشاف',      'Exploration Dir.', 'NH', COLORS['purple']),
    ('دکتر کاوه صادقی',   'Dr. K. Sadeghi','مدیر محیط زیست',   'Environment Dir.', 'KS', COLORS['blue']),
]

TeamMember.objects.all().delete()
for i,(nfa,nen,rfa,ren,initials,color) in enumerate(team_data):
    member = TeamMember(
        name_fa=nfa, name_en=nen,
        role_fa=rfa, role_en=ren,
        bio_fa=f'{nfa}، با بیش از ۱۵ سال تجربه در صنعت معدن، در حوزه {rfa} فعالیت می‌نماید.',
        linkedin='https://linkedin.com/in/example',
        order=i, is_active=True
    )
    img_content = make_avatar(initials, color)
    member.photo.save(f'team_{initials.lower()}.jpg', img_content, save=False)
    member.save()
print("  ✓ ۶ عضو تیم")

# ── Services ─────────────────────────────────────────────
Service.objects.all().delete()
services = [
    ('🔍','اکتشاف و پیمایش','Exploration','مطالعات زمین‌شناسی، ژئوفیزیک و حفاری اکتشافی', COLORS['navy']),
    ('⛏️','استخراج معدن','Mining','عملیات استخراج روباز و زیرزمینی', COLORS['gold']),
    ('⚙️','فرآوری مواد معدنی','Processing','تولید کنسانتره و بهره‌وری از ذخایر', COLORS['teal']),
    ('🌿','محیط زیست و HSE','HSE','پایش محیطی و ایمنی معادن', COLORS['green']),
    ('📊','امکانسنجی','Feasibility','ارزیابی اقتصادی-فنی پروژه‌های معدنی', COLORS['purple']),
    ('🚛','لجستیک معدن','Logistics','حمل و نقل و زنجیره تامین مواد معدنی', COLORS['orange']),
]
for i,(icon,tfa,ten,dfa,color) in enumerate(services):
    svc = Service(icon_fa=icon,title_fa=tfa,title_en=ten,desc_fa=dfa,order=i,is_active=True)
    img = make_image(tfa[:8], color, (600,400))
    svc.image.save(f'service_{i}.jpg', img, save=False)
    svc.save()
print("  ✓ ۶ خدمت")

# ── Awards ──────────────────────────────────────────────
Award.objects.all().delete()
awards = [
    ('واحد نمونه صنعت معدن','وزارت صنعت، معدن و تجارت','1401', COLORS['gold']),
    ('گواهینامه ایمنی و بهداشت','سازمان نظام مهندسی معدن','1400', COLORS['green']),
    ('تقدیرنامه حفاظت محیط زیست','سازمان حفاظت محیط زیست','1402', COLORS['teal']),
    ('برنده جایزه استخراج پایدار','انجمن مهندسی معدن ایران','1401', COLORS['navy']),
    ('گواهی ISO 45001','موسسه بین‌المللی استاندارد','1399', COLORS['blue']),
    ('رتبه اول اکتشاف استان یزد','سازمان صنعت معدن یزد','1402', COLORS['purple']),
]
for i,(tfa,issuer,year,color) in enumerate(awards):
    award = Award(title_fa=tfa,issuer_fa=issuer,year=year,order=i,
                  desc_fa=f'دریافت {tfa} از {issuer} به پاس خدمات ارزنده در صنعت معدن کشور')
    img = make_image('🏆', color, (300,300))
    award.image.save(f'award_{i}.jpg', img, save=False)
    award.save()
print("  ✓ ۶ تقدیرنامه")

# ── ExpertiseArea ────────────────────────────────────────
ExpertiseArea.objects.all().delete()
exp_data = [
    ('⛏️','اکتشاف معدنی','Mineral Exploration','مطالعات ژئوفیزیکی و حفاری اکتشافی'),
    ('🏔️','استخراج','Mining Operations','استخراج روباز و زیرزمینی با روش‌های نوین'),
    ('⚙️','فرآوری','Mineral Processing','تولید کنسانتره و محصولات معدنی'),
    ('🌿','محیط زیست','Environmental','پایش زیست‌محیطی و ایمنی'),
    ('📊','امکانسنجی','Feasibility','ارزیابی اقتصادی پروژه‌ها'),
]
for i,(icon,tfa,ten,dfa) in enumerate(exp_data):
    ExpertiseArea.objects.get_or_create(title_fa=tfa,defaults={'icon_fa':icon,'title_en':ten,'desc_fa':dfa,'order':i,'is_active':True})
print("  ✓ حوزه‌های تخصصی")

# ── Sliders ──────────────────────────────────────────────
if not Slider.objects.exists():
    sliders = [
        ('اکتشاف، استخراج و فرآوری مواد معدنی','بازوی فعال هلدینگ صنایع و معادن غدیر',COLORS['navy']),
        ('معدن سنگ آهن ساغند','قرارداد استخراج بالای ۴ میلیون تن',COLORS['teal']),
        ('فعالیت در ۱۱ استان کشور','پیشرو در صنعت معدن ایران',COLORS['gold']),
    ]
    for i,(tfa,sfa,color) in enumerate(sliders):
        sl = Slider(title_fa=tfa,subtitle_fa=sfa,order=i,is_active=True)
        img = make_image(tfa[:10], color, (1200,600))
        sl.image.save(f'slider_{i}.jpg', img, save=False)
        sl.save()
    print("  ✓ ۳ اسلایدر")

print("""
✅ داده‌های نمونه ساخته شدند!
   سایت را refresh کنید تا بخش‌های جدید را ببینید.
""")