修改代码生成接口,optional的设值为None

This commit is contained in:
wcq 2023-12-06 16:11:36 +08:00
parent 31a57ce329
commit b1fa11fc9f
6 changed files with 36 additions and 36 deletions

View File

@ -13,7 +13,7 @@ class ReceiveFileAdd(BaseModel):
name: str
receive_order_id: str
receive_subject_file_id: int
file_path: Optional[str]
file_path: Optional[str] = None
class ReceiveFileAddOptional(ReceiveFileAdd, metaclass=AllOptional):

View File

@ -10,21 +10,21 @@ class ReceiveOrderId(BaseModel):
class ReceiveOrderAdd(BaseModel):
subject: Optional[Any]
subject: Optional[Any] = None
subject_id: int
sender_company: str
sender_phone: Optional[str]
sender_email: Optional[str]
sender_name: Optional[str]
sender_wx: Optional[str]
contact_user_id: Optional[str]
url_code: Optional[str]
code: Optional[str]
sender_phone: Optional[str] = None
sender_email: Optional[str] = None
sender_name: Optional[str] = None
sender_wx: Optional[str] = None
contact_user_id: Optional[str] = None
url_code: Optional[str] = None
code: Optional[str] = None
state: Any
files: Optional[List[Any]]
create_time: Optional[datetime]
finish_time: Optional[datetime]
notice_logs: Optional[List[Any]]
files: Optional[List[Any]] = None
create_time: Optional[datetime] = None
finish_time: Optional[datetime] = None
notice_logs: Optional[List[Any]] = None
class ReceiveOrderAddOptional(ReceiveOrderAdd, metaclass=AllOptional):

View File

@ -13,7 +13,7 @@ class ReceiveOrderNoticeLogAdd(BaseModel):
subject_id: str
account: str
notice_type: Any
notice_time: Optional[datetime]
notice_time: Optional[datetime]= None
class ReceiveOrderNoticeLogAddOptional(ReceiveOrderNoticeLogAdd, metaclass=AllOptional):

View File

@ -11,13 +11,13 @@ class ReceiveSubjectId(BaseModel):
class ReceiveSubjectAdd(BaseModel):
name: str
start_time: Optional[datetime]
finish_time: Optional[datetime]
start_time: Optional[datetime]= None
finish_time: Optional[datetime]= None
state: Any
folder_name: Optional[str]
create_time: Optional[datetime]
folder_name: Optional[str]= None
create_time: Optional[datetime]= None
des: Any
files: Optional[List[Any]]
files: Optional[List[Any]]= None
class ReceiveSubjectAddOptional(ReceiveSubjectAdd, metaclass=AllOptional):

View File

@ -14,8 +14,8 @@ class ReceiveSubjectFileAdd(BaseModel):
subject_id: int
type: str
des: Any
template_path: Optional[str]
file_size_limit: Optional[float]
template_path: Optional[str] = None
file_size_limit: Optional[float] = None
class ReceiveSubjectFileAddOptional(ReceiveSubjectFileAdd, metaclass=AllOptional):

View File

@ -88,18 +88,18 @@ def make_code_from_model(template_path, save_path, model: Type[DeclarativeMeta],
continue
if col_base_type in ['int', 'float', 'date', 'datetime', 'bool', 'str']:
if col.server_default or col.default or col.nullable:
add_code_line = f"{key}: Optional[{col_base_type}]"
add_code_line = f"{key}: Optional[{col_base_type}] = None"
else:
add_code_line = f"{key}: {col_base_type}"
elif col_base_type in ['relation']:
if col_config.relation.relation_type == 'o2o':
add_code_line = f"{key}: {'Optional[Any]'}"
add_code_line = f"{key}: {'Optional[Any]'} = None"
elif col_config.relation.relation_type == 'o2m':
add_code_line = f"{key}: {'Optional[List[Any]]'}"
add_code_line = f"{key}: {'Optional[List[Any]]'} = None"
elif col_config.relation.relation_type == 'm2m':
add_code_line = f"{key}: {'Optional[List[Any]]'}"
add_code_line = f"{key}: {'Optional[List[Any]]'} = None"
elif col_config.relation.relation_type == 'm2o':
add_code_line = f"{key}: {'Optional[Any]'}"
add_code_line = f"{key}: {'Optional[Any]'} = None"
else:
if col_base_type in ["json", 'jsonb']:
add_code_line = f"{key}: {'Optional[Dict[Any,Any]]'} = {'{}'}"