feat(pets]): modifies db setup to add pets at app start
This commit is contained in:
@@ -4,6 +4,9 @@ from app.models.pet import Pet
|
|||||||
from app.models.pet_kind import PetKind
|
from app.models.pet_kind import PetKind
|
||||||
from app.models.adoptions import Adoptions
|
from app.models.adoptions import Adoptions
|
||||||
from app.models.adoption_status import AdoptionStatus
|
from app.models.adoption_status import AdoptionStatus
|
||||||
|
from datetime import date
|
||||||
|
from random import randrange
|
||||||
|
from werkzeug.security import generate_password_hash
|
||||||
|
|
||||||
class DBUtils():
|
class DBUtils():
|
||||||
def setup_db(app):
|
def setup_db(app):
|
||||||
@@ -11,10 +14,43 @@ class DBUtils():
|
|||||||
Exception("this function needs app context to be ran, please provide app")
|
Exception("this function needs app context to be ran, please provide app")
|
||||||
with app.app_context():
|
with app.app_context():
|
||||||
db.create_all()
|
db.create_all()
|
||||||
|
|
||||||
|
"""User"""
|
||||||
|
new_user = User(
|
||||||
|
name="Xavi",
|
||||||
|
lastname="Alonso",
|
||||||
|
email="x@d.co",
|
||||||
|
password=generate_password_hash("123456"),
|
||||||
|
birth_date= date(1990,12,30),
|
||||||
|
address="1234 Main St",
|
||||||
|
phone_number="1123873991"
|
||||||
|
)
|
||||||
|
db.session.add(new_user)
|
||||||
|
db.session.commit()
|
||||||
|
|
||||||
|
"""Pet kinds"""
|
||||||
dog = PetKind(name="Dog")
|
dog = PetKind(name="Dog")
|
||||||
cat = PetKind(name="Cat")
|
cat = PetKind(name="Cat")
|
||||||
db.session.add_all([dog,cat])
|
db.session.add_all([dog,cat])
|
||||||
db.session.commit()
|
db.session.commit()
|
||||||
|
|
||||||
|
"""Pets"""
|
||||||
|
pets = []
|
||||||
|
for i in range(40):
|
||||||
|
kind_id=randrange(1,3,1)
|
||||||
|
pets.append(Pet(
|
||||||
|
name=f"Pet {str(i)}",
|
||||||
|
kind_id=kind_id,
|
||||||
|
age=randrange(0,8),
|
||||||
|
img_src="img/placeholder-dog.jpg" if kind_id == 1 else "img/placeholder-cat.jpg",
|
||||||
|
weight=1,
|
||||||
|
location="Shelbyville",
|
||||||
|
sex= "M" if randrange(3) > 1.5 else "F",
|
||||||
|
height=1,
|
||||||
|
registrar_id=1
|
||||||
|
))
|
||||||
|
db.session.add_all(pets)
|
||||||
|
db.session.commit()
|
||||||
db.session.flush()
|
db.session.flush()
|
||||||
|
|
||||||
def drop_db(app):
|
def drop_db(app):
|
||||||
|
|||||||
Reference in New Issue
Block a user