这段代码主要问题在于重复代码过多，以及路由处理方式不够优雅。 soft、device、wuhan 三个路由几乎完全相同，只是IP白名单和部分PM列表不同。gantt相关的路由也存在大量重复。 此外，错误处理可以更完善。
以下提供优化方案：
- 重构 IP 白名单:
将 ip_whitelist 改为更易于管理的结构，例如字典嵌套列表：
ip_whitelist = {
'soft': [
('孔鲁文', "172.16.33.191"),
('李嘉政', "172.16.27.248"),
# ...
],
'device': [
('孔鲁文', "172.16.33.191"),
# ...
],
# ...
}这样可以简化IP校验逻辑。
- 创建辅助函数:
创建辅助函数来处理重复的逻辑，例如：
def process_project_data(ip_list, pm_list, remove_projects, is_normal):
"""处理项目数据，用于 soft, device, wuhan 路由"""
removeTemoList = [f" name <> '{project}'" for project in remove_projects]
removeProject = 'AND'.join(removeTemoList)
joinTemp = [f" PmUserId = '{pm_id}'" for pm_name, pm_id, enabled in pm_list if enabled]
PM = 'OR'.join(joinTemp) if joinTemp else "1=0" #避免空字符串导致错误
tempDatajson = {
'Normal': {
'count': None,
'DetailedList': SerachProject(PM, removeProject, True)
},
'over': {
'count': None,
'DetailedList': SerachProject(PM, removeProject, False)
}
}
return render_template('base.html', tempDatajson=tempDatajson) # 创建一个 base.html 模板
def check_ip_and_auth(ip, ip_list, route_name):
"""检查IP地址和权限"""
if ip in [ip_addr for _, ip_addr in ip_list]:
whiteKeyName = [name for name, ip_addr in ip_list if ip_addr == ip][0]
logger.info(f'{whiteKeyName} | {ip} 访问/{route_name}')
return True
else:
logger.info(f'未授权 | {ip} 访问/{route_name}')
return False
def auth_required(func):
"""装饰器，用于检查权限"""
def wrapper(*args, **kwargs):
ip = request.remote_addr
if not check_ip_and_auth(ip, ip_whitelist[kwargs['route_name']], kwargs['route_name']):
return render_template("AuthFailed.html")
return func(*args, **kwargs)
return wrapper- 使用装饰器简化路由:
使用装饰器 auth_required 来简化路由的权限检查：
@app.route('/device', methods=['GET', 'POST'])
@auth_required
def device():
return process_project_data(ip_whitelist['device'], device_pm_list, remove_projects, True)
@app.route('/soft', methods=['GET', 'POST'])
@auth_required
def soft():
return process_project_data(ip_whitelist['soft'], soft_pm_list, remove_projects, True)
@app.route('/wuhan', methods=['GET', 'POST'])
@auth_required
def wuhan():
return process_project_data(ip_whitelist['wuhan'], wuhan_pm_list, remove_projects, True)你需要定义 device_pm_list, soft_pm_list, wuhan_pm_list 和 remove_projects 变量。
- 重构 Gantt 路由:
gantt 相关的路由也存在大量重复，可以提取公共部分：
def handle_gantt(redis_name, pm_list, statement, template_name):
today = datetime.now().strftime("%Y-%m-%d")
redis_name = f'{redis_name}_{today}'
try:
random_string = request.args.get('token')
if random_string == settings.gantt_tokens[template_name]: # 将token放到settings中
tempDatajson = project_gantt_public(redis_name, pm_list, statement)
return render_template(template_name, tempDatajson=tempDatajson, today=today)
else:
return render_template("AuthFailed.html")
except Exception as e:
print(f"Error in gantt(): {e}")
return "重新请求尝试"
@app.route('/gantt', methods=['GET', 'POST'])
def project_gantt_all_in_one():
return handle_gantt('project_gantt', ['李嘉政', '王清臣', ...], '', 'gantt.html')
@app.route('/gantt_huishiyun', methods=['GET', 'POST'])
def project_gantt_huishiyun():
return handle_gantt('project_gantt_huishiyun', ['李嘉政'], " AND a.ProjectBaseName NOT LIKE '%平台软件日常支撑(2024)%'", 'gantt.html')
# ... 其他 Gantt 路由类似处理 ...在 settings.py 中定义 gantt_tokens 字典来存储每个 Gantt 路由的 token。
- 改进错误处理:
使用 Flask 的 errorhandler 装饰器来处理异常，提供更友好的错误页面。
- 定时任务:
定时任务部分代码可以优化，例如使用更简洁的数据库查询和数据处理方式。 可以考虑使用更强大的数据库ORM框架（如SQLAlchemy）来简化数据库操作。
- 代码规范:
使用更具描述性的变量名。
保持代码缩进一致。
添加必要的注释。
总而言之，通过重构、提取公共方法、使用装饰器和改进错误处理，可以显著提高代码的可读性、可维护性和可扩展性。 记住，代码优化是一个迭代的过程，需要逐步改进。 我提供的方案是一个起点，你可以根据实际情况进行调整和完善。 你需要根据自己的需求创建 base.html 模板以及其他必要的修改。 也请确保你的 settings.py 文件包含了必要的配置信息。
提供一个更结构化、更易于维护的代码框架，你需要根据你的实际环境和数据库配置进行调整。
- settings.py:
import os
# 数据库配置
DATABASE_URL = os.environ.get('DATABASE_URL') or '你的数据库连接字符串'
# Redis 配置
REDIS_HOST = '192.168.10.100'
REDIS_PORT = 6379
REDIS_PASSWORD = 'lijiazheng'
REDIS_DB = 0
# 日志文件路径
LOG_FILE = '/home/bugsSerachZentao/logfile.log'
LOG_MAX_BYTES = 1e6
LOG_BACKUP_COUNT = 3
# Gantt token
gantt_tokens = {
'gantt.html': 'AoVv9Pba3sGjZoGEMjKgs6Bhaw29AaZnzMsEQdwHyqmWY_aio',
'gantt.html': 'N2ALqcECzjiZv9N6kJYBPBW9EJMNHVqL9QvtXTrCdZrmP_lijzh',
'gantt.html': '65KEq8sgQGUK8TQiUCB2MzZJjnlK7TVVevE8TZqk4WVBv_wqc',
'gantt.html': 'RaqWTAGqke46WHdmRep6N9g3fuGxQDPnjE973H7mCaYST_lcy',
'gantt.html': 'or8Alc3NZt9jx5fR79GFQdWrsDAqZDuCldAX4pDgimG73_platform',
'gantt.html': 'Qqu9w3fhJQfWd82U23Vpf8AoVUViYMevdexqQHC3FqERT_testing',
'test.html': 'test_46QYcgwBmmqAkRWL95VkH6aasd7WsYziQiChdWv4f3LbNcsfNGduudyDrgW5uUaV_test',
'iorz.html': '^gwUSSNwzUoRhgg7CZwSUQQUDiWG4XxZqT526DUgaaZhjb94Y95tTUz3w%%oc',
'ptyf.html': '^gwUSSNwzUoRhgg7CZwSUQQUDiWG4XxZqT526DUgaaZhjb94Y95tTUz3w%%oa',
'beziquePlus.html': 'V98cD8H7CZwSUQQUDiWG4XxZqT526DUgHGkHb94Y95t6Wz',
}
# ...其他配置...
- app.py:
from flask import Flask, request, render_template, redirect
from DBmodel.bugs import get_project_list # 假设这个模块存在
from DBmodel.info import zantao_project_baseInfo # 假设这个模块存在
import settings, warnings
from logzero import logger, logfile
from package.tools import Tools # 假设这个模块存在
from Model.project_info import SerachProject # 假设这个模块存在
from Model.timeLine import timelines # 假设这个模块存在
from Model.orz import orzs # 假设这个模块存在
from Model.ptyf import ptyfb # 假设这个模块存在
from Model.beziquePlus import bezique # 假设这个模块存在
from Model.project_gantt import gantt # 假设这个模块存在
from datetime import datetime
from apscheduler.schedulers.background import BackgroundScheduler
import redis, json
from DBmodel.db_report_oa import MysqlDB # 假设这个模块存在
from decimal import Decimal
from collections import defaultdict
warnings.filterwarnings("ignore")
# ... (IP 白名单部分，参考之前的重构方案) ...
app = Flask(__name__)
app.config.from_object(settings)
redis_client = redis.StrictRedis(host=settings.REDIS_HOST, port=settings.REDIS_PORT, password=settings.REDIS_PASSWORD, db=settings.REDIS_DB)
tools = Tools()
logfile(settings.LOG_FILE, maxBytes=settings.LOG_MAX_BYTES, backupCount=settings.LOG_BACKUP_COUNT)
# ... (辅助函数，参考之前的方案) ...
# 路由
@app.route('/', methods=['GET', 'POST'])
def index():
# ... (index 路由逻辑，参考之前的代码，简化重复部分) ...
@app.route('/device', methods=['GET', 'POST'])
@auth_required('device')
def device():
return process_project_data(ip_whitelist['device'], device_pm_list, remove_projects, True, 'device.html')
@app.route('/soft', methods=['GET', 'POST'])
@auth_required('soft')
def soft():
return process_project_data(ip_whitelist['soft'], soft_pm_list, remove_projects, True, 'soft.html')
@app.route('/wuhan', methods=['GET', 'POST'])
@auth_required('wuhan')
def wuhan():
return process_project_data(ip_whitelist['wuhan'], wuhan_pm_list, remove_projects, True, 'wuhan.html')
@app.route('/timeline', methods=['GET', 'POST'])
def timeline():
tempDatajson = timelines()
return render_template('timeline.html', tempDatajson=tempDatajson)
@app.route('/orz', methods=['GET', 'POST'])
@auth_required('jiaban')
def orz():
tempDatajson = orzs()
return render_template('orz.html', tempDatajson=tempDatajson)
@app.route('/ptyf', methods=['GET', 'POST'])
def ptyf():
# ... (ptyf 路由逻辑) ...
@app.route('/beziquePlus', methods=['GET', 'POST'])
def ptbeziquePlus():
# ... (beziquePlus 路由逻辑) ...
@app.route('/gantt', methods=['GET', 'POST'])
def project_gantt_all_in_one():
return handle_gantt('project_gantt', ['李嘉政', '王清臣', ...], '', 'gantt.html')
@app.route('/gantt_huishiyun', methods=['GET', 'POST'])
def project_gantt_huishiyun():
return handle_gantt('project_gantt_huishiyun', ['李嘉政'], " AND a.ProjectBaseName NOT LIKE '%平台软件日常支撑(2024)%'", 'gantt.html')
# ... 其他 Gantt 路由 ...
@app.route('/test', methods=['GET', 'POST'])
def test():
return handle_gantt('project_gantt_huishiyun', ['李嘉政', '王清臣', ...], " AND a.ProjectBaseName NOT LIKE '%平台软件日常支撑(2024)%'", 'test.html')
@app.route('/iorz', methods=['GET', 'POST'])
@auth_required('jiaban')
def iorz():
# ... (iorz 路由逻辑) ...
# ... (错误处理) ...
# ... (定时任务) ...
if __name__ == '__main__':
scheduler = BackgroundScheduler()
scheduler.add_job(cron_OvertimeData, 'cron', hour=16, minute=37)
scheduler.add_job(cron_getProjectTaskDetails, 'cron', hour=7, minute=30)
scheduler.start()
app.run(host='0.0.0.0', port=8080, debug=True)
- 其他模块 (Model, DBmodel 等):
你需要根据你的实际情况完善这些模块。 确保你的数据库连接、数据模型以及其他依赖都正确配置。
注意: 你需要替换占位符，例如数据库连接字符串，以及完善 device_pm_list, soft_pm_list, wuhan_pm_list, remove_projects 等变量，以及 handle_gantt 函数中的 pm_list 和 statement 参数。 还需要创建 base.html, device.html, soft.html, wuhan.html, gantt.html, test.html, iorz.html, timeline.html, ptyf.html, beziquePlus.html 等模板文件。 这个代码框架只是为了展示如何改进代码结构，你仍然需要根据你的实际需求进行调整和完善。 尤其要仔细检查数据库查询语句，避免SQL注入漏洞。 建议使用参数化查询来防止SQL注入。
评论