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 HouseImages(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=True)
url = db.Column(db.String(256), nullable=True) # 图片的路径
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=True) # 设施名字
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(BaseModel, db.Model):
__tablename__ = 'ih_user'
id = db.Column(db.INTEGER, primary_key=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)) # 头像
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')
@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_basic_dict(self):
data = {
'id': self.id,
'phone': self.phone,
'name': self.name,
'avatar': self.avatar if self.avatar else '',
}
return data
class Order(BaseModel, db.Model):
__tablename__ = 'ihome_order'
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer,
db.ForeignKey('ih_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) # 总价
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 {
'order_id': self.id,
'house_title': self.house.title,
'image':
self.house.index_img_url if self.house.index_img_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,
}
class House(BaseModel, db.Model):
"""房屋信息"""
__tablename__ = 'ihome_house'
id = db.Column(db.Integer, primary_key=True)
# 房屋主人用户编号
user_id = db.Column(db.Integer, db.ForeignKey('ih_user.id'), nullable=True)
# 归属地编号
area_id = db.Column(db.Integer,
db.ForeignKey('ihome_area.id'),
nullable=True)
title = db.Column(db.String(64), nullable=True)
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_img_url = db.Column(db.String(256), default='') # 房屋主图片路径
# 房屋设施
facilities = db.relationship('Facility', secondary=ihome_house_facility)
images = db.relationship('HouseImages') # 房屋图片
orders = db.relationship('Order', backref='house')
def to_dict(self):
return {
'id': self.id,
'title': self.title,
'image': self.index_img_url if self.index_img_url else '',
'area': self.area.name,
'price': self.price,
'create_time': self.create_time.strftime('%Y-%m-%d %H:%M:%S'),
'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.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],
'facitities': [facility.to_dict() for facility in self.facilities],
}
return check_password_hash(self.pwd_hash, pwd)
def to_basic_dict(self):
data = {
'id': self.id,
'phone': self.phone,
'name': self.name,
'avatar': self.avatar if self.avatar else '',
}
return data
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)
# 房屋主人用户编号
user_id = db.Column(db.Integer, db.ForeignKey('ih_user.id'), nullable=True)