130 lines
3.9 KiB
Python
Executable File
130 lines
3.9 KiB
Python
Executable File
# Python library imports
|
|
from datetime import datetime
|
|
|
|
# Library imports
|
|
from flask import Flask
|
|
from flask_sqlalchemy import SQLAlchemy
|
|
|
|
# Import accounts blueprint
|
|
from accounts import accounts
|
|
|
|
# Import db
|
|
from accounts.models import db as _db
|
|
|
|
# Import db models
|
|
from accounts.models.dog import Dog
|
|
from accounts.models.dog.breed_size import BreedSize
|
|
from accounts.models.user import User
|
|
from accounts.models.visit import Visit
|
|
from accounts.models.visit.visit_type import VisitType
|
|
|
|
def create_app(config_class='development_config.py'):
|
|
accounts_app = Flask(__name__, template_folder='tests/templates')
|
|
accounts_app.register_blueprint(accounts)
|
|
accounts_app.config.from_pyfile(config_class)
|
|
|
|
return accounts_app
|
|
|
|
def add_test_data(db):
|
|
add_test_users(db)
|
|
add_test_dogs(db)
|
|
add_test_visits(db)
|
|
|
|
def add_test_users(db):
|
|
man_man = User(name='Man Man',
|
|
username='mamanman',
|
|
email='man_man@email.com',
|
|
address='123 Home Ln. City, AA 11223',
|
|
password='manword')
|
|
|
|
real_person = User(name='Real Person',
|
|
username='imarealperson',
|
|
email='real_person@email.com',
|
|
address='113 Park St. City, AA 13433',
|
|
password='realpassword')
|
|
|
|
db.session.add(man_man)
|
|
db.session.add(real_person)
|
|
|
|
db.session.commit()
|
|
|
|
def add_test_dogs(db):
|
|
man_man = db.session.execute(
|
|
db.select(User).where(User.name == 'Man Man')).scalar_one()
|
|
real_person = db.session.execute(
|
|
db.select(User).where(User.name == 'Real Person')).scalar_one()
|
|
|
|
lassey = Dog(name='Lassey',
|
|
breed_size=BreedSize.LARGE,
|
|
birth_date=datetime(2010, 10, 4),
|
|
owner_id=man_man.id)
|
|
|
|
rufus = Dog(name='Rufus',
|
|
breed_size=BreedSize.MEDIUM,
|
|
birth_date=datetime(2020, 4, 20),
|
|
owner_id=man_man.id)
|
|
|
|
air_bud = Dog(name='Air Bud',
|
|
breed_size=BreedSize.LARGE,
|
|
birth_date=datetime(2018, 7, 16),
|
|
owner_id=real_person.id)
|
|
|
|
db.session.add(lassey)
|
|
db.session.add(rufus)
|
|
db.session.add(air_bud)
|
|
|
|
db.session.commit()
|
|
|
|
def add_test_visits(db):
|
|
# User objects
|
|
man_man = db.session.execute(
|
|
db.select(User).where(User.name == 'Man Man')).scalar_one()
|
|
real_person = db.session.execute(
|
|
db.select(User).where(User.name == 'Real Person')).scalar_one()
|
|
|
|
# Dog objects
|
|
lassey = db.session.execute(
|
|
db.select(Dog).where(Dog.name == 'Lassey')).scalar_one()
|
|
rufus = db.session.execute(
|
|
db.select(Dog).where(Dog.name == 'Rufus')).scalar_one()
|
|
air_bud = db.session.execute(
|
|
db.select(Dog).where(Dog.name == 'Air Bud')).scalar_one()
|
|
|
|
# Visit with the date in the past
|
|
test_visit1 = Visit(date_time=datetime(2023,12,5,10,30),
|
|
owner_id=man_man.id,
|
|
dog_id=rufus.id)
|
|
|
|
# Visit with the date in the future
|
|
test_visit2 = Visit(date_time=datetime(2034,12,5,10,30),
|
|
owner_id=man_man.id,
|
|
dog_id=lassey.id)
|
|
|
|
# Visit with non-default visit type
|
|
test_visit3 = Visit(date_time=datetime(2034,12,5,10,30),
|
|
owner_id=real_person.id,
|
|
visit_type=VisitType.HOUSE_SITTING,
|
|
dog_id=air_bud.id)
|
|
|
|
# Visit with more than one dog
|
|
#test_visit4 = Visit(date_time=datetime(2034,12,5,10,30),
|
|
# owner_id=man_man.id,
|
|
# dog_id=[lassey.id, rufus.id])
|
|
|
|
db.session.add(test_visit1)
|
|
db.session.add(test_visit2)
|
|
db.session.add(test_visit3)
|
|
|
|
db.session.commit()
|
|
|
|
if __name__ == '__main__':
|
|
app = create_app()
|
|
_db.init_app(app)
|
|
|
|
with app.app_context():
|
|
_db.create_all()
|
|
add_test_data(_db)
|
|
|
|
app.run()
|
|
|