Browse Source

241114-2代码。

baoyubo 7 months ago
parent
commit
eed027b168
2 changed files with 32 additions and 1 deletions
  1. 1 1
      models/pattern_base.py
  2. 31 0
      routers/api/pattern/__init__.py

+ 1 - 1
models/pattern_base.py

@@ -57,7 +57,7 @@ class TpPatternWSGroupList(Base):
     group_id = Column(String(255), primary_key=True)
     group_name = Column(String(255), nullable=True, comment='分组名称')
     pattern_id = Column(String(255), nullable=True, comment='图案id')
-    pattern_name = Column(String(255), nullable=True, comment='图案名称')
+    pattern_name = Column(String(255), comment='图案名称')
     del_flag = Column(String(1), default='0', comment='删除标志(0代表存在 2代表删除)')
     create_time = Column(DateTime, default=datetime.now, comment='数据创建时间')
     update_time = Column(DateTime, default=datetime.now, onupdate=datetime.now, comment='数据更新时间')

+ 31 - 0
routers/api/pattern/__init__.py

@@ -500,6 +500,37 @@ async def rollback_pattern(
         traceback.print_exc()
         raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
 
+
+@router.put("/ws/delete_all_user")
+async def rollback_pattern(
+    user_id=Depends(valid_access_token),
+    body = Depends(remove_xss_json),
+    db: Session = Depends(get_db)
+):
+    try:
+        pattern_info = pattern_id_get_tp_pattern_ws_group_def_info(body['pattern_id'],db)
+        if pattern_info is None:
+            return JSONResponse(status_code=404, content={
+                                'code': 404,
+                                'msg': '抱歉,pattern_id不存在,请联系系统管理员'
+                            })
+        if pattern_info.create_by!=user_id:
+            return JSONResponse(status_code=404, content={
+                                'code': 404,
+                                'msg': '抱歉,您无权限,请联系系统管理员'
+                            })
+        users = pattern_id_get_tp_pattern_ws_user_list(body['pattern_id'],db)
+        for user in users:
+            user.del_flag='2'
+            user.update_by = user_id
+        db.commit()
+        return {"code": 200, "msg": "关闭协同成功", "data": None}
+    except Exception as e:
+        traceback.print_exc()
+        raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
+
+
+
 @router.post("/ws/add_user")
 async def rollback_pattern(
     user_id=Depends(valid_access_token),