admin_event.py 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875
  1. import time
  2. import numpy as np
  3. from tool.typing import *
  4. from sql import DBBit
  5. from sql.db import DB
  6. from sql.user import find_user_by_name
  7. from core.user import User
  8. from core.garbage import GarbageBag, GarbageType
  9. from .event import TkEventBase, TkThreading
  10. from . import admin
  11. from . import admin_program
  12. class AdminEventBase(TkEventBase):
  13. def __init__(self, station):
  14. super(AdminEventBase, self).__init__()
  15. self.station: admin.AdminStationBase = station
  16. self._db: DB = station.get_db()
  17. def get_title(self) -> str:
  18. return "AdminEvent"
  19. class LoginEvent(AdminEventBase):
  20. def __init__(self, station):
  21. super().__init__(station)
  22. self.thread: Optional[TkThreading] = None
  23. def login(self, name, passwd):
  24. return find_user_by_name(name, passwd, self._db)
  25. def start(self, name, passwd):
  26. self.thread = TkThreading(self.login, name, passwd)
  27. return self
  28. def is_end(self) -> bool:
  29. return not self.thread.is_alive()
  30. def done_after_event(self):
  31. self.station.login(self.thread.wait_event())
  32. class TestProgressEvent(AdminEventBase):
  33. @staticmethod
  34. def func(sleep_time):
  35. time.sleep(sleep_time)
  36. def __init__(self, station):
  37. super(TestProgressEvent, self).__init__(station)
  38. self.thread = TkThreading(self.func, 5)
  39. class CreateUserEvent(AdminEventBase):
  40. def func(self, name, passwd, phone, is_manager):
  41. return self.station.create_user(name, passwd, phone, is_manager)
  42. def __init__(self, station):
  43. super(CreateUserEvent, self).__init__(station)
  44. self._name = None
  45. def start(self, name, passwd, phone, is_manager):
  46. self._name = name
  47. self.thread = TkThreading(self.func, name, passwd, phone, is_manager)
  48. return self
  49. def done_after_event(self):
  50. res: Optional[User] = self.thread.wait_event()
  51. if res is None:
  52. self.station.show_warning("创建用户错误", f"无法创建用户 user: {self._name}")
  53. else:
  54. name = res.get_name()
  55. self.station.show_msg("创建用户成功", f"成功创建 {name} 新用户")
  56. class CreateGarbageEvent(AdminEventBase):
  57. def func(self, path, count):
  58. return self.station.create_garbage(path, count)
  59. def __init__(self, station):
  60. super(CreateGarbageEvent, self).__init__(station)
  61. self._name = None
  62. def start(self, path, count):
  63. self.thread = TkThreading(self.func, path, count)
  64. return self
  65. def done_after_event(self):
  66. res: list[tuple[str, Optional[GarbageBag]]] = self.thread.wait_event()
  67. self.station.show_msg("创建垃圾袋", f"成功创建{len(res)}个垃圾袋")
  68. class ExportGarbageByIDEvent(AdminEventBase):
  69. def func(self, path, gid):
  70. return self.station.export_garbage_by_gid(path, gid)
  71. def __init__(self, station):
  72. super(ExportGarbageByIDEvent, self).__init__(station)
  73. self._name = None
  74. def start(self, path, gid):
  75. self.thread = TkThreading(self.func, path, gid)
  76. return self
  77. def done_after_event(self):
  78. res: tuple[str, Optional[GarbageBag]] = self.thread.wait_event()
  79. if res[1] is None:
  80. self.station.show_warning("导出错误", f"无法导出垃圾袋二维码")
  81. else:
  82. self.station.show_msg("导出成功", f"成功导出垃圾袋二维码")
  83. class ExportGarbageAdvancedEvent(AdminEventBase):
  84. def func(self, path, where):
  85. return self.station.export_garbage(path, where)
  86. def __init__(self, station):
  87. super(ExportGarbageAdvancedEvent, self).__init__(station)
  88. self._name = None
  89. def start(self, path, where):
  90. self.thread = TkThreading(self.func, path, where)
  91. return self
  92. def done_after_event(self):
  93. res: list[tuple[str]] = self.thread.wait_event()
  94. self.station.show_msg("导出完成", f"导出{len(res)}个垃圾袋二维码")
  95. class ExportUserByIDEvent(AdminEventBase):
  96. def func(self, path, uid):
  97. return self.station.export_user_by_uid(path, uid)
  98. def __init__(self, station):
  99. super(ExportUserByIDEvent, self).__init__(station)
  100. self._name = None
  101. def start(self, path, uid):
  102. self.thread = TkThreading(self.func, path, uid)
  103. return self
  104. def done_after_event(self):
  105. res: tuple[str, Optional[GarbageBag]] = self.thread.wait_event()
  106. if res[1] is None:
  107. self.station.show_warning("导出错误", f"无法导出用户二维码")
  108. else:
  109. self.station.show_msg("导出成功", f"成功导出用户二维码")
  110. class ExportUserAdvancedEvent(AdminEventBase):
  111. def func(self, path, where):
  112. res = self.station.export_user(path, where)
  113. return res
  114. def __init__(self, station):
  115. super(ExportUserAdvancedEvent, self).__init__(station)
  116. self._name = None
  117. def start(self, path, where):
  118. self.thread = TkThreading(self.func, path, where)
  119. return self
  120. def done_after_event(self):
  121. res: list[tuple[str]] = self.thread.wait_event()
  122. self.station.show_msg("导出完成", f"导出{len(res)}个用户二维码")
  123. class CreateUserFromCSVEvent(AdminEventBase):
  124. def func(self, path):
  125. return self.station.create_user_from_csv(path)
  126. def __init__(self, station):
  127. super(CreateUserFromCSVEvent, self).__init__(station)
  128. self._name = None
  129. def start(self, path):
  130. self.thread = TkThreading(self.func, path)
  131. return self
  132. def done_after_event(self):
  133. res: list[User] = self.thread.wait_event()
  134. self.station.show_msg("创建完成", f"从CSV创建{len(res)}个新用户")
  135. class CreateAutoUserFromCSVEvent(AdminEventBase):
  136. def func(self, path):
  137. return self.station.create_auto_user_from_csv(path)
  138. def __init__(self, station):
  139. super(CreateAutoUserFromCSVEvent, self).__init__(station)
  140. self._name = None
  141. def start(self, path):
  142. self.thread = TkThreading(self.func, path)
  143. return self
  144. def done_after_event(self):
  145. res: list[User] = self.thread.wait_event()
  146. self.station.show_msg("创建完成", f"从CSV创建{len(res)}个新用户")
  147. class DelUserEvent(AdminEventBase):
  148. def func(self, uid):
  149. return self.station.del_user(uid)
  150. def __init__(self, station):
  151. super(DelUserEvent, self).__init__(station)
  152. self._name = None
  153. def start(self, uid):
  154. self.thread = TkThreading(self.func, uid)
  155. return self
  156. def done_after_event(self):
  157. res: bool = self.thread.wait_event()
  158. if res:
  159. self.station.show_msg("删除成功", f"删除用户成功")
  160. else:
  161. self.station.show_warning("删除失败", f"删除用户失败")
  162. class DelUserFromWhereScanEvent(AdminEventBase):
  163. def func(self, where):
  164. return self.station.del_user_from_where_scan(where)
  165. def __init__(self, station):
  166. super(DelUserFromWhereScanEvent, self).__init__(station)
  167. def start(self, where):
  168. self.thread = TkThreading(self.func, where)
  169. return self
  170. def done_after_event(self):
  171. res: int = self.thread.wait_event()
  172. if res != -1:
  173. self.station.show_msg("扫描结果", f"符合删除要素的用户个数: {res}")
  174. else:
  175. self.station.show_warning("扫描结果", f"获取扫描结果失败")
  176. class DelUserFromWhereEvent(AdminEventBase):
  177. def func(self, where):
  178. return self.station.del_user_from_where(where)
  179. def __init__(self, station):
  180. super(DelUserFromWhereEvent, self).__init__(station)
  181. def start(self, where):
  182. self.thread = TkThreading(self.func, where)
  183. return self
  184. def done_after_event(self):
  185. res: int = self.thread.wait_event()
  186. if res != -1:
  187. self.station.show_msg("扫描结果", f"成功删除{res}个用户")
  188. else:
  189. self.station.show_warning("扫描结果", f"删除用户失败")
  190. class DelGarbageEvent(AdminEventBase):
  191. def func(self, gid, where):
  192. if where == 0:
  193. return self.station.del_garbage(gid)
  194. elif where == 1:
  195. return self.station.del_garbage_not_use(gid)
  196. elif where == 2:
  197. return self.station.del_garbage_wait_check(gid)
  198. elif where == 3:
  199. return self.station.del_garbage_has_check(gid)
  200. return False
  201. def __init__(self, station):
  202. super(DelGarbageEvent, self).__init__(station)
  203. def start(self, gid, where):
  204. self.thread = TkThreading(self.func, gid, where)
  205. return self
  206. def done_after_event(self):
  207. res: bool = self.thread.wait_event()
  208. if res:
  209. self.station.show_msg("删除成功", f"成功删除垃圾袋")
  210. else:
  211. self.station.show_warning("删除失败", f"删除垃圾袋失败")
  212. class DelGarbageWhereEvent(AdminEventBase):
  213. def func(self, where, where_sql):
  214. if where == 1:
  215. return self.station.del_garbage_where_not_use(where_sql)
  216. elif where == 2:
  217. return self.station.del_garbage_where_wait_check(where_sql)
  218. elif where == 3:
  219. return self.station.del_garbage_where_has_check(where_sql)
  220. return -1
  221. def __init__(self, station):
  222. super(DelGarbageWhereEvent, self).__init__(station)
  223. def start(self, where, where_sql):
  224. self.thread = TkThreading(self.func, where, where_sql)
  225. return self
  226. def done_after_event(self):
  227. res: int = self.thread.wait_event()
  228. if res != -1:
  229. self.station.show_msg("删除成功", f"成功删除{res}个垃圾袋")
  230. else:
  231. self.station.show_warning("删除失败", f"垃圾袋删除失败")
  232. class DelGarbageWhereScanEvent(AdminEventBase):
  233. def func(self, where, where_sql):
  234. if where == 1:
  235. return self.station.del_garbage_where_scan_not_use(where_sql)
  236. elif where == 2:
  237. return self.station.del_garbage_where_scan_wait_check(where_sql)
  238. elif where == 3:
  239. return self.station.del_garbage_where_scan_has_check(where_sql)
  240. return -1
  241. def __init__(self, station):
  242. super(DelGarbageWhereScanEvent, self).__init__(station)
  243. def start(self, where, where_sql):
  244. self.thread = TkThreading(self.func, where, where_sql)
  245. return self
  246. def done_after_event(self):
  247. res: int = self.thread.wait_event()
  248. if res != -1:
  249. self.station.show_msg("扫描结果", f"符合删除要素的垃圾袋个数: {res}")
  250. else:
  251. self.station.show_warning("扫描结果", f"获取扫描结果失败")
  252. class DelAllGarbageScanEvent(AdminEventBase):
  253. def func(self):
  254. return self.station.del_all_garbage_scan()
  255. def __init__(self, station):
  256. super(DelAllGarbageScanEvent, self).__init__(station)
  257. self.thread = TkThreading(self.func)
  258. def done_after_event(self):
  259. res: int = self.thread.wait_event()
  260. if res != -1:
  261. self.station.show_msg("扫描结果", f"全部垃圾袋个数: {res}")
  262. else:
  263. self.station.show_warning("扫描结果", f"获取扫描结果失败")
  264. class DelAllGarbageEvent(AdminEventBase):
  265. def func(self):
  266. return self.station.del_all_garbage()
  267. def __init__(self, station):
  268. super(DelAllGarbageEvent, self).__init__(station)
  269. self.thread = TkThreading(self.func)
  270. def done_after_event(self):
  271. res: int = self.thread.wait_event()
  272. if res != -1:
  273. self.station.show_msg("删除成功", f"成功删除所用[共计{res}个]垃圾袋")
  274. else:
  275. self.station.show_warning("删除失败", f"删除所有垃圾袋失败")
  276. def set_user_search_result(i):
  277. i = list(i)
  278. if i[-1] is not None:
  279. i[-1] = "是" if i[-1] == DBBit.BIT_1 else "否"
  280. return i
  281. class SearchUserEvent(AdminEventBase):
  282. def func(self, columns, uid, name, phone):
  283. return self.station.search_user(columns, uid, name, phone)
  284. def __init__(self, station):
  285. super(SearchUserEvent, self).__init__(station)
  286. self.program: Optional[admin_program.SearchUserProgram] = None
  287. def start(self, columns, uid, name, phone, program):
  288. self.thread = TkThreading(self.func, columns, uid, name, phone)
  289. self.program = program
  290. return self
  291. def done_after_event(self):
  292. res: list[list] = self.thread.wait_event()
  293. if res is None or self.program is None:
  294. self.station.show_warning("搜索失败", f"搜索用户失败")
  295. return
  296. for i in self.program.view.get_children():
  297. self.program.view.delete(i)
  298. for i in res:
  299. i = set_user_search_result(i)
  300. self.program.view.insert('', 'end', values=i)
  301. class SearchUserAdvancedEvent(AdminEventBase):
  302. def func(self, columns, sql):
  303. return self.station.search_user_advanced(columns, sql)
  304. def __init__(self, station):
  305. super(SearchUserAdvancedEvent, self).__init__(station)
  306. self.program: Optional[admin_program.SearchUserAdvancedProgram] = None
  307. def start(self, columns, sql, program):
  308. self.thread = TkThreading(self.func, columns, sql)
  309. self.program = program
  310. return self
  311. def done_after_event(self):
  312. res: Optional[List[List[any]]] = self.thread.wait_event()
  313. if res is None or self.program is None:
  314. self.station.show_warning("高级搜索失败", f"高级搜索用户失败")
  315. return
  316. for i in self.program.view.get_children():
  317. self.program.view.delete(i)
  318. for i in res:
  319. i = set_user_search_result(i)
  320. self.program.view.insert('', 'end', values=i)
  321. def set_garbage_search_result(data):
  322. data = list(data)
  323. if data[6] is not None:
  324. tmp: bytes = data[6]
  325. data[6] = GarbageType.GarbageTypeStrList_ch[int(tmp.decode('utf-8'))]
  326. if data[7] is not None:
  327. data[7] = "投放正确" if data[7] == DBBit.BIT_1 else "投放错误"
  328. for index, i in enumerate(data):
  329. if i is None:
  330. data[index] = "无"
  331. return data
  332. class SearchGarbageEvent(AdminEventBase):
  333. def func(self, columns, key_values: dict):
  334. return self.station.search_garbage(columns, key_values)
  335. def __init__(self, station):
  336. super(SearchGarbageEvent, self).__init__(station)
  337. self.program: Optional[admin_program.SearchUserProgram] = None
  338. def start(self, columns, key_values: dict, program):
  339. self.thread = TkThreading(self.func, columns, key_values)
  340. self.program = program
  341. return self
  342. def done_after_event(self):
  343. res: list[list] = self.thread.wait_event()
  344. if res is None or self.program is None:
  345. self.station.show_warning("搜索失败", f"搜索垃圾袋失败")
  346. return
  347. for i in self.program.view.get_children():
  348. self.program.view.delete(i)
  349. for i in res:
  350. i = set_garbage_search_result(i)
  351. self.program.view.insert('', 'end', values=i)
  352. class SearchGarbageAdvancedEvent(AdminEventBase):
  353. def func(self, columns, sql):
  354. return self.station.search_garbage_advanced(columns, sql)
  355. def __init__(self, station):
  356. super(SearchGarbageAdvancedEvent, self).__init__(station)
  357. self.program: Optional[admin_program.SearchGarbageAdvancedProgram] = None
  358. def start(self, columns, sql, program):
  359. self.thread = TkThreading(self.func, columns, sql)
  360. self.program = program
  361. return self
  362. def done_after_event(self):
  363. res: list[list] = self.thread.wait_event()
  364. if res is None or self.program is None:
  365. self.station.show_warning("高级搜索失败", f"高级搜索垃圾袋失败")
  366. return
  367. for i in self.program.view.get_children():
  368. self.program.view.delete(i)
  369. for i in res:
  370. i = set_garbage_search_result(i)
  371. self.program.view.insert('', 'end', values=i)
  372. def set_advanced_search_result(data):
  373. data = list(data)
  374. if data[-2] is not None:
  375. tmp: bytes = data[-2]
  376. data[-2] = GarbageType.GarbageTypeStrList_ch[int(tmp.decode('utf-8'))]
  377. if data[-1] is not None:
  378. data[-1] = "投放正确" if data[-1] == DBBit.BIT_1 else "投放错误"
  379. data = list(data)
  380. for index, i in enumerate(data):
  381. if i is None:
  382. data[index] = "无"
  383. return data
  384. class SearchAdvancedEvent(AdminEventBase):
  385. def func(self, columns, sql):
  386. return self.station.search_advanced(columns, sql)
  387. def __init__(self, station):
  388. super(SearchAdvancedEvent, self).__init__(station)
  389. self.program: Optional[admin_program.SearchAdvancedProgram] = None
  390. def start(self, columns, sql, program):
  391. self.thread = TkThreading(self.func, columns, sql)
  392. self.program = program
  393. return self
  394. def done_after_event(self):
  395. res: list[list] = self.thread.wait_event()
  396. if res is None or self.program is None:
  397. self.station.show_warning("高级搜索失败", f"高级搜索失败")
  398. return
  399. for i in self.program.view.get_children():
  400. self.program.view.delete(i)
  401. for i in res:
  402. i = set_advanced_search_result(i)
  403. self.program.view.insert('', 'end', values=i)
  404. class UpdateUserScoreEvent(AdminEventBase):
  405. def func(self, score, where):
  406. return self.station.update_user_score(score, where)
  407. def __init__(self, station):
  408. super(UpdateUserScoreEvent, self).__init__(station)
  409. def start(self, score, where):
  410. self.thread = TkThreading(self.func, score, where)
  411. return self
  412. def done_after_event(self):
  413. res: int = self.thread.wait_event()
  414. if res == -1:
  415. self.station.show_warning("更新失败", f"更新用户-积分失败")
  416. else:
  417. self.station.show_msg("更新成功", f"成功更新{res}个用户-积分")
  418. class UpdateUserReputationEvent(AdminEventBase):
  419. def func(self, reputation, where):
  420. return self.station.update_user_reputation(reputation, where)
  421. def __init__(self, station):
  422. super(UpdateUserReputationEvent, self).__init__(station)
  423. def start(self, reputation, where):
  424. self.thread = TkThreading(self.func, reputation, where)
  425. return self
  426. def done_after_event(self):
  427. res: int = self.thread.wait_event()
  428. if res == -1:
  429. self.station.show_warning("更新失败", f"更新用户-垃圾分类信用失败")
  430. else:
  431. self.station.show_msg("更新成功", f"成功更新{res}个用户-垃圾分类信用")
  432. class UpdateGarbageTypeEvent(AdminEventBase):
  433. def func(self, type_, where):
  434. return self.station.update_garbage_type(type_, where)
  435. def __init__(self, station):
  436. super(UpdateGarbageTypeEvent, self).__init__(station)
  437. def start(self, type_, where):
  438. self.thread = TkThreading(self.func, type_, where)
  439. return self
  440. def done_after_event(self):
  441. res: int = self.thread.wait_event()
  442. if res == -1:
  443. self.station.show_warning("更新失败", f"更新垃圾袋-垃圾类型失败")
  444. else:
  445. self.station.show_msg("更新成功", f"成功更新{res}个垃圾袋-垃圾类型")
  446. class UpdateGarbageCheckEvent(AdminEventBase):
  447. def func(self, check, where):
  448. return self.station.update_garbage_check(check, where)
  449. def __init__(self, station):
  450. super(UpdateGarbageCheckEvent, self).__init__(station)
  451. def start(self, check, where):
  452. self.thread = TkThreading(self.func, check, where)
  453. return self
  454. def done_after_event(self):
  455. res: int = self.thread.wait_event()
  456. if res == -1:
  457. self.station.show_warning("更新失败", f"更新垃圾袋-检测结果失败")
  458. else:
  459. self.station.show_msg("更新成功", f"成功更新{res}个垃圾袋-检测结果")
  460. class CountTimeEvent(AdminEventBase):
  461. """
  462. 任务: 按时段统计数据
  463. """
  464. def func(self, column: List, get_name: Callable):
  465. res = {}
  466. cur = self._db.search(columns=["DATE_FORMAT(UseTime,'%H') AS days", "count(GarbageID) AS count", *column],
  467. table="garbage",
  468. group_by=["days", *column],
  469. order_by=[(c, "DESC") for c in column] + [("days", "ASC")],
  470. where="UseTime IS NOT NULL")
  471. if cur is None:
  472. return None
  473. loc_list = cur.fetchall()
  474. loc_type = []
  475. for i in loc_list:
  476. name = get_name(i)
  477. if name not in loc_type:
  478. loc_type.append(name)
  479. lst: List = res.get(name, list())
  480. lst.append(i)
  481. res[name] = lst
  482. res['res_type'] = loc_type
  483. return res, loc_list
  484. def __init__(self, gb_station):
  485. super().__init__(gb_station)
  486. self.thread = None
  487. self._program: Optional[admin_program.StatisticsTimeProgramBase] = None
  488. def start(self, column: List, get_name: Callable, program):
  489. self.thread = TkThreading(self.func, column, get_name)
  490. self._program = program
  491. return self
  492. def is_end(self) -> bool:
  493. return not self.thread.is_alive()
  494. def done_after_event(self):
  495. res: Optional[Tuple[Dict[str, str], List]] = self.thread.wait_event()
  496. if res is None:
  497. self.station.show_warning("数据分析", "数据获取时发生错误")
  498. else:
  499. self._program.show_result(res[0], res[1])
  500. class CountScoreReputationTinyEvent(AdminEventBase):
  501. """
  502. 任务: 统计积分和用户数据
  503. """
  504. def func(self):
  505. cur = self._db.search(columns=["ceil(Score / 5) AS ScoreGB",
  506. "ceil(Reputation / 10) AS ReputationGB",
  507. "count(UserID) AS count"],
  508. table="user",
  509. group_by=["ScoreGB", "ReputationGB"],
  510. order_by=[("ScoreGB", "DESC"), ("ReputationGB", "DESC")],
  511. where="IsManager=0")
  512. if cur is None:
  513. return None
  514. lst = np.zeros((100, 100))
  515. for i in range(cur.rowcount):
  516. res = cur.fetchone()
  517. x = res[0]
  518. y = res[1]
  519. if x == 100:
  520. x = 99 # 边界值
  521. if y == 100:
  522. y = 99 # 边界值
  523. lst[x, y] = int(res[2])
  524. return lst
  525. def __init__(self, gb_station):
  526. super().__init__(gb_station)
  527. self.thread = None
  528. self._program: Optional[admin_program.StatisticsUserTinyProgram] = None
  529. def start(self, program):
  530. self.thread = TkThreading(self.func)
  531. self._program = program
  532. return self
  533. def is_end(self) -> bool:
  534. return not self.thread.is_alive()
  535. def done_after_event(self):
  536. res: Optional[np.array] = self.thread.wait_event()
  537. if res is None:
  538. self.station.show_warning("数据分析", "数据获取时发生错误")
  539. else:
  540. self._program.show_result(res)
  541. class CountScoreReputationLargeEvent(AdminEventBase):
  542. """
  543. 任务: 统计积分和用户数据
  544. """
  545. def func(self):
  546. cur = self._db.search(columns=["ceil(Score / 50) AS ScoreGB",
  547. "ceil(Reputation / 100) AS ReputationGB",
  548. "count(UserID) AS count"],
  549. table="user",
  550. group_by=["ScoreGB", "ReputationGB"],
  551. order_by=[("ScoreGB", "DESC"), ("ReputationGB", "DESC")],
  552. where="IsManager=0")
  553. if cur is None:
  554. return None
  555. lst = np.zeros((10, 10))
  556. for i in range(cur.rowcount):
  557. res = cur.fetchone()
  558. x = res[0]
  559. y = res[1]
  560. if x == 10:
  561. x = 9 # 边界值
  562. if y == 10:
  563. y = 9 # 边界值
  564. lst[x, y] = int(res[2])
  565. return lst
  566. def __init__(self, gb_station):
  567. super().__init__(gb_station)
  568. self.thread = None
  569. self._program: Optional[admin_program.StatisticsUserLargeProgram] = None
  570. def start(self, program):
  571. self.thread = TkThreading(self.func)
  572. self._program = program
  573. return self
  574. def is_end(self) -> bool:
  575. return not self.thread.is_alive()
  576. def done_after_event(self):
  577. res: Optional[np.array] = self.thread.wait_event()
  578. if res is None:
  579. self.station.show_warning("数据分析", "数据获取时发生错误")
  580. else:
  581. self._program.show_result(res)
  582. class ScoreReputationDistributedEvent(AdminEventBase):
  583. """
  584. 任务: 统计积分或信用分布
  585. """
  586. def func(self, which):
  587. cur = self._db.search(columns=[which],
  588. table="user",
  589. order_by=[(which, "DESC")],
  590. where="IsManager=0")
  591. if cur is None:
  592. return None
  593. return cur.fetchall()
  594. def __init__(self, gb_station):
  595. super().__init__(gb_station)
  596. self.thread = None
  597. self._program: Optional[admin_program.StatisticsScoreDistributedProgram] = None
  598. def start(self, which, program):
  599. self.thread = TkThreading(self.func, which)
  600. self._program = program
  601. return self
  602. def is_end(self) -> bool:
  603. return not self.thread.is_alive()
  604. def done_after_event(self):
  605. res: Optional[np.array] = self.thread.wait_event()
  606. if res is None:
  607. self.station.show_warning("数据分析", "数据获取时发生错误")
  608. else:
  609. lst = [int(i[0]) for i in res]
  610. self._program.show_result(lst)
  611. class PassingRateEvent(AdminEventBase):
  612. def func(self, columns, where, where_select, order_by):
  613. where_str = " AND ".join(["g.CheckResult is not null", *where_select])
  614. columns += [f"get_avg(count(GarbageID), "
  615. f"(SELECT count(g.GarbageID) FROM garbage AS g WHERE {where_str})) AS count"]
  616. where += ["CheckResult is not null", "CheckResult=1"]
  617. cur = self._db.search(columns=columns,
  618. table="garbage",
  619. where=where,
  620. order_by=[(i, "DESC") for i in order_by] + [("count", "DESC")],
  621. group_by=order_by if len(order_by) != 0 else None)
  622. if cur is None:
  623. return None
  624. return cur.fetchall()
  625. def __init__(self, gb_station):
  626. super().__init__(gb_station)
  627. self.thread = None
  628. self._program: Optional[admin_program.StatisticsScoreDistributedProgram] = None
  629. def start(self, columns, where, where_select, order_by, program):
  630. self.thread = TkThreading(self.func, columns, where, where_select, order_by)
  631. self._program = program
  632. return self
  633. def is_end(self) -> bool:
  634. return not self.thread.is_alive()
  635. def done_after_event(self):
  636. res: Optional[List] = self.thread.wait_event()
  637. if res is None:
  638. self.station.show_warning("数据分析", "数据获取时发生错误")
  639. else:
  640. self._program.show_result(res)
  641. class CountDateEvent(AdminEventBase):
  642. """
  643. 任务: 按日期统计数据
  644. """
  645. def func(self, days: int, column: List, get_name: Callable):
  646. res = {}
  647. start = time.strftime("%Y-%m-%d", time.localtime(time.time()))
  648. cur = self._db.search(columns=["days", "count(GarbageID) AS count", *column],
  649. table=f"garbage_{days}d",
  650. group_by=["days", *column],
  651. order_by=[(c, "DESC") for c in column] + [("days", "ASC")],
  652. where="UseTime IS NOT NULL")
  653. if cur is None:
  654. return None
  655. end = time.strftime("%Y-%m-%d", time.localtime(time.time()))
  656. if end != start: # 时间校验, 确保调用前后时间一致
  657. return None
  658. loc_list = cur.fetchall()
  659. loc_type = []
  660. for i in loc_list:
  661. name = get_name(i)
  662. if name not in loc_type:
  663. loc_type.append(name)
  664. lst: List = res.get(name, list())
  665. lst.append(i)
  666. res[name] = lst
  667. res['res_type'] = loc_type
  668. return res, loc_list, end
  669. def __init__(self, gb_station):
  670. super().__init__(gb_station)
  671. self.thread = None
  672. self._program: Optional[admin_program.StatisticsDateProgramBase] = None
  673. def start(self, days, column: List, get_name: Callable, program):
  674. self.thread = TkThreading(self.func, days, column, get_name)
  675. self._program = program
  676. return self
  677. def is_end(self) -> bool:
  678. return not self.thread.is_alive()
  679. def done_after_event(self):
  680. res: Optional[Tuple[Dict[str, str], List, str]] = self.thread.wait_event()
  681. if res is None:
  682. self.station.show_warning("数据分析", "数据获取时发生错误")
  683. else:
  684. self._program.show_result(res[0], res[1], res[2])