# coding=utf-8 from __future__ import division import json from datetime import timedelta,datetime from collections import defaultdict jobnum_region_dict = { "10": "省公司", "110": "武汉分公司", "170": "襄阳分公司", "130": "鄂州分公司", "260": "孝感分公司", "250": "黄冈分公司", "120": "黄石分公司", "190": "咸宁分公司", "200": "荆州分公司", "140": "宜昌分公司", "150": "恩施分公司", "160": "十堰分公司", "240": "随州分公司", "230": "荆门分公司", "1801": "江汉分公司", "1802": "潜江分公司", "1803": "天门分公司" } def keep_digits_filter(code): """ 工号可能有字面，去掉字面，以数字起算 """ return ''.join(filter(str.isdigit, str(code))) def find_region_by_code(code, region_dict): """ 查询工号对应公司 未查询到 返回错误工号 """ code_str = keep_digits_filter(code) # 使用生成器表达式和next函数尝试找到匹配的前缀 company = next( (region_dict.get(code_str[:i]) for i in range(2, min(len(code_str), 5)) if code_str[:i] in region_dict), "错误工号") return company def ip_summary_data_format(ip_summary_data): """ ip维度数据转换方法 """ result = {"summary": {"ip": []}, "detail": {"ip": {}}} grouped_data = defaultdict(lambda: {"reqs": 0, "ips": set()}) ip_detail_dict = defaultdict(lambda: defaultdict(lambda: {"req_frequency": 0})) # 全部账号元组 ips_total = set() for ip_data in ip_summary_data: company = find_region_by_code(ip_data["jobnum"], jobnum_region_dict) count = ip_data["count"] ip = ip_data["ip"] jobnum = ip_data["jobnum"] ip_detail_dict_key = "{}{}".format(ip, jobnum) # 更新统计数据 grouped_data[company]["reqs"] += count grouped_data[company]["ips"].add(ip) ips_total.add(ip) # 构建下钻详情 ip_detail_dict[company][ip_detail_dict_key]["req_ip"] = ip ip_detail_dict[company][ip_detail_dict_key]["req_jobnum"] = jobnum ip_detail_dict[company][ip_detail_dict_key]["req_frequency"] += count # 统计总请求次数和独立IP数 reqs_total = sum(data["reqs"] for data in grouped_data.values()) # 请求为0抛出 if reqs_total == 0: return result # 构建summary部分 ip_data_list = [ { "company": company, "req_frequency": data["reqs"], # 本公司的 请求次数/所有公司 请求次数的合计 "frequency_rate": round(data["reqs"] / reqs_total, 4), "ip_count": len(data["ips"]), # 本公司的 ip个数/所有公司 ip个数的合计 "ip_rate": round(len(data["ips"]) / len(ips_total), 4), # 本公司的 请求次数/本公司 ip个数的合计 "ip_avg": data["reqs"] // len(data["ips"]), } for company, data in grouped_data.items() ] result["summary"]["ip"] = sorted(ip_data_list, key=lambda x: x["req_frequency"], reverse=True) # 构建detail部分 result["detail"]["ip"] = { company: sorted(data.values(), key=lambda x: x['req_frequency'], reverse=True) for company, data in ip_detail_dict.items() } return result def account_summary_data_format(account_summary_data): """ 账号维度数据转换方法 """ result = {"summary": {"account": []}, "detail": {"account": {}}} grouped_data = defaultdict(lambda: {"reqs": 0, "accounts": set()}) account_detail_dict = defaultdict(lambda: defaultdict(lambda: {"req_frequency": 0})) accounts_total = set() for account_data in account_summary_data: company = find_region_by_code(account_data["jobnum"], jobnum_region_dict) count = account_data["count"] account = account_data["account"] jobnum = account_data["jobnum"] account_detail_dict_key = "{}{}".format(account, jobnum) # 更新统计数据 grouped_data[company]["reqs"] += count grouped_data[company]["accounts"].add(account) accounts_total.add(account) # 更新下钻详情 account_detail_dict[company][account_detail_dict_key]["req_account"] = account account_detail_dict[company][account_detail_dict_key]["req_jobnum"] = jobnum account_detail_dict[company][account_detail_dict_key]["req_frequency"] += count # 统计总请求次数和独立账号数 reqs_total = sum(data["reqs"] for data in grouped_data.values()) # 请求为0抛出 if reqs_total == 0: return result # 构建summary部分 account_data_list = [ { "company": company, "req_frequency": data["reqs"], # 本公司的 请求次数/所有公司 请求次数的合计 "frequency_rate": round(data["reqs"] / reqs_total, 4), "account_count": len(data["accounts"]), # 本公司的 账号次数/所有公司 账号次数的合计 "account_rate": round(len(data["accounts"]) / len(accounts_total), 4), # 本公司的 请求次数/本公司 账号次数的合计 "account_avg": data["reqs"] // len(data["accounts"]), } for company, data in grouped_data.items() ] result["summary"]["account"] = sorted(account_data_list, key=lambda x: x["req_frequency"], reverse=True) # 构建detail部分 result["detail"]["account"] = {company: sorted(data.values(), key=lambda x: x['req_frequency'], reverse=True) for company, data in account_detail_dict.items()} return result def interface_summary_data_format(interface_summary_data): """ 接口维度数据转换方法 """ result = {"summary": {"interface": []}, "detail": {"interface": {}}} grouped_data = defaultdict(lambda: {"reqs": 0}) interface_detail_dict = defaultdict(lambda: defaultdict(lambda: {"req_frequency": 0})) for interface_data in interface_summary_data: count = interface_data["count"] interface = interface_data["interface"] jobnum = interface_data["jobnum"] account = interface_data["account"] ip = interface_data["ip"] interface_detail_dict_key = "{}{}".format(ip, account, jobnum) # 更新统计数据 grouped_data[interface]["reqs"] += count # 构建下钻详情 interface_detail_dict[interface][interface_detail_dict_key]["interface_addr"] = interface interface_detail_dict[interface][interface_detail_dict_key]["req_ip"] = ip interface_detail_dict[interface][interface_detail_dict_key]["req_account"] = account interface_detail_dict[interface][interface_detail_dict_key]["req_jobnum"] = jobnum interface_detail_dict[interface][interface_detail_dict_key]["req_frequency"] += count # 统计总请求次数 reqs_total = sum(data["reqs"] for data in grouped_data.values()) # 请求为0抛出 if reqs_total == 0: return result # 构建summary部分 interface_data_list = [ { "interface_addr": interface, "req_frequency": data["reqs"], # 本接口的 请求次数/所有接口 请求次数的合计 "frequency_rate": round(data["reqs"] / reqs_total, 4), # 本接口的 请求次数/ 20 查询top20接口 "frequency_avg": data["reqs"] // 20, } for company, data in grouped_data.items() ] result["summary"]["interface"] = sorted(interface_data_list, key=lambda x: x["req_frequency"], reverse=True) # 构建detail部分 result["detail"]["interface"] = { company: sorted(data.values(), key=lambda x: x["req_frequency"], reverse=True) for company, data in interface_detail_dict.items() } return result def menu_summary_data_format(menu_summary_data): """ 菜单维度数据转换方法 """ result = {"summary": {"menu": []}, "detail": {"menu": {}}} grouped_data = defaultdict(lambda: {"reqs": 0, "menu": set()}) menu_detail_dict = defaultdict(lambda: defaultdict(lambda: {"req_frequency": 0})) menu_total = set() for menu_data in menu_summary_data: count = menu_data["count"] menu = menu_data["menu"] jobnum = menu_data["jobnum"] account = menu_data["account"] ip = menu_data["ip"] menu_detail_dict_key = "{}{}{}".format(ip, account, jobnum) # 更新统计数据和独立菜单数 grouped_data[menu]["reqs"] += count grouped_data[menu]["menu"].add(menu) menu_total.add(menu) # 构建下钻详情 menu_detail_dict[menu][menu_detail_dict_key]["menu_name"] = menu menu_detail_dict[menu][menu_detail_dict_key]["req_ip"] = ip menu_detail_dict[menu][menu_detail_dict_key]["req_account"] = account menu_detail_dict[menu][menu_detail_dict_key]["req_jobnum"] = jobnum menu_detail_dict[menu][menu_detail_dict_key]["req_frequency"] += count # 统计总请求次数 reqs_total = sum(data["reqs"] for data in grouped_data.values()) # 请求为0抛出 if reqs_total == 0 or menu_total == 0: return result # 构建summary部分 menu_data_list = [ { "menu_name": menu, "req_frequency": data["reqs"], # 本菜单的 请求次数 /所有菜单 请求次数的合计 "frequency_rate": round(data["reqs"] / reqs_total, 4), # 本菜单的 请求次数 /所有菜单 个数的合计 "frequency_avg": data["reqs"] // len(menu_total), } for company, data in grouped_data.items() ] result["summary"]["menu"] = sorted(menu_data_list, key=lambda x: x["req_frequency"], reverse=True) # 构建detail部分 result["detail"]["menu"] = {company: sorted(data.values(), key=lambda x: x["req_frequency"], reverse=True) for company, data in menu_detail_dict.items()} return result #调整时间 def adjust_times(start_time, end_time): # 计算开始时间和结束时间之间的天数差 delta_days = (end_time - start_time).days # 从开始和结束时间各自减去这个天数差 adjusted_start_time = start_time - timedelta(days=delta_days) adjusted_end_time = end_time - timedelta(days=delta_days) return adjusted_start_time, adjusted_end_time