from database import SessionLocal, engine
from models import Sign

db = SessionLocal()

signs_to_create = [
    {"name": "Aries", "icon": "♈"},
    {"name": "Taurus", "icon": "♉"},
    {"name": "Gemini", "icon": "♊"},
    {"name": "Cancer", "icon": "♋"},
    {"name": "Leo", "icon": "♌"},
    {"name": "Virgo", "icon": "♍"},
    {"name": "Libra", "icon": "♎"},
    {"name": "Scorpio", "icon": "♏"},
    {"name": "Sagittarius", "icon": "♐"},
    {"name": "Capricorn", "icon": "♑"},
    {"name": "Aquarius", "icon": "♒"},
    {"name": "Pisces", "icon": "♓"},
]

for sign_data in signs_to_create:
    sign = db.query(Sign).filter_by(name=sign_data["name"]).first()
    if not sign:
        db.add(Sign(**sign_data))

db.commit()
db.close()

print("Initial data created successfully.")