Exemple #1
0
def unique_email(form, field):
    session = Session()
    email = session.query(User).filter(User.email == field.data).first()
    session.close()
    if email:
        raise ValidationError('%s[%s] is already registered' %
                              (field.name, field.data))
Exemple #2
0
def stock_exist_valid(code):
    if(code in ('sz','sh')):
        return True
    sql = "select code,name from [stock].[comp_basic] where code='{code}'".format(code=code)
    s = Session()
    r = s.execute(sql).fetchall()
    return r.__len__() > 0
Exemple #3
0
def stock_exist_valid(code):
    if (code in ('sz', 'sh')):
        return True
    sql = "select code,name from [stock].[comp_basic] where code='{code}'".format(
        code=code)
    s = Session()
    r = s.execute(sql).fetchall()
    return r.__len__() > 0
Exemple #4
0
def load_user(user_id):
    from app.models import User
    from app.db_info import Session
    session = Session()
    user = session.query(User).options(subqueryload(
        User.address)).filter(User.id == user_id).first()
    session.close()
    return user
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_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()
Exemple #8
0
class ApptForm(Form):
    id = IntegerField("Appointment ID")
    appt_date = DateField("Appointment Date")
    appt_timeslot = NoValidationSelectField("Time Slot",
                                            coerce=int,
                                            choices=[(1, "Select date first")])
    appt_service = SelectMultipleField("Services",
                                       coerce=int,
                                       choices=Session().query(
                                           Service.id, Service.type,
                                           Service.desc).all())
    pet = NoValidationSelectField("Pet to Serve",
                                  coerce=int,
                                  choices=[],
                                  validators=[DataRequired()])

    def pre_validate(self, form):
        pass
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()
def run_sql_via_sqlalchemy(sql):
    session = Session()
    r = session.execute(sql)
    result = r.fetchall()
    session.close()
    return result
Exemple #11
0
def run_sql_via_sqlalchemy(sql):
    session = Session()
    r = session.execute(sql)
    result = r.fetchall()
    session.close()
    return result