from .db import engine
from .models import Announcement, UserDatabase, Trashbox, Base


Base.metadata.create_all(bind=engine)

db = SessionLocal()

# Пробный чувак
tom = UserDatabase(name="Tom", phone="89999999", email="pupka", password="1234", surname="Smith")
# db.add(tom)     # добавляем в бд
# db.commit()     # сохраняем изменения
# db.refresh(tom)  # обновляем состояние объекта

# Пробное объявление 1
a1 = Announcement(user_id=1, category="cat", best_by="201223", address="abd", longtitude=23, latitude=22,
description="abv", src="111", metro="Lesnaya", booked_by=2)
# Пробное объявление 2
a2 = Announcement(user_id=1, category="dog", best_by="221223", address="abd", longtitude=50, latitude=12, 
description="vvv", src="110", metro="Petrogradskaya", booked_by=2)

a3 = Announcement(user_id=1, category="a", best_by="221223", address="abd", longtitude=20, latitude=25,
description="vvv", src="101", metro="metro", booked_by=2)

trash1 = Trashbox(name="Tom", address="abd", longtitude=23, latitude=22, category="indisposable")

# db.add(a1)     # добавляем в бд
# db.add(a2)     # добавляем в бд
# db.add(a3)     # добавляем в бд
# db.add(trash1)     # добавляем в бд
# db.commit()     # сохраняем изменения
# db.refresh(a1)  # обновляем состояние объекта
# db.refresh(a2)  # обновляем состояние объекта
# db.refresh(a3)  # обновляем состояние объекта
# db.refresh(trash1)  # обновляем состояние объекта

# # Удалить все
# db.query(User).delete()
# db.query(Announcement).delete()
# db.commit()