123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875 |
- import time
- import numpy as np
- from tool.typing import *
- from sql import DBBit
- from sql.db import DB
- from sql.user import find_user_by_name
- from core.user import User
- from core.garbage import GarbageBag, GarbageType
- from .event import TkEventBase, TkThreading
- from . import admin
- from . import admin_program
- class AdminEventBase(TkEventBase):
- def __init__(self, station):
- super(AdminEventBase, self).__init__()
- self.station: admin.AdminStationBase = station
- self._db: DB = station.get_db()
- def get_title(self) -> str:
- return "AdminEvent"
- class LoginEvent(AdminEventBase):
- def __init__(self, station):
- super().__init__(station)
- self.thread: Optional[TkThreading] = None
- def login(self, name, passwd):
- return find_user_by_name(name, passwd, self._db)
- def start(self, name, passwd):
- self.thread = TkThreading(self.login, name, passwd)
- return self
- def is_end(self) -> bool:
- return not self.thread.is_alive()
- def done_after_event(self):
- self.station.login(self.thread.wait_event())
- class TestProgressEvent(AdminEventBase):
- @staticmethod
- def func(sleep_time):
- time.sleep(sleep_time)
- def __init__(self, station):
- super(TestProgressEvent, self).__init__(station)
- self.thread = TkThreading(self.func, 5)
- class CreateUserEvent(AdminEventBase):
- def func(self, name, passwd, phone, is_manager):
- return self.station.create_user(name, passwd, phone, is_manager)
- def __init__(self, station):
- super(CreateUserEvent, self).__init__(station)
- self._name = None
- def start(self, name, passwd, phone, is_manager):
- self._name = name
- self.thread = TkThreading(self.func, name, passwd, phone, is_manager)
- return self
- def done_after_event(self):
- res: Optional[User] = self.thread.wait_event()
- if res is None:
- self.station.show_warning("创建用户错误", f"无法创建用户 user: {self._name}")
- else:
- name = res.get_name()
- self.station.show_msg("创建用户成功", f"成功创建 {name} 新用户")
- class CreateGarbageEvent(AdminEventBase):
- def func(self, path, count):
- return self.station.create_garbage(path, count)
- def __init__(self, station):
- super(CreateGarbageEvent, self).__init__(station)
- self._name = None
- def start(self, path, count):
- self.thread = TkThreading(self.func, path, count)
- return self
- def done_after_event(self):
- res: list[tuple[str, Optional[GarbageBag]]] = self.thread.wait_event()
- self.station.show_msg("创建垃圾袋", f"成功创建{len(res)}个垃圾袋")
- class ExportGarbageByIDEvent(AdminEventBase):
- def func(self, path, gid):
- return self.station.export_garbage_by_gid(path, gid)
- def __init__(self, station):
- super(ExportGarbageByIDEvent, self).__init__(station)
- self._name = None
- def start(self, path, gid):
- self.thread = TkThreading(self.func, path, gid)
- return self
- def done_after_event(self):
- res: tuple[str, Optional[GarbageBag]] = self.thread.wait_event()
- if res[1] is None:
- self.station.show_warning("导出错误", f"无法导出垃圾袋二维码")
- else:
- self.station.show_msg("导出成功", f"成功导出垃圾袋二维码")
- class ExportGarbageAdvancedEvent(AdminEventBase):
- def func(self, path, where):
- return self.station.export_garbage(path, where)
- def __init__(self, station):
- super(ExportGarbageAdvancedEvent, self).__init__(station)
- self._name = None
- def start(self, path, where):
- self.thread = TkThreading(self.func, path, where)
- return self
- def done_after_event(self):
- res: list[tuple[str]] = self.thread.wait_event()
- self.station.show_msg("导出完成", f"导出{len(res)}个垃圾袋二维码")
- class ExportUserByIDEvent(AdminEventBase):
- def func(self, path, uid):
- return self.station.export_user_by_uid(path, uid)
- def __init__(self, station):
- super(ExportUserByIDEvent, self).__init__(station)
- self._name = None
- def start(self, path, uid):
- self.thread = TkThreading(self.func, path, uid)
- return self
- def done_after_event(self):
- res: tuple[str, Optional[GarbageBag]] = self.thread.wait_event()
- if res[1] is None:
- self.station.show_warning("导出错误", f"无法导出用户二维码")
- else:
- self.station.show_msg("导出成功", f"成功导出用户二维码")
- class ExportUserAdvancedEvent(AdminEventBase):
- def func(self, path, where):
- res = self.station.export_user(path, where)
- return res
- def __init__(self, station):
- super(ExportUserAdvancedEvent, self).__init__(station)
- self._name = None
- def start(self, path, where):
- self.thread = TkThreading(self.func, path, where)
- return self
- def done_after_event(self):
- res: list[tuple[str]] = self.thread.wait_event()
- self.station.show_msg("导出完成", f"导出{len(res)}个用户二维码")
- class CreateUserFromCSVEvent(AdminEventBase):
- def func(self, path):
- return self.station.create_user_from_csv(path)
- def __init__(self, station):
- super(CreateUserFromCSVEvent, self).__init__(station)
- self._name = None
- def start(self, path):
- self.thread = TkThreading(self.func, path)
- return self
- def done_after_event(self):
- res: list[User] = self.thread.wait_event()
- self.station.show_msg("创建完成", f"从CSV创建{len(res)}个新用户")
- class CreateAutoUserFromCSVEvent(AdminEventBase):
- def func(self, path):
- return self.station.create_auto_user_from_csv(path)
- def __init__(self, station):
- super(CreateAutoUserFromCSVEvent, self).__init__(station)
- self._name = None
- def start(self, path):
- self.thread = TkThreading(self.func, path)
- return self
- def done_after_event(self):
- res: list[User] = self.thread.wait_event()
- self.station.show_msg("创建完成", f"从CSV创建{len(res)}个新用户")
- class DelUserEvent(AdminEventBase):
- def func(self, uid):
- return self.station.del_user(uid)
- def __init__(self, station):
- super(DelUserEvent, self).__init__(station)
- self._name = None
- def start(self, uid):
- self.thread = TkThreading(self.func, uid)
- return self
- def done_after_event(self):
- res: bool = self.thread.wait_event()
- if res:
- self.station.show_msg("删除成功", f"删除用户成功")
- else:
- self.station.show_warning("删除失败", f"删除用户失败")
- class DelUserFromWhereScanEvent(AdminEventBase):
- def func(self, where):
- return self.station.del_user_from_where_scan(where)
- def __init__(self, station):
- super(DelUserFromWhereScanEvent, self).__init__(station)
- def start(self, where):
- self.thread = TkThreading(self.func, where)
- return self
- def done_after_event(self):
- res: int = self.thread.wait_event()
- if res != -1:
- self.station.show_msg("扫描结果", f"符合删除要素的用户个数: {res}")
- else:
- self.station.show_warning("扫描结果", f"获取扫描结果失败")
- class DelUserFromWhereEvent(AdminEventBase):
- def func(self, where):
- return self.station.del_user_from_where(where)
- def __init__(self, station):
- super(DelUserFromWhereEvent, self).__init__(station)
- def start(self, where):
- self.thread = TkThreading(self.func, where)
- return self
- def done_after_event(self):
- res: int = self.thread.wait_event()
- if res != -1:
- self.station.show_msg("扫描结果", f"成功删除{res}个用户")
- else:
- self.station.show_warning("扫描结果", f"删除用户失败")
- class DelGarbageEvent(AdminEventBase):
- def func(self, gid, where):
- if where == 0:
- return self.station.del_garbage(gid)
- elif where == 1:
- return self.station.del_garbage_not_use(gid)
- elif where == 2:
- return self.station.del_garbage_wait_check(gid)
- elif where == 3:
- return self.station.del_garbage_has_check(gid)
- return False
- def __init__(self, station):
- super(DelGarbageEvent, self).__init__(station)
- def start(self, gid, where):
- self.thread = TkThreading(self.func, gid, where)
- return self
- def done_after_event(self):
- res: bool = self.thread.wait_event()
- if res:
- self.station.show_msg("删除成功", f"成功删除垃圾袋")
- else:
- self.station.show_warning("删除失败", f"删除垃圾袋失败")
- class DelGarbageWhereEvent(AdminEventBase):
- def func(self, where, where_sql):
- if where == 1:
- return self.station.del_garbage_where_not_use(where_sql)
- elif where == 2:
- return self.station.del_garbage_where_wait_check(where_sql)
- elif where == 3:
- return self.station.del_garbage_where_has_check(where_sql)
- return -1
- def __init__(self, station):
- super(DelGarbageWhereEvent, self).__init__(station)
- def start(self, where, where_sql):
- self.thread = TkThreading(self.func, where, where_sql)
- return self
- def done_after_event(self):
- res: int = self.thread.wait_event()
- if res != -1:
- self.station.show_msg("删除成功", f"成功删除{res}个垃圾袋")
- else:
- self.station.show_warning("删除失败", f"垃圾袋删除失败")
- class DelGarbageWhereScanEvent(AdminEventBase):
- def func(self, where, where_sql):
- if where == 1:
- return self.station.del_garbage_where_scan_not_use(where_sql)
- elif where == 2:
- return self.station.del_garbage_where_scan_wait_check(where_sql)
- elif where == 3:
- return self.station.del_garbage_where_scan_has_check(where_sql)
- return -1
- def __init__(self, station):
- super(DelGarbageWhereScanEvent, self).__init__(station)
- def start(self, where, where_sql):
- self.thread = TkThreading(self.func, where, where_sql)
- return self
- def done_after_event(self):
- res: int = self.thread.wait_event()
- if res != -1:
- self.station.show_msg("扫描结果", f"符合删除要素的垃圾袋个数: {res}")
- else:
- self.station.show_warning("扫描结果", f"获取扫描结果失败")
- class DelAllGarbageScanEvent(AdminEventBase):
- def func(self):
- return self.station.del_all_garbage_scan()
- def __init__(self, station):
- super(DelAllGarbageScanEvent, self).__init__(station)
- self.thread = TkThreading(self.func)
- def done_after_event(self):
- res: int = self.thread.wait_event()
- if res != -1:
- self.station.show_msg("扫描结果", f"全部垃圾袋个数: {res}")
- else:
- self.station.show_warning("扫描结果", f"获取扫描结果失败")
- class DelAllGarbageEvent(AdminEventBase):
- def func(self):
- return self.station.del_all_garbage()
- def __init__(self, station):
- super(DelAllGarbageEvent, self).__init__(station)
- self.thread = TkThreading(self.func)
- def done_after_event(self):
- res: int = self.thread.wait_event()
- if res != -1:
- self.station.show_msg("删除成功", f"成功删除所用[共计{res}个]垃圾袋")
- else:
- self.station.show_warning("删除失败", f"删除所有垃圾袋失败")
- def set_user_search_result(i):
- i = list(i)
- if i[-1] is not None:
- i[-1] = "是" if i[-1] == DBBit.BIT_1 else "否"
- return i
- class SearchUserEvent(AdminEventBase):
- def func(self, columns, uid, name, phone):
- return self.station.search_user(columns, uid, name, phone)
- def __init__(self, station):
- super(SearchUserEvent, self).__init__(station)
- self.program: Optional[admin_program.SearchUserProgram] = None
- def start(self, columns, uid, name, phone, program):
- self.thread = TkThreading(self.func, columns, uid, name, phone)
- self.program = program
- return self
- def done_after_event(self):
- res: list[list] = self.thread.wait_event()
- if res is None or self.program is None:
- self.station.show_warning("搜索失败", f"搜索用户失败")
- return
- for i in self.program.view.get_children():
- self.program.view.delete(i)
- for i in res:
- i = set_user_search_result(i)
- self.program.view.insert('', 'end', values=i)
- class SearchUserAdvancedEvent(AdminEventBase):
- def func(self, columns, sql):
- return self.station.search_user_advanced(columns, sql)
- def __init__(self, station):
- super(SearchUserAdvancedEvent, self).__init__(station)
- self.program: Optional[admin_program.SearchUserAdvancedProgram] = None
- def start(self, columns, sql, program):
- self.thread = TkThreading(self.func, columns, sql)
- self.program = program
- return self
- def done_after_event(self):
- res: Optional[List[List[any]]] = self.thread.wait_event()
- if res is None or self.program is None:
- self.station.show_warning("高级搜索失败", f"高级搜索用户失败")
- return
- for i in self.program.view.get_children():
- self.program.view.delete(i)
- for i in res:
- i = set_user_search_result(i)
- self.program.view.insert('', 'end', values=i)
- def set_garbage_search_result(data):
- data = list(data)
- if data[6] is not None:
- tmp: bytes = data[6]
- data[6] = GarbageType.GarbageTypeStrList_ch[int(tmp.decode('utf-8'))]
- if data[7] is not None:
- data[7] = "投放正确" if data[7] == DBBit.BIT_1 else "投放错误"
- for index, i in enumerate(data):
- if i is None:
- data[index] = "无"
- return data
- class SearchGarbageEvent(AdminEventBase):
- def func(self, columns, key_values: dict):
- return self.station.search_garbage(columns, key_values)
- def __init__(self, station):
- super(SearchGarbageEvent, self).__init__(station)
- self.program: Optional[admin_program.SearchUserProgram] = None
- def start(self, columns, key_values: dict, program):
- self.thread = TkThreading(self.func, columns, key_values)
- self.program = program
- return self
- def done_after_event(self):
- res: list[list] = self.thread.wait_event()
- if res is None or self.program is None:
- self.station.show_warning("搜索失败", f"搜索垃圾袋失败")
- return
- for i in self.program.view.get_children():
- self.program.view.delete(i)
- for i in res:
- i = set_garbage_search_result(i)
- self.program.view.insert('', 'end', values=i)
- class SearchGarbageAdvancedEvent(AdminEventBase):
- def func(self, columns, sql):
- return self.station.search_garbage_advanced(columns, sql)
- def __init__(self, station):
- super(SearchGarbageAdvancedEvent, self).__init__(station)
- self.program: Optional[admin_program.SearchGarbageAdvancedProgram] = None
- def start(self, columns, sql, program):
- self.thread = TkThreading(self.func, columns, sql)
- self.program = program
- return self
- def done_after_event(self):
- res: list[list] = self.thread.wait_event()
- if res is None or self.program is None:
- self.station.show_warning("高级搜索失败", f"高级搜索垃圾袋失败")
- return
- for i in self.program.view.get_children():
- self.program.view.delete(i)
- for i in res:
- i = set_garbage_search_result(i)
- self.program.view.insert('', 'end', values=i)
- def set_advanced_search_result(data):
- data = list(data)
- if data[-2] is not None:
- tmp: bytes = data[-2]
- data[-2] = GarbageType.GarbageTypeStrList_ch[int(tmp.decode('utf-8'))]
- if data[-1] is not None:
- data[-1] = "投放正确" if data[-1] == DBBit.BIT_1 else "投放错误"
- data = list(data)
- for index, i in enumerate(data):
- if i is None:
- data[index] = "无"
- return data
- class SearchAdvancedEvent(AdminEventBase):
- def func(self, columns, sql):
- return self.station.search_advanced(columns, sql)
- def __init__(self, station):
- super(SearchAdvancedEvent, self).__init__(station)
- self.program: Optional[admin_program.SearchAdvancedProgram] = None
- def start(self, columns, sql, program):
- self.thread = TkThreading(self.func, columns, sql)
- self.program = program
- return self
- def done_after_event(self):
- res: list[list] = self.thread.wait_event()
- if res is None or self.program is None:
- self.station.show_warning("高级搜索失败", f"高级搜索失败")
- return
- for i in self.program.view.get_children():
- self.program.view.delete(i)
- for i in res:
- i = set_advanced_search_result(i)
- self.program.view.insert('', 'end', values=i)
- class UpdateUserScoreEvent(AdminEventBase):
- def func(self, score, where):
- return self.station.update_user_score(score, where)
- def __init__(self, station):
- super(UpdateUserScoreEvent, self).__init__(station)
- def start(self, score, where):
- self.thread = TkThreading(self.func, score, where)
- return self
- def done_after_event(self):
- res: int = self.thread.wait_event()
- if res == -1:
- self.station.show_warning("更新失败", f"更新用户-积分失败")
- else:
- self.station.show_msg("更新成功", f"成功更新{res}个用户-积分")
- class UpdateUserReputationEvent(AdminEventBase):
- def func(self, reputation, where):
- return self.station.update_user_reputation(reputation, where)
- def __init__(self, station):
- super(UpdateUserReputationEvent, self).__init__(station)
- def start(self, reputation, where):
- self.thread = TkThreading(self.func, reputation, where)
- return self
- def done_after_event(self):
- res: int = self.thread.wait_event()
- if res == -1:
- self.station.show_warning("更新失败", f"更新用户-垃圾分类信用失败")
- else:
- self.station.show_msg("更新成功", f"成功更新{res}个用户-垃圾分类信用")
- class UpdateGarbageTypeEvent(AdminEventBase):
- def func(self, type_, where):
- return self.station.update_garbage_type(type_, where)
- def __init__(self, station):
- super(UpdateGarbageTypeEvent, self).__init__(station)
- def start(self, type_, where):
- self.thread = TkThreading(self.func, type_, where)
- return self
- def done_after_event(self):
- res: int = self.thread.wait_event()
- if res == -1:
- self.station.show_warning("更新失败", f"更新垃圾袋-垃圾类型失败")
- else:
- self.station.show_msg("更新成功", f"成功更新{res}个垃圾袋-垃圾类型")
- class UpdateGarbageCheckEvent(AdminEventBase):
- def func(self, check, where):
- return self.station.update_garbage_check(check, where)
- def __init__(self, station):
- super(UpdateGarbageCheckEvent, self).__init__(station)
- def start(self, check, where):
- self.thread = TkThreading(self.func, check, where)
- return self
- def done_after_event(self):
- res: int = self.thread.wait_event()
- if res == -1:
- self.station.show_warning("更新失败", f"更新垃圾袋-检测结果失败")
- else:
- self.station.show_msg("更新成功", f"成功更新{res}个垃圾袋-检测结果")
- class CountTimeEvent(AdminEventBase):
- """
- 任务: 按时段统计数据
- """
- def func(self, column: List, get_name: Callable):
- res = {}
- cur = self._db.search(columns=["DATE_FORMAT(UseTime,'%H') AS days", "count(GarbageID) AS count", *column],
- table="garbage",
- group_by=["days", *column],
- order_by=[(c, "DESC") for c in column] + [("days", "ASC")],
- where="UseTime IS NOT NULL")
- if cur is None:
- return None
- loc_list = cur.fetchall()
- loc_type = []
- for i in loc_list:
- name = get_name(i)
- if name not in loc_type:
- loc_type.append(name)
- lst: List = res.get(name, list())
- lst.append(i)
- res[name] = lst
- res['res_type'] = loc_type
- return res, loc_list
- def __init__(self, gb_station):
- super().__init__(gb_station)
- self.thread = None
- self._program: Optional[admin_program.StatisticsTimeProgramBase] = None
- def start(self, column: List, get_name: Callable, program):
- self.thread = TkThreading(self.func, column, get_name)
- self._program = program
- return self
- def is_end(self) -> bool:
- return not self.thread.is_alive()
- def done_after_event(self):
- res: Optional[Tuple[Dict[str, str], List]] = self.thread.wait_event()
- if res is None:
- self.station.show_warning("数据分析", "数据获取时发生错误")
- else:
- self._program.show_result(res[0], res[1])
- class CountScoreReputationTinyEvent(AdminEventBase):
- """
- 任务: 统计积分和用户数据
- """
- def func(self):
- cur = self._db.search(columns=["ceil(Score / 5) AS ScoreGB",
- "ceil(Reputation / 10) AS ReputationGB",
- "count(UserID) AS count"],
- table="user",
- group_by=["ScoreGB", "ReputationGB"],
- order_by=[("ScoreGB", "DESC"), ("ReputationGB", "DESC")],
- where="IsManager=0")
- if cur is None:
- return None
- lst = np.zeros((100, 100))
- for i in range(cur.rowcount):
- res = cur.fetchone()
- x = res[0]
- y = res[1]
- if x == 100:
- x = 99 # 边界值
- if y == 100:
- y = 99 # 边界值
- lst[x, y] = int(res[2])
- return lst
- def __init__(self, gb_station):
- super().__init__(gb_station)
- self.thread = None
- self._program: Optional[admin_program.StatisticsUserTinyProgram] = None
- def start(self, program):
- self.thread = TkThreading(self.func)
- self._program = program
- return self
- def is_end(self) -> bool:
- return not self.thread.is_alive()
- def done_after_event(self):
- res: Optional[np.array] = self.thread.wait_event()
- if res is None:
- self.station.show_warning("数据分析", "数据获取时发生错误")
- else:
- self._program.show_result(res)
- class CountScoreReputationLargeEvent(AdminEventBase):
- """
- 任务: 统计积分和用户数据
- """
- def func(self):
- cur = self._db.search(columns=["ceil(Score / 50) AS ScoreGB",
- "ceil(Reputation / 100) AS ReputationGB",
- "count(UserID) AS count"],
- table="user",
- group_by=["ScoreGB", "ReputationGB"],
- order_by=[("ScoreGB", "DESC"), ("ReputationGB", "DESC")],
- where="IsManager=0")
- if cur is None:
- return None
- lst = np.zeros((10, 10))
- for i in range(cur.rowcount):
- res = cur.fetchone()
- x = res[0]
- y = res[1]
- if x == 10:
- x = 9 # 边界值
- if y == 10:
- y = 9 # 边界值
- lst[x, y] = int(res[2])
- return lst
- def __init__(self, gb_station):
- super().__init__(gb_station)
- self.thread = None
- self._program: Optional[admin_program.StatisticsUserLargeProgram] = None
- def start(self, program):
- self.thread = TkThreading(self.func)
- self._program = program
- return self
- def is_end(self) -> bool:
- return not self.thread.is_alive()
- def done_after_event(self):
- res: Optional[np.array] = self.thread.wait_event()
- if res is None:
- self.station.show_warning("数据分析", "数据获取时发生错误")
- else:
- self._program.show_result(res)
- class ScoreReputationDistributedEvent(AdminEventBase):
- """
- 任务: 统计积分或信用分布
- """
- def func(self, which):
- cur = self._db.search(columns=[which],
- table="user",
- order_by=[(which, "DESC")],
- where="IsManager=0")
- if cur is None:
- return None
- return cur.fetchall()
- def __init__(self, gb_station):
- super().__init__(gb_station)
- self.thread = None
- self._program: Optional[admin_program.StatisticsScoreDistributedProgram] = None
- def start(self, which, program):
- self.thread = TkThreading(self.func, which)
- self._program = program
- return self
- def is_end(self) -> bool:
- return not self.thread.is_alive()
- def done_after_event(self):
- res: Optional[np.array] = self.thread.wait_event()
- if res is None:
- self.station.show_warning("数据分析", "数据获取时发生错误")
- else:
- lst = [int(i[0]) for i in res]
- self._program.show_result(lst)
- class PassingRateEvent(AdminEventBase):
- def func(self, columns, where, where_select, order_by):
- where_str = " AND ".join(["g.CheckResult is not null", *where_select])
- columns += [f"get_avg(count(GarbageID), "
- f"(SELECT count(g.GarbageID) FROM garbage AS g WHERE {where_str})) AS count"]
- where += ["CheckResult is not null", "CheckResult=1"]
- cur = self._db.search(columns=columns,
- table="garbage",
- where=where,
- order_by=[(i, "DESC") for i in order_by] + [("count", "DESC")],
- group_by=order_by if len(order_by) != 0 else None)
- if cur is None:
- return None
- return cur.fetchall()
- def __init__(self, gb_station):
- super().__init__(gb_station)
- self.thread = None
- self._program: Optional[admin_program.StatisticsScoreDistributedProgram] = None
- def start(self, columns, where, where_select, order_by, program):
- self.thread = TkThreading(self.func, columns, where, where_select, order_by)
- self._program = program
- return self
- def is_end(self) -> bool:
- return not self.thread.is_alive()
- def done_after_event(self):
- res: Optional[List] = self.thread.wait_event()
- if res is None:
- self.station.show_warning("数据分析", "数据获取时发生错误")
- else:
- self._program.show_result(res)
- class CountDateEvent(AdminEventBase):
- """
- 任务: 按日期统计数据
- """
- def func(self, days: int, column: List, get_name: Callable):
- res = {}
- start = time.strftime("%Y-%m-%d", time.localtime(time.time()))
- cur = self._db.search(columns=["days", "count(GarbageID) AS count", *column],
- table=f"garbage_{days}d",
- group_by=["days", *column],
- order_by=[(c, "DESC") for c in column] + [("days", "ASC")],
- where="UseTime IS NOT NULL")
- if cur is None:
- return None
- end = time.strftime("%Y-%m-%d", time.localtime(time.time()))
- if end != start: # 时间校验, 确保调用前后时间一致
- return None
- loc_list = cur.fetchall()
- loc_type = []
- for i in loc_list:
- name = get_name(i)
- if name not in loc_type:
- loc_type.append(name)
- lst: List = res.get(name, list())
- lst.append(i)
- res[name] = lst
- res['res_type'] = loc_type
- return res, loc_list, end
- def __init__(self, gb_station):
- super().__init__(gb_station)
- self.thread = None
- self._program: Optional[admin_program.StatisticsDateProgramBase] = None
- def start(self, days, column: List, get_name: Callable, program):
- self.thread = TkThreading(self.func, days, column, get_name)
- self._program = program
- return self
- def is_end(self) -> bool:
- return not self.thread.is_alive()
- def done_after_event(self):
- res: Optional[Tuple[Dict[str, str], List, str]] = self.thread.wait_event()
- if res is None:
- self.station.show_warning("数据分析", "数据获取时发生错误")
- else:
- self._program.show_result(res[0], res[1], res[2])
|