graphene_testing/database.py at master · microwavenby/graphene_testing · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
import sqlalchemy as sa
from sqlalchemy import orm
import models as m
from models import Base

Engine = sa.create_engine('sqlite:///test.db', convert_unicode=True)
Session = orm.scoped_session(orm.sessionmaker(
    autocommit=False, autoflush=False, bind=Engine))
Base.query = Session.query_property()


def create(engine=Engine):
    Base.metadata.drop_all(bind=engine)
    Base.metadata.create_all(bind=engine)


def hydrate(engine=Engine, session=Session):

    accounts = {
        'bk': m.Account(name='Burger Kwik'),
        'cl': m.Account(name='Chikin Likin'),
        'pf': m.Account(name='Pizza Face')}

    locations = [
        m.Location(
            name='BK1',
            address='314 Circle Ct',
            account=accounts['bk']),
        m.Location(
            name='BK2',
            address='1350 American Way',
            account=accounts['bk']),
        m.Location(
            name='CL1',
            address='2718 Euler Blvd',
            account=accounts['cl'])]

    users = [
        m.User(
            name='Peter',
            account=accounts['bk']),
        m.User(
            name='Paul',
            account=accounts['bk']),
        m.User(
            name='Mary',
            account=accounts['bk']),
        m.User(
            name='Karen',
            account=accounts['pf']),
        m.User(
            name='Richard',
            account=accounts['pf'])]

    session.add_all(accounts.values())
    session.add_all(locations)
    session.add_all(users)
    session.commit()