class Area(BaseModel, db.Model):
"""城区"""
__tablename__ = "ihome_area"
id = db.Column(db.Integer, primary_key=True) # 区域编号
name = db.Column(db.String(32), nullable=False) # 区域名字
houses = db.relationship("House", backref="area") # 区域的房屋
def to_dict(self):
return {'id': self.id, 'name': self.name}
class HouseImage(BaseModel, db.Model):
"""房屋图片"""
__tablename__ = "ihome_house_image"
id = db.Column(db.Integer, primary_key=True)
# 房屋编号
house_id = db.Column(db.Integer,
db.ForeignKey("ihome_house.id"),
nullable=False)
url = db.Column(db.String(256), nullable=False) # 图片的路径
class BaseModel(object):
# 定义基础模型
create_time = db.Column(db.DATETIME, default=datetime.now())
update_time = db.Column(db.DATETIME,
default=datetime.now(),
onupdate=datetime.now())
def add_update(self):
db.session.add(self)
db.session.commit()
def delete(self):
db.session.delete(self)
db.session.commit()
class Facility(BaseModel, db.Model):
"""设施信息, 房间规格等信息"""
__tablename__ = "ihome_facility"
id = db.Column(db.Integer, primary_key=True) # 设施编号
name = db.Column(db.String(32), nullable=False) # 设施名字
css = db.Column(db.String(30), nullable=True) # 设施展示的图标
def to_dict(self):
return {'id': self.id, 'name': self.name, 'css': self.css}
def to_house_dict(self):
return {'id': self.id}
class User(db.Model):
id = db.Column(db.Integer, primary_key=True)
chat_id = db.Column(db.Integer, unique=True)
username = db.Column(db.String(128), unique=True)
name = db.Column(db.String(128), unique=True)
connected_date = db.Column(db.DateTime)
last_update = db.Column(db.DateTime, onupdate=date.today())
current = db.Column(db.Integer, default=1)
albums = db.relationship('Album',
secondary=albums,
backref=db.backref('user', lazy='joined'))
def __repr__(self):
return self.name
def favourite_list(self):
string = 'Избранное:\n'
for item, album in enumerate(self.albums):
string += '{}. {}\n'.format(item + 1, album.name)
return 'Список \'Избранное\' пуст!' if not self.albums else string
def next(self):
if Album.query.count() > self.current:
self.current += 1
db.session.commit()
def prev(self):
if self.current > 1:
self.current -= 1
db.session.commit()
class Album(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(64), unique=True)
genre = db.Column(db.String(32))
country = db.Column(db.String(32))
date = db.Column(db.Date)
image = db.Column(db.String(128))
download_link = db.Column(db.String(128))
song_list = db.Column(db.String(256))
info = db.Column(db.String(64))
def __init__(self, args):
self.name = args['name'] or 'Empty'
self.genre = args['genre'] or 'Empty'
self.country = args['country'] or 'Empty'
self.download_link = args['download'] or 'Empty'
self.image = args['img'] or 'Empty'
self.song_list = args.song_str() or 'Empty'
self.date = args['date']
self.info = args['info']
def __str__(self):
date_format = self.date.strftime('%Y %B %d') or 'Empty'
return '*{}*\n\nGenre: {}\n' \
'Country: {}\nRelease date: {}\n\n' \
'Song list:\n{}'.format(self.name, self.genre, self.country,
date_format, self.song_list)
def __repr__(self):
return self.name
def __eq__(self, other):
return self.name == other
def load_img(self):
try:
img_dir = 'media/logo.png'
file = open(img_dir, 'wb')
file.write(request.urlopen(self.image).read())
file.close()
except Exception as e:
img_dir = 'media/default.png'
return img_dir
class User(BaseModel, db.Model):
__tablename__ = 'ihome_user'
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
phone = db.Column(db.String(11), unique=True)
pwd_hash = db.Column(db.String(200))
name = db.Column(db.String(30), unique=True)
avatar = db.Column(db.String(100)) # T头像
id_name = db.Column(db.String(30)) # 实名认证的姓名
id_card = db.Column(db.String(18), unique=True) # 实名认证的身份证号码
houses = db.relationship('House', backref='user')
orders = db.relationship('Order', backref='user')
def save(self):
db.session.add(self)
db.session.commit()
@property
def password(self):
return ''
# 写
@password.setter
def password(self, pwd):
self.pwd_hash = generate_password_hash(pwd)
# 对比
def check_pwd(self, pwd):
return check_password_hash(self.pwd_hash, pwd)
def to_auth_dict(self):
return {'id_name': self.id_name, 'id_card': self.id_card}
def to_basic_dict(self):
return {
'id': self.id,
'avatar': self.avatar if self.avatar else '',
'name': self.name,
'phone': self.phone
}
from datetime import date
from urllib import request
from utils.app import db
albums = db.Table('user_albums',
db.Column('user_id', db.Integer, db.ForeignKey('user.id')),
db.Column('album_id', db.Integer, db.ForeignKey('album.id')))
class User(db.Model):
id = db.Column(db.Integer, primary_key=True)
chat_id = db.Column(db.Integer, unique=True)
username = db.Column(db.String(128), unique=True)
name = db.Column(db.String(128), unique=True)
connected_date = db.Column(db.DateTime)
last_update = db.Column(db.DateTime, onupdate=date.today())
current = db.Column(db.Integer, default=1)
albums = db.relationship('Album',
secondary=albums,
backref=db.backref('user', lazy='joined'))
def __repr__(self):
return self.name
def favourite_list(self):
string = 'Избранное:\n'
for item, album in enumerate(self.albums):
string += '{}. {}\n'.format(item + 1, album.name)
return 'Список \'Избранное\' пуст!' if not self.albums else string
class House(BaseModel, db.Model):
"""房屋信息"""
__tablename__ = "ihome_house"
id = db.Column(db.Integer, primary_key=True) # 房屋编号
# 房屋主人的用户编号
user_id = db.Column(db.Integer,
db.ForeignKey("ihome_user.id"),
nullable=False)
# 归属地的区域编号
area_id = db.Column(db.Integer,
db.ForeignKey("ihome_area.id"),
nullable=False)
title = db.Column(db.String(64), nullable=False) # 标题
price = db.Column(db.Integer, default=0) # 单价，单位：分
address = db.Column(db.String(512), default="") # 地址
room_count = db.Column(db.Integer, default=1) # 房间数目
acreage = db.Column(db.Integer, default=0) # 房屋面积
unit = db.Column(db.String(32), default="") # 房屋单元， 如几室几厅
capacity = db.Column(db.Integer, default=1) # 房屋容纳的人数
beds = db.Column(db.String(64), default="") # 房屋床铺的配置
deposit = db.Column(db.Integer, default=0) # 房屋押金
min_days = db.Column(db.Integer, default=1) # 最少入住天数
max_days = db.Column(db.Integer, default=0) # 最多入住天数，0表示不限制
order_count = db.Column(db.Integer, default=0) # 预订完成的该房屋的订单数
index_image_url = db.Column(db.String(256), default="") # 房屋主图片的路径
is_delected = db.Column(db.Boolean, default=False)
# 房屋的设施
facilities = db.relationship("Facility", secondary=ihome_house_facility)
images = db.relationship("HouseImage") # 房屋的图片
orders = db.relationship('Order', backref='house')
def to_dict(self):
return {
'id': self.id,
'title': self.title,
'image': self.index_image_url if self.index_image_url else '',
'area': self.area.name,
'price': self.price,
'create_time': self.create_time.strftime('%Y-%m-%d %H:%M:%S'),
# 'avatar':current_app.config['QINIU_URL']+self.user.avatar if self.user.avatar else '',
'room': self.room_count,
'order_count': self.order_count,
'address': self.address
}
def to_full_dict(self):
return {
'id': self.id,
'user_avatar': self.user.avatar if self.user.avatar else '',
'user_name': self.user.name,
'title': self.title,
'price': self.price,
'address': self.area.name + self.address,
'room_count': self.room_count,
'acreage': self.acreage,
'unit': self.unit,
'capacity': self.capacity,
'beds': self.beds,
'deposit': self.deposit,
'min_days': self.min_days,
'max_days': self.max_days,
'order_count': self.order_count,
'images': [image.url for image in self.images],
'facilities': [facility.to_dict() for facility in self.facilities],
}
class Order(BaseModel, db.Model):
__tablename__ = "ihome_order"
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer,
db.ForeignKey("ihome_user.id"),
nullable=False)
house_id = db.Column(db.Integer,
db.ForeignKey("ihome_house.id"),
nullable=False)
begin_date = db.Column(db.DateTime, nullable=False) # 入住时间
end_date = db.Column(db.DateTime, nullable=False) # 离店时间
days = db.Column(db.Integer, nullable=False) # 住多少天
house_price = db.Column(db.Integer, nullable=False) # 房间价格
amount = db.Column(db.Integer, nullable=False) # 总价格
trade_id = db.Column(db.Text, nullable=True) # 订单编号
is_delected = db.Column(db.Boolean, default=False) # 是否完成订单
status = db.Column(
db.Enum(
"WAIT_ACCEPT", # 待接单,
"WAIT_PAYMENT", # 待支付
"PAID", # 已支付
"WAIT_COMMENT", # 待评价
"COMPLETE", # 已完成
"CANCELED", # 已取消
"REJECTED" # 已拒单
),
default="WAIT_ACCEPT",
index=True)
comment = db.Column(db.Text) # 评论
def to_dict(self):
return {
'user_id': self.user_id,
'house_id': self.house_id,
'order_id': self.id,
'house_title': self.house.title,
'image':
self.house.index_image_url if self.house.index_image_url else '',
'create_date': self.create_time.strftime('%Y-%m-%d'),
'begin_date': self.begin_date.strftime('%Y-%m-%d'),
'end_date': self.end_date.strftime('%Y-%m-%d'),
'amount': self.amount,
'days': self.days,
'status': self.status,
'comment': self.comment,
'trade_id': self.trade_id,
}
return {'id_name': self.id_name, 'id_card': self.id_card}
def to_basic_dict(self):
return {
'id': self.id,
'avatar': self.avatar if self.avatar else '',
'name': self.name,
'phone': self.phone
}
ihome_house_facility = db.Table(
"ihome_house_facility",
db.Column("house_id",
db.Integer,
db.ForeignKey("ihome_house.id"),
primary_key=True), # 房屋编号
db.Column("facility_id",
db.Integer,
db.ForeignKey("ihome_facility.id"),
primary_key=True) # 设施编号
)
class House(BaseModel, db.Model):
"""房屋信息"""
__tablename__ = "ihome_house"
id = db.Column(db.Integer, primary_key=True) # 房屋编号
# 房屋主人的用户编号