operate.py 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. from model.DateUtils import DateUtils
  2. from model.DataBaseUtils import *
  3. from model.log import logger
  4. from model.CommonUtils import *
  5. du = DateUtils()
  6. def user_channel(arg):
  7. db = MysqlUtils()
  8. # 返回所有公众号
  9. sql = "select id,nick_name from t_mp_account where del_flag=0"
  10. return db.zx.getData_json(sql)
  11. def user_channel_group(arg):
  12. db = MysqlUtils()
  13. user_id = arg.get('user_id')
  14. if user_id:
  15. # 该用户拥有的公众号组
  16. sql = f"""SELECT id,name from channel_group where id in
  17. (select channel_group_id from user_channel_group_auth where user_id={user_id})"""
  18. return db.quchen_text.getData_json(sql)
  19. else:
  20. # 返回所有公众号组
  21. sql = "select id,name from channel_group"
  22. return db.quchen_text.getData_json(sql)
  23. def user(arg):
  24. db = MysqlUtils()
  25. # 返回所有用户
  26. sql = "SELECT user_id,nick_name from t_sys_user where del_flag=0 and status=0"
  27. return db.zx.getData_json(sql)
  28. def group_add(arg):
  29. db = MysqlUtils()
  30. name = arg.get("name")
  31. ids = arg.get("channel_ids")
  32. if name and ids:
  33. sql = "insert into channel_group (name,channel_ids) values (%s,%s)"
  34. db.quchen_text.execute(sql,(name,ids))
  35. def group_del(arg):
  36. db = MysqlUtils()
  37. id = arg.get("id")
  38. if id:
  39. sql = f"delete from channel_group where id={id}"
  40. db.quchen_text.execute(sql)
  41. def group_auth_add(arg):
  42. db = MysqlUtils()
  43. user_id = arg.get("user_id")
  44. group_id = arg.get("group_id")
  45. if user_id and group_id:
  46. sql = "insert into user_channel_group_auth (user_id,channel_group_id) values (%s,%s)"
  47. db.quchen_text.execute(sql, (user_id, group_id))
  48. def group_auth_del(arg):
  49. db = MysqlUtils()
  50. user_id = arg.get("user_id")
  51. group_id = arg.get("group_id")
  52. if user_id and group_id:
  53. sql = f"delete from user_channel_group_auth where user_id={user_id} and channel_group_id={group_id}"
  54. # print(sql)
  55. db.quchen_text.execute(sql)