-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmodels.py
50 lines (43 loc) · 1.89 KB
/
models.py
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
from sqlalchemy import *
from sqlalchemy import select
from sqlalchemy.orm import (scoped_session, sessionmaker, relationship,
backref)
from sqlalchemy.ext.declarative import declarative_base
engine = create_engine('postgresql://postgres:test123@db:5432/lexus', convert_unicode=True)
db_session = scoped_session(sessionmaker(autocommit=False,
autoflush=False,
bind=engine))
Base = declarative_base()
# We will need this for querying
Base.query = db_session.query_property()
class Feedback(Base):
__tablename__ = 'feedback'
id = Column(Integer, primary_key=True)
customer_name = Column(String(200))
rating = Column(Integer)
comments = Column(Text())
order_id = Column(Integer, ForeignKey('order.id'))
dealer_id = Column(Integer, ForeignKey('dealer.id'))
customer_id = Column(Integer, ForeignKey('customer.id'))
class Order(Base):
__tablename__ = 'order'
id = Column(Integer, primary_key=True)
price = Column(Integer)
customer_id = Column(Integer, ForeignKey('customer.id'))
dealer_id = Column(Integer, ForeignKey('dealer.id'))
class Dealer(Base):
__tablename__ = 'dealer'
id = Column(Integer, primary_key=True)
name = Column(String(200))
country = Column(String(200))
averageRating = Column(Integer)
numofRatings = Column(Integer)
feedbacks = relationship('Feedback', backref = 'dealer', cascade='all, delete-orphan', lazy=True)
orders = relationship('Order', backref = 'dealer', cascade='all, delete-orphan', lazy=True)
class Customer(Base):
__tablename__ = 'customer'
id = Column(Integer, primary_key=True)
name = Column(String(200))
feedbacks = relationship('Feedback', backref = 'customer', lazy = True)
orders = relationship('Order', backref = 'customer', lazy = True)
Base.metadata.create_all(bind=engine)