class Child_cmt(BaseModel, db.Model):
__tablename__ = "ccmt"
id = db.Column(db.Integer, primary_key=True)
review_title = db.Column(db.String(50))
review_content = db.Column(db.Text())
fcmt_id = db.Column(db.Integer, db.ForeignKey('comment.id'))
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
class Follow(BaseModel, db.Model):
__tablename__ = 'follows'
follower_id = db.Column(db.Integer,
db.ForeignKey('user.id'),
primary_key=True)
followed_id = db.Column(db.Integer,
db.ForeignKey('user.id'),
primary_key=True)
timestamp = db.Column(db.DateTime, default=datetime.utcnow)
class Love(BaseModel, db.Model):
_tablename__ = "love"
id = db.Column(db.Integer, primary_key=True)
focus = db.Column(db.Integer)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
store_id = db.Column(db.Integer, db.ForeignKey('ta.id'))
def __init__(self, **kwargs):
self.focus = kwargs.get("focus", None)
self.author = kwargs.get("author", None)
self.store = kwargs.get("store", None)
class Comment_like(BaseModel, db.Model):
__tablename__ = "comment_like"
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
comment_id = db.Column(db.Integer, db.ForeignKey('comment.id'))
def __init__(self, **kwargs):
self.user_id = kwargs.get("user_id", None)
self.comment_id = kwargs.get("comment_id", None)
def find_all(self, id):
return self.query.filter_by(user_id=id).all()
class Reservation(BaseModel, db.Model):
__tablename__ = "reservation"
id = db.Column(db.Integer, primary_key=True)
title_name = db.Column(db.String(32))
people = db.Column(db.Integer())
booking_date = db.Column(db.Date)
booking_time = db.Column(db.String(10))
order_id = db.Column(db.String(40))
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
restaurant_id = db.Column(db.Integer, db.ForeignKey("ta.id"))
def cancel_order(self, id):
row = self.query.filter_by(order_id=id).first()
dao.delete(row)
def find_by_orderId(self, id):
return self.query.filter_by(order_id=id).first()
def update(self, id, **kwargs):
self.query.filter_by(order_id=id).update(kwargs)
return dao.update()
class User(UserMixin, db.Model):
__tablename__ = 'user'
__table_args__ = {"useexisting": True}
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(20),
unique=True,
index=True,
nullable=False)
email = db.Column(db.String(40), unique=True, nullable=False)
cellphone = db.Column(db.String(10), nullable=False)
password_hash = db.Column(db.String(128))
click_store = db.relationship('Click', backref='user', lazy="dynamic")
reservation = db.relationship('Reservation',
backref='user',
lazy='dynamic')
comment_son = db.relationship("Child_cmt",
backref="author",
lazy="dynamic")
comment_like = db.relationship('Comment_like',
backref='user',
lazy="dynamic")
restaurant_love = db.relationship('Love', backref='author', lazy="dynamic")
restaurant_comment = db.relationship('Comment',
backref='author',
lazy='dynamic')
confirmed = db.Column(db.Boolean, default=False)
city = db.Column(db.String(10))
website = db.Column(db.String(128))
about_me = db.Column(db.String(100))
followed = db.relationship('Follow',
foreign_keys=[Follow.follower_id],
backref=db.backref('follower', lazy='joined'),
lazy='dynamic',
cascade='all, delete-orphan')
followers = db.relationship('Follow',
foreign_keys=[Follow.followed_id],
backref=db.backref('followed', lazy='joined'),
lazy='dynamic',
cascade='all, delete-orphan')
@cache.memoize(CACHE_TIMEOUT)
def find_by_id(self, id):
return db.session.query(User).filter_by(id=id).first()
@cache.memoize(CACHE_TIMEOUT)
def find_by_email(self, email):
return self.query.filter_by(email=email).first()
@cache.memoize(CACHE_TIMEOUT)
def find_by_username(self, username):
return self.query.filter_by(username=username).first()
@cache.memoize(CACHE_TIMEOUT)
def find_by_cellphone(self, cellphone):
return self.query.filter_by(cellphone=cellphone).first()
@cache.memoize(CACHE_TIMEOUT)
def find_by_cellphone_and_email(self, cellphone, email):
return self.query.filter(
and_(User.cellphone == cellphone, User.email == email)).first()
def update(self, id, **kwargs):
self.query.filter_by(id=id).update(kwargs)
return dao.update()
@property
def password(self):
raise AttributeError('password is not a readable attribute')
@password.setter
def password(self, password):
self.password_hash = generate_password_hash(password)
def verify_password(self, password):
return check_password_hash(self.password_hash, password)
def generate_confirmation_token(self, expiration=3600):
s = Serializer(current_app.config["SECRET_KEY"], expiration)
return s.dumps({'confirm': self.id})
def ping(self):
self.last_seen = datetime.utcnow()
db.session.add(self)
def follow(self, user):
if not self.is_following(user):
f = Follow(follower=self, followed=user)
dao.save(f)
def unfollow(self, user):
f = self.followed.filter_by(followed_id=user.id).first()
if f:
dao.delete(f)
def is_following(self, user):
return self.followed.filter_by(followed_id=user.id).first() is not None
def is_followed_by(self, user):
return self.followers.filter_by(
followed_id=user.id).first() is not None
def __repr__(self):
return '<User %r>' % self.username
class Click(BaseModel, db.Model):
__tablename__ = "click"
id = db.Column(db.Integer, primary_key=True)
store_id = db.Column(db.Integer, db.ForeignKey('ta.id'))
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
class Comment(BaseModel, db.Model):
__tablename__ = "comment"
id = db.Column(db.Integer, primary_key=True)
rating = db.Column(db.Integer)
review_title = db.Column(db.String(50))
review_content = db.Column(db.Text())
friend = db.Column(db.Integer)
booking_date = db.Column(db.Date)
takeout = db.Column(db.Integer)
vegetable = db.Column(db.Integer)
service = db.Column(db.Integer)
disabled = db.Column(db.Integer)
star1 = db.Column(db.Integer)
star2 = db.Column(db.Integer)
star3 = db.Column(db.Integer)
recommend_dish = db.Column(db.String(50))
author_id = db.Column(db.Integer, db.ForeignKey('user.id'))
store_id = db.Column(db.Integer, db.ForeignKey('ta.id'))
child_cmt = db.relationship("Child_cmt", backref="fcmt", lazy="dynamic")
user_like = db.relationship("Comment_like",
backref="comment",
lazy="dynamic")
def find_by_id(self, id):
return self.query.get_or_404(id)
def find_all(self, id):
return self.query.filter_by(store_id=id).all()
def get_comments(self, id):
return self.query.filter_by(store_id=id).limit(3)
def to_dict(self):
return {
"rating": self.rating,
"author": self.author.username,
"title": self.review_title,
"content": self.review_content
}
class TripAdvisor(db.Model):
__tablename__ = "ta"
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(64))
res_type = db.Column(db.String(32))
rating_count = db.Column(db.Integer())
info_url = db.Column(db.Text())
cellphone = db.Column(db.String(20))
address = db.Column(db.String(128))
city = db.Column(db.String(16))
area = db.Column(db.String(16))
street = db.Column(db.String(64))
rating = db.Column(db.Float())
comment = db.Column(db.Text())
read_count = db.Column(db.Integer, server_default='0')
open_time = db.Column(db.String(64))
reservation = db.relationship('Reservation',
backref='store',
lazy='dynamic')
user_comment = db.relationship("Comment",
backref="restaurant",
lazy="dynamic")
following = db.relationship("Love", backref="store", lazy="dynamic")
clicked_user = db.relationship("Click", backref="store", lazy="dynamic")
def find_by_id(self, id):
return self.query.get_or_404(id)
def find_by_name(self, title):
return self.query.filter_by(title=title).first()
def to_dict(self):
return {
"id": self.id,
"title": self.title,
"address": self.address,
"street": self.street,
"rating": self.rating,
"res_type": self.res_type,
"cellphone": self.cellphone,
"rating_count": self.rating_count,
"comment": self.comment.split(","),
"info_url": self.info_url.split(",")
}
def to_comment_dict(self):
return {
"id": self.id,
"title": self.title,
"address": self.address,
"image": self.info_url.split(",")[2]
}
class BaseModel(object):
""""模型基類，為每個模型補充創建時間與更新時間"""
create_time = db.Column(db.DateTime, default=datetime.now)
update_time = db.Column(db.DateTime,
default=datetime.now,
onupdate=datetime.now)