from flask import Blueprint, request, jsonify from sqlalchemy.orm import sessionmaker from sqlalchemy import create_engine, and_ from backend.models.models import TimeRecord, Project, CutoffPeriod from backend.models.utils import * from datetime import datetime, date, timedelta from collections import defaultdict statistics_bp = Blueprint('statistics', __name__) def get_db_session(): """获取数据库会话""" engine = create_engine('sqlite:///data/timetrack.db') Session = sessionmaker(bind=engine) return Session() @statistics_bp.route('/api/statistics/weekly', methods=['GET']) def get_weekly_statistics(): """获取周统计数据""" try: session = get_db_session() # 获取查询参数 period_id = request.args.get('period_id') start_date = request.args.get('start_date') end_date = request.args.get('end_date') # 如果指定了周期ID，使用周期的日期范围 if period_id: period = session.query(CutoffPeriod).get(int(period_id)) if not period: session.close() return jsonify({'success': False, 'error': '周期不存在'}), 404 start_date = period.start_date end_date = period.end_date target_hours = period.target_hours period_info = period.to_dict() else: # 使用指定的日期范围 if not start_date or not end_date: session.close() return jsonify({'success': False, 'error': '请提供开始日期和结束日期'}), 400 start_date = datetime.strptime(start_date, '%Y-%m-%d').date() end_date = datetime.strptime(end_date, '%Y-%m-%d').date() target_hours = 40 # 默认目标工时 period_info = { 'period_name': f"{start_date.strftime('%m月%d日')}-{end_date.strftime('%m月%d日')}", 'start_date': start_date.isoformat(), 'end_date': end_date.isoformat(), 'target_hours': target_hours } # 查询该时间范围内的所有工时记录 records = session.query(TimeRecord).filter( and_(TimeRecord.date >= start_date, TimeRecord.date <= end_date) ).order_by(TimeRecord.date).all() # 生成完整的日期范围 current_date = start_date daily_records = [] record_dict = {} # 将记录按日期分组 for record in records: if record.date not in record_dict: record_dict[record.date] = [] record_dict[record.date].append(record) # 生成每日汇总 while current_date <= end_date: day_records = record_dict.get(current_date, []) if day_records: # 如果有记录，汇总该日的数据 total_hours = sum([format_hours_to_decimal(r.hours) for r in day_records if r.hours not in ['-', '0:00']]) event_descriptions = [r.event_description for r in day_records if r.event_description and r.event_description != '-'] activity_nums = [r.activity_num for r in day_records if r.activity_num and r.activity_num != '-'] project_names = [] for r in day_records: if r.project and r.project.project_name: if r.project.project_type.value == 'traditional': project_names.append(f"{r.project.customer_name} {r.project.project_code}") else: project_names.append(f"{r.project.customer_name} {r.project.contract_number}") # 获取第一条记录的时间信息 first_record = day_records[0] daily_record = { 'date': current_date.isoformat(), 'day_of_week': get_day_of_week_chinese(current_date), 'event': '; '.join(event_descriptions) if event_descriptions else '-', 'project': '; '.join(set(project_names)) if project_names else '-', 'start_time': first_record.start_time.strftime('%H:%M') if first_record.start_time else '-', 'end_time': first_record.end_time.strftime('%H:%M') if first_record.end_time else '-', 'activity_num': '; '.join(activity_nums) if activity_nums else '-', 'hours': format_decimal_to_hours(total_hours) if total_hours > 0 else ('0:00' if first_record.is_holiday else '-'), 'is_holiday': first_record.is_holiday, 'holiday_type': first_record.holiday_type, 'is_working_on_holiday': any(r.is_working_on_holiday for r in day_records) } else: # 如果没有记录，生成默认记录 holidays = session.query(Holiday).all() holiday_info = is_holiday(current_date, holidays) daily_record = { 'date': current_date.isoformat(), 'day_of_week': get_day_of_week_chinese(current_date), 'event': '-', 'project': '-', 'start_time': '-', 'end_time': '-', 'activity_num': '-', 'hours': '0:00' if holiday_info['is_holiday'] else '-', 'is_holiday': holiday_info['is_holiday'], 'holiday_type': holiday_info['holiday_type'], 'is_working_on_holiday': False } daily_records.append(daily_record) current_date += timedelta(days=1) # 计算统计数据 workday_total = 0 holiday_total = 0 working_days = 0 holiday_work_days = 0 rest_days = 0 for daily in daily_records: hours_decimal = format_hours_to_decimal(daily['hours']) if daily['is_holiday']: if daily['is_working_on_holiday']: holiday_total += hours_decimal holiday_work_days += 1 else: rest_days += 1 else: if hours_decimal > 0: workday_total += hours_decimal working_days += 1 else: rest_days += 1 # 项目工时统计 project_hours = defaultdict(float) for record in records: if record.project and record.hours not in ['-', '0:00']: hours_decimal = format_hours_to_decimal(record.hours) if record.project.project_type.value == 'traditional': project_key = f"{record.project.customer_name} {record.project.project_code}" else: project_key = f"{record.project.customer_name} {record.project.contract_number}" project_hours[project_key] += hours_decimal total_project_hours = sum(project_hours.values()) project_stats = [] for project_name, hours in project_hours.items(): percentage = (hours / total_project_hours * 100) if total_project_hours > 0 else 0 project_stats.append({ 'project': project_name, 'hours': format_decimal_to_hours(hours), 'percentage': round(percentage, 1) }) project_stats.sort(key=lambda x: x['percentage'], reverse=True) result = { 'period': period_info, 'daily_records': daily_records, 'project_hours': project_stats, 'workday_total': format_decimal_to_hours(workday_total), 'holiday_total': format_decimal_to_hours(holiday_total), 'weekly_total': format_decimal_to_hours(workday_total + holiday_total), 'working_days': working_days, 'holiday_work_days': holiday_work_days, 'rest_days': rest_days, 'target_hours': target_hours, 'completion_rate': round((workday_total + holiday_total) / target_hours * 100, 1) if target_hours > 0 else 0 } session.close() return jsonify({'success': True, 'data': result}) except Exception as e: return jsonify({'success': False, 'error': str(e)}), 500 @statistics_bp.route('/api/statistics/periods', methods=['GET']) def get_cutoff_periods(): """获取Cut-Off周期列表""" try: session = get_db_session() periods = session.query(CutoffPeriod).order_by(CutoffPeriod.start_date.desc()).all() result = [period.to_dict() for period in periods] session.close() return jsonify({'success': True, 'data': result}) except Exception as e: return jsonify({'success': False, 'error': str(e)}), 500 @statistics_bp.route('/api/statistics/periods', methods=['POST']) def create_cutoff_period(): """创建Cut-Off周期""" try: data = request.json session = get_db_session() # 验证必填字段 if not all(key in data for key in ['period_name', 'start_date', 'end_date']): return jsonify({'success': False, 'error': '周期名称、开始日期和结束日期为必填项'}), 400 start_date = datetime.strptime(data['start_date'], '%Y-%m-%d').date() end_date = datetime.strptime(data['end_date'], '%Y-%m-%d').date() if start_date >= end_date: return jsonify({'success': False, 'error': '开始日期必须早于结束日期'}), 400 # 计算周数 days_diff = (end_date - start_date).days + 1 weeks = round(days_diff / 7) period = CutoffPeriod( period_name=data['period_name'], start_date=start_date, end_date=end_date, target_hours=data.get('target_hours', weeks * 40), weeks=weeks, year=start_date.year, month=start_date.month ) session.add(period) session.commit() result = period.to_dict() session.close() return jsonify({'success': True, 'data': result}) except Exception as e: return jsonify({'success': False, 'error': str(e)}), 500 @statistics_bp.route('/api/statistics/periods/', methods=['DELETE']) def delete_cutoff_period(period_id): """删除Cut-Off周期""" try: session = get_db_session() period = session.query(CutoffPeriod).get(period_id) if not period: session.close() return jsonify({'success': False, 'error': '周期不存在'}), 404 session.delete(period) session.commit() session.close() return jsonify({'success': True, 'message': '周期已删除'}) except Exception as e: return jsonify({'success': False, 'error': str(e)}), 500