def mock(): with app.app_context(): recreate_db() mock_user_pet() mock_appt_slot() mock_service() mock_appt_bill()
def mock_service(): with app.app_context(): session = Session() for s in SERVICES: service = Service(s[0], s[1], s[2]) session.add(service) session.commit()
def mock_appt_slot(): with app.app_context(): session = Session() slots = ["9:30 - 10:00", "10:00 - 11:30", "12:30 - 14:00", "14:00 - 15:30", "15:30 - 17:00", "17:00 - 18:30"] for s in slots: time_slot = ApptTimeSlot(s) session.add(time_slot) session.commit()
def mock_appt_bill(): with app.app_context(): session = Session() appts = [ [2, 1, dt.date(2018, 7, 17), 2,[1,2]], [2, 1, dt.date(2018, 7, 18), 3,[2,3]], [2, 1, dt.date(2018, 7, 19), 1,[1,4]], ] for a in appts: appt = Appt(a[0],a[1],a[2],a[3]) session.add(appt) total_fee = 0 for s in a[4]: appt_service = Appt2Ser(appt, s) total_fee += SERVICES[s][2] session.add(appt_service) bill = Bill(appt,2,total_fee) session.add(bill) session.commit()
def mock_user_pet(): with app.app_context(): session = Session() address = Address("Melbourne", "Queen", 3000) md5 = hashlib.md5() md5.update("pwd".encode('utf-8')) password = md5.hexdigest() tom = User(email="*****@*****.**", password=password, first_name="Tom", last_name="Unimelb", dob=dt.date(1989, 9, 8), gender="M", phone="000000", home_number="000000", work_number="000000", address=address, active=True, admin=True, register_date=dt.datetime.now()) user1 = User(email="*****@*****.**", password=password, first_name="User1", last_name="Unimelb", dob=dt.date(1999, 12, 8), gender="M", phone="000000", home_number="000000", work_number="000000", address=address, active=True, admin=False, register_date=dt.datetime.now()) user2 = User(email="*****@*****.**", password=password, first_name="User2", last_name="Unimelb", dob=dt.date(1988, 7, 17), gender="F", phone="000000", home_number="000000", work_number="000000", address=address, active=True, admin=False, register_date=dt.datetime.now()) pet1 = Pet(user1, "Puppy1", "breed1", "M", dt.date(2017, 7, 11)) pet2 = Pet(user1, "Puppy2", "breed2", "M", dt.date(2017, 7, 12)) pet3 = Pet(user1, "Puppy3", "breed3", "F", dt.date(2017, 7, 13)) card1 = Card(user1, "1234567890123456", "ANZ") card2 = Card(user1, "2234567890123456", "ANZ") session.add(address) session.add(tom) session.add(user1) session.add(user2) session.add(pet1) session.add(pet2) session.add(pet3) session.add(card1) session.add(card2) session.commit()