30 lines
806 B
Python
30 lines
806 B
Python
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.")
|