133 lines
5.0 KiB
Python
133 lines
5.0 KiB
Python
# -*- coding: utf-8 -*-
|
||
# created by mmmy on 2025-09-30
|
||
|
||
from typing import Dict
|
||
|
||
from sqlalchemy import Column, Integer, String, Text, DateTime, func
|
||
|
||
from utils.rldb import RLDBBaseModel
|
||
|
||
class PeopleRLDBModel(RLDBBaseModel):
|
||
__tablename__ = 'peoples'
|
||
id = Column(String(36), primary_key=True)
|
||
name = Column(String(255), index=True)
|
||
contact = Column(String(255), index=True)
|
||
gender = Column(String(10))
|
||
age = Column(Integer)
|
||
height = Column(Integer)
|
||
marital_status = Column(String(20))
|
||
match_requirement = Column(Text)
|
||
introduction = Column(Text)
|
||
comments = Column(Text)
|
||
cover = Column(String(255), nullable=True)
|
||
created_at = Column(DateTime(timezone=True), server_default=func.now(), nullable=False)
|
||
updated_at = Column(DateTime(timezone=True), server_default=func.now(), onupdate=func.now(), nullable=False)
|
||
deleted_at = Column(DateTime(timezone=True), nullable=True, index=True)
|
||
|
||
|
||
class People:
|
||
# 数据库 ID
|
||
id: str
|
||
# 姓名
|
||
name: str
|
||
# 联系人
|
||
contact: str
|
||
# 性别
|
||
gender: str
|
||
# 年龄
|
||
age: int
|
||
# 身高(cm)
|
||
height: int
|
||
# 婚姻状况
|
||
marital_status: str
|
||
# 择偶要求
|
||
match_requirement: str
|
||
# 个人介绍
|
||
introduction: Dict[str, str]
|
||
# 总结评价
|
||
comments: Dict[str, str]
|
||
# 封面
|
||
cover: str = None
|
||
|
||
def __init__(self, **kwargs):
|
||
# 初始化所有属性，从kwargs中获取值，如果不存在则设置默认值
|
||
self.id = kwargs.get('id', '') if kwargs.get('id', '') is not None else ''
|
||
self.name = kwargs.get('name', '') if kwargs.get('name', '') is not None else ''
|
||
self.contact = kwargs.get('contact', '') if kwargs.get('contact', '') is not None else ''
|
||
self.gender = kwargs.get('gender', '') if kwargs.get('gender', '') is not None else ''
|
||
self.age = kwargs.get('age', 0) if kwargs.get('age', 0) is not None else 0
|
||
self.height = kwargs.get('height', 0) if kwargs.get('height', 0) is not None else 0
|
||
self.marital_status = kwargs.get('marital_status', '') if kwargs.get('marital_status', '') is not None else ''
|
||
self.match_requirement = kwargs.get('match_requirement', '') if kwargs.get('match_requirement', '') is not None else ''
|
||
self.introduction = kwargs.get('introduction', {}) if kwargs.get('introduction', {}) is not None else {}
|
||
self.comments = kwargs.get('comments', {}) if kwargs.get('comments', {}) is not None else {}
|
||
self.cover = kwargs.get('cover', None) if kwargs.get('cover', None) is not None else None
|
||
|
||
def __str__(self) -> str:
|
||
# 返回对象的字符串表示，包含所有属性
|
||
return (f"People(id={self.id}, name={self.name}, contact={self.contact}, gender={self.gender}, "
|
||
f"age={self.age}, height={self.height}, marital_status={self.marital_status}, "
|
||
f"match_requirement={self.match_requirement}, introduction={self.introduction}, "
|
||
f"comments={self.comments}, cover={self.cover})")
|
||
|
||
@classmethod
|
||
def from_dict(cls, data: dict):
|
||
if 'created_at' in data:
|
||
# 移除 created_at 字段，避免类型错误
|
||
del data['created_at']
|
||
if 'updated_at' in data:
|
||
# 移除 updated_at 字段，避免类型错误
|
||
del data['updated_at']
|
||
if 'deleted_at' in data:
|
||
# 移除 deleted_at 字段，避免类型错误
|
||
del data['deleted_at']
|
||
return cls(**data)
|
||
|
||
@classmethod
|
||
def from_rldb_model(cls, data: PeopleRLDBModel):
|
||
# 将关系数据库模型转换为对象
|
||
return cls(
|
||
id=data.id,
|
||
name=data.name,
|
||
contact=data.contact,
|
||
gender=data.gender,
|
||
age=data.age,
|
||
height=data.height,
|
||
marital_status=data.marital_status,
|
||
match_requirement=data.match_requirement,
|
||
introduction=data.introduction,
|
||
comments=data.comments,
|
||
cover=data.cover,
|
||
)
|
||
|
||
def to_dict(self) -> dict:
|
||
# 将对象转换为字典格式
|
||
return {
|
||
'id': self.id,
|
||
'name': self.name,
|
||
'contact': self.contact,
|
||
'gender': self.gender,
|
||
'age': self.age,
|
||
'height': self.height,
|
||
'marital_status': self.marital_status,
|
||
'match_requirement': self.match_requirement,
|
||
'introduction': self.introduction,
|
||
'comments': self.comments,
|
||
'cover': self.cover,
|
||
}
|
||
|
||
def to_rldb_model(self) -> PeopleRLDBModel:
|
||
# 将对象转换为关系数据库模型
|
||
return PeopleRLDBModel(
|
||
id=self.id,
|
||
name=self.name,
|
||
contact=self.contact,
|
||
gender=self.gender,
|
||
age=self.age,
|
||
height=self.height,
|
||
marital_status=self.marital_status,
|
||
match_requirement=self.match_requirement,
|
||
introduction=self.introduction,
|
||
comments=self.comments,
|
||
cover=self.cover,
|
||
) |