You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

62 lines
2.3 KiB

9 months ago
9 months ago
9 months ago
9 months ago
  1. from fastapi import Depends, APIRouter, status, Query, Path, HTTPException
  2. from internal.models import *
  3. from internal.database import fetch_one, fetch_all, execute_query, response_success, raise_if_exists,raise_if_not_found
  4. from dependencies import get_current_active_user
  5. router = APIRouter(
  6. prefix="/comlinks",
  7. tags=['链接管理']
  8. )
  9. # 获取列表
  10. @router.get("/list")
  11. async def comlink_list():
  12. select_query = "SELECT id,linktext,linkurl,descr FROM comlinks;"
  13. comlink_list = fetch_all(select_query)
  14. return response_success(comlink_list, "comlink get list success")
  15. # 新增
  16. @router.post("/add")
  17. async def link_add(link:ComLink):
  18. insert_query="""INSERT INTO comlinks (linktext,linkurl,descr) VALUES(%s,%s,%s)"""
  19. insert_value=(link.linktext,link.linkurl,link.descr)
  20. execute_query(insert_query,insert_value)
  21. return response_success(data=link,message="link create success")
  22. # 单条数据查询
  23. @router.get("/list/search")
  24. async def link_search(linktext:str=Query(description="链接名称")):
  25. select_query="SELECT id,linktext,linkurl,descr FROM comlinks WHERE 1=1 "
  26. params=[]
  27. if linktext:
  28. select_query+="AND linktext LIKE %s"
  29. params.append(f"%{linktext}%")
  30. link_query=fetch_all(select_query,params=params,fetchall=True)
  31. return response_success(data=link_query,message="link search success")
  32. # 链接修改
  33. @router.put("/update/{id}")
  34. async def link_put(link:ComLink,id: str = Path(description="链接id")):
  35. update_query = (
  36. "UPDATE comlinks SET linktext=%s,linkurl=%s,descr=%s WHERE id=%s;"
  37. )
  38. update_data = (link.linktext, link.linkurl,
  39. link.descr,id)
  40. execute_query(update_query, update_data)
  41. return response_success("link update sucess")
  42. # 链接删除
  43. @router.delete("/delete/{id}")
  44. async def link_del(id: str = Path(description="链接id")):
  45. update_query = (
  46. "DELETE FROM comlinks WHERE id=%s;"
  47. )
  48. update_data = (id)
  49. execute_query(update_query, update_data)
  50. return response_success()
  51. # 根据id查询链接
  52. @router.get("/list/search/{id}")
  53. async def link_search_id(id:str=Path(description="链接id")):
  54. select_query="SELECT * FROM comlinks WHERE id=%s"
  55. link_query=fetch_one(select_query,(id,))
  56. return response_success(data=link_query,message="link search success")