42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
from datetime import datetime, timedelta
|
|
|
|
from app.infrastructure.database import Note, Reservation, SessionLocal, Task, init_db
|
|
from app.memory.memory_manager import MemoryManager
|
|
|
|
|
|
def seed() -> None:
|
|
init_db()
|
|
session = SessionLocal()
|
|
try:
|
|
if session.query(Task).count() == 0:
|
|
session.add_all(
|
|
[
|
|
Task(title="Review venue shortlist", assignee="Maria", status="pending"),
|
|
Task(title="Confirm catering tasting", assignee="Nikos", status="in_progress"),
|
|
]
|
|
)
|
|
if session.query(Note).count() == 0:
|
|
session.add_all(
|
|
[
|
|
Note(content="Customer prefers white flowers.", category="preference"),
|
|
Note(content="Budget conversation should stay under 20,000 EUR.", category="budget"),
|
|
]
|
|
)
|
|
if session.query(Reservation).count() == 0:
|
|
session.add(
|
|
Reservation(
|
|
vendor="Aegean Garden Venue",
|
|
reservation_date=datetime.utcnow() + timedelta(days=45),
|
|
status="pending",
|
|
)
|
|
)
|
|
session.commit()
|
|
memory = MemoryManager()
|
|
memory.seed_demo()
|
|
finally:
|
|
session.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
seed()
|