新增账目类型

app/api/account/crud.py

@@ -165,7 +165,7 @@ def get_account_type(db: Session, data):
 
 
 def update_account_type(db: Session, data):
-
+    """修改账目类型"""
     try:
         db.query(AccountType).filter(AccountType.id == data.id).update({AccountType.key_name: data.key_name,
                                                                         AccountType.key_value: data.key_value,
@@ -175,3 +175,22 @@ def update_account_type(db: Session, data):
     except Exception as e:
         print(e)
         return False
+
+
+def get_account_type_value(db: Session, key):
+    """新建类型 查询是否有重复数据"""
+    return db.query(AccountType).filter(AccountType.key_value == key).first()
+
+
+def create_type(db: Session, data):
+    """创建新账目类型"""
+    try:
+        db_type = AccountType(key_name=data.key_name, key_value=data.key_value, type=data.type,
+                              create_time=datetime.now())
+        db.add(db_type)
+        db.commit()
+        db.refresh(db_type)
+    except Exception as e:
+        print(e)
+        return {}
+    return db_type

app/api/account/schemas.py

@@ -59,3 +59,9 @@ class AccountTypeUpdate(BaseModel):
 class AccountTypeList(AccountTypeUpdate):
     page: int = None
     size: int = None
+
+
+class CreateType(BaseModel):
+    key_name: str
+    key_value: str
+    type: int

app/api/account/views.py

@@ -85,3 +85,15 @@ def finance_fix(data: schemas.AccountTypeList, db: Session = Depends(get_db)):
     if res:
         return HttpResultResponse()
     return HttpResultResponse(code=422, msg='修改失败')
+
+
+@router.post("/create/type")
+def create_user(data: schemas.CreateType, db: Session = Depends(get_db)):
+    """添加账目类型"""
+    res_type = crud.get_account_type_value(db, data.key_value)
+    if res_type:
+        return HttpResultResponse(code=400, msg=HttpMessage.TYPE_EXIST)
+    res = crud.create_type(db, data)
+    if not res:
+        return HttpResultResponse(code=500, msg=res)
+    return HttpResultResponse(data=res.id)

libs/result_format.py

@@ -14,6 +14,7 @@ class HttpMessage(object):
     NO_DELETE_ROLE = "不能删除已绑定用户的角色！"
 
     ACCOUNT_EXIST = "账号已存在"
+    TYPE_EXIST = "类目标识已存在"
 
     DELETE_LOSE = '删除失败'
     LOSE = '操作失败'