mirror of
https://github.com/opsre/LiteOps.git
synced 2026-02-23 00:20:46 +08:00
288 lines
11 KiB
Python
288 lines
11 KiB
Python
import logging
|
||
from datetime import datetime, timedelta
|
||
from django.http import JsonResponse
|
||
from django.views import View
|
||
from django.utils.decorators import method_decorator
|
||
from django.views.decorators.csrf import csrf_exempt
|
||
from django.db.models import Count
|
||
from ..models import Project, BuildTask, BuildHistory, User, Environment
|
||
|
||
logger = logging.getLogger('apps')
|
||
|
||
@method_decorator(csrf_exempt, name='dispatch')
|
||
class DashboardStatsView(View):
|
||
"""首页统计数据接口"""
|
||
|
||
def get(self, request):
|
||
"""获取首页统计数据"""
|
||
try:
|
||
# 获取项目总数
|
||
project_count = Project.objects.count()
|
||
|
||
# 获取构建任务总数
|
||
task_count = BuildTask.objects.count()
|
||
|
||
# 获取用户总数
|
||
user_count = User.objects.count()
|
||
|
||
# 获取环境总数
|
||
env_count = Environment.objects.count()
|
||
|
||
# 获取总构建数量
|
||
total_builds_count = BuildHistory.objects.count()
|
||
|
||
# 获取最近7天的构建成功率
|
||
seven_days_ago = datetime.now() - timedelta(days=7)
|
||
recent_builds = BuildHistory.objects.filter(create_time__gte=seven_days_ago)
|
||
total_recent_builds = recent_builds.count()
|
||
success_recent_builds = recent_builds.filter(status='success').count()
|
||
|
||
success_rate = 0
|
||
if total_recent_builds > 0:
|
||
success_rate = round((success_recent_builds / total_recent_builds) * 100, 2)
|
||
|
||
return JsonResponse({
|
||
'code': 200,
|
||
'message': '获取首页统计数据成功',
|
||
'data': {
|
||
'project_count': project_count,
|
||
'task_count': task_count,
|
||
'user_count': user_count,
|
||
'env_count': env_count,
|
||
'total_builds_count': total_builds_count,
|
||
'success_rate': success_rate,
|
||
'total_recent_builds': total_recent_builds
|
||
}
|
||
})
|
||
except Exception as e:
|
||
logger.error(f'获取首页统计数据失败: {str(e)}', exc_info=True)
|
||
return JsonResponse({
|
||
'code': 500,
|
||
'message': f'服务器错误: {str(e)}'
|
||
})
|
||
|
||
|
||
@method_decorator(csrf_exempt, name='dispatch')
|
||
class BuildTrendView(View):
|
||
"""构建任务趋势接口"""
|
||
|
||
def get(self, request):
|
||
"""获取构建任务趋势数据"""
|
||
try:
|
||
# 获取时间范围参数，默认为最近7天
|
||
days = int(request.GET.get('days', 7))
|
||
|
||
# 计算日期范围：包含今天在内的最近 days 天
|
||
today = datetime.now().date() # 获取今天的日期部分
|
||
start_date = today - timedelta(days=days - 1) # 开始日期是今天往前 days-1 天
|
||
|
||
# 准备日期列表和结果数据
|
||
date_list = []
|
||
success_data = []
|
||
failed_data = []
|
||
|
||
# 生成从 start_date 到 today 的日期列表
|
||
current_date = start_date
|
||
while current_date <= today:
|
||
date_str = current_date.strftime('%Y-%m-%d')
|
||
date_list.append(date_str)
|
||
|
||
# 查询当天的构建数据
|
||
day_start = datetime.combine(current_date, datetime.min.time())
|
||
day_end = datetime.combine(current_date, datetime.max.time())
|
||
|
||
# 成功构建数
|
||
success_count = BuildHistory.objects.filter(
|
||
create_time__gte=day_start,
|
||
create_time__lte=day_end,
|
||
status='success'
|
||
).count()
|
||
|
||
# 失败构建数
|
||
failed_count = BuildHistory.objects.filter(
|
||
create_time__gte=day_start,
|
||
create_time__lte=day_end,
|
||
status='failed'
|
||
).count()
|
||
|
||
success_data.append(success_count)
|
||
failed_data.append(failed_count)
|
||
|
||
current_date += timedelta(days=1)
|
||
|
||
return JsonResponse({
|
||
'code': 200,
|
||
'message': '获取构建任务趋势数据成功',
|
||
'data': {
|
||
'dates': date_list,
|
||
'success': success_data,
|
||
'failed': failed_data
|
||
}
|
||
})
|
||
except Exception as e:
|
||
logger.error(f'获取构建任务趋势数据失败: {str(e)}', exc_info=True)
|
||
return JsonResponse({
|
||
'code': 500,
|
||
'message': f'服务器错误: {str(e)}'
|
||
})
|
||
|
||
|
||
@method_decorator(csrf_exempt, name='dispatch')
|
||
class BuildDetailView(View):
|
||
"""构建详细数据接口"""
|
||
|
||
def get(self, request):
|
||
"""获取指定日期的构建详细数据"""
|
||
try:
|
||
# 获取日期参数
|
||
date_str = request.GET.get('date')
|
||
if not date_str:
|
||
return JsonResponse({
|
||
'code': 400,
|
||
'message': '日期参数不能为空'
|
||
})
|
||
|
||
# 解析日期
|
||
try:
|
||
date = datetime.strptime(date_str, '%Y-%m-%d')
|
||
day_start = datetime(date.year, date.month, date.day, 0, 0, 0)
|
||
day_end = datetime(date.year, date.month, date.day, 23, 59, 59)
|
||
except ValueError:
|
||
return JsonResponse({
|
||
'code': 400,
|
||
'message': '日期格式不正确，应为YYYY-MM-DD'
|
||
})
|
||
|
||
# 查询当天的构建历史
|
||
builds = BuildHistory.objects.filter(
|
||
create_time__gte=day_start,
|
||
create_time__lte=day_end
|
||
).select_related('task', 'operator').order_by('-create_time')
|
||
|
||
build_list = []
|
||
for build in builds:
|
||
# 计算构建耗时
|
||
duration = '未完成'
|
||
if build.build_time and 'total_duration' in build.build_time:
|
||
total_seconds = int(build.build_time.get('total_duration', 0))
|
||
minutes = total_seconds // 60
|
||
seconds = total_seconds % 60
|
||
duration = f"{minutes}分{seconds}秒"
|
||
|
||
build_list.append({
|
||
'id': build.history_id,
|
||
'build_number': build.build_number,
|
||
'task_name': build.task.name,
|
||
'status': build.status,
|
||
'branch': build.branch,
|
||
'version': build.version,
|
||
'start_time': build.build_time.get('start_time') if build.build_time else build.create_time.strftime('%Y-%m-%d %H:%M:%S'),
|
||
'duration': duration,
|
||
'operator': build.operator.name if build.operator else '系统'
|
||
})
|
||
|
||
return JsonResponse({
|
||
'code': 200,
|
||
'message': '获取构建详细数据成功',
|
||
'data': build_list
|
||
})
|
||
except Exception as e:
|
||
logger.error(f'获取构建详细数据失败: {str(e)}', exc_info=True)
|
||
return JsonResponse({
|
||
'code': 500,
|
||
'message': f'服务器错误: {str(e)}'
|
||
})
|
||
|
||
|
||
@method_decorator(csrf_exempt, name='dispatch')
|
||
class RecentBuildsView(View):
|
||
"""最近构建任务接口"""
|
||
|
||
def get(self, request):
|
||
"""获取最近构建任务数据"""
|
||
try:
|
||
# 获取数量参数，默认为10条
|
||
limit = int(request.GET.get('limit', 5))
|
||
|
||
# 查询最近的构建历史
|
||
recent_builds = BuildHistory.objects.select_related(
|
||
'task', 'task__environment', 'operator' # 关联环境信息
|
||
).order_by('-create_time')[:limit]
|
||
|
||
build_list = []
|
||
for build in recent_builds:
|
||
# 计算构建耗时
|
||
duration = '未完成'
|
||
if build.build_time and 'total_duration' in build.build_time:
|
||
total_seconds = int(build.build_time.get('total_duration', 0))
|
||
minutes = total_seconds // 60
|
||
seconds = total_seconds % 60
|
||
duration = f"{minutes}分 {seconds}秒"
|
||
|
||
build_list.append({
|
||
'id': build.history_id,
|
||
'build_number': build.build_number,
|
||
'task_name': build.task.name,
|
||
'status': build.status,
|
||
'branch': build.branch,
|
||
'version': build.version,
|
||
'environment': build.task.environment.name if build.task.environment else None, # 添加环境名称
|
||
'requirement': build.requirement,
|
||
'start_time': build.build_time.get('start_time') if build.build_time else build.create_time.strftime('%Y-%m-%d %H:%M:%S'),
|
||
'duration': duration,
|
||
'operator': build.operator.name if build.operator else '系统'
|
||
})
|
||
|
||
return JsonResponse({
|
||
'code': 200,
|
||
'message': '获取最近构建任务数据成功',
|
||
'data': build_list
|
||
})
|
||
except Exception as e:
|
||
logger.error(f'获取最近构建任务数据失败: {str(e)}', exc_info=True)
|
||
return JsonResponse({
|
||
'code': 500,
|
||
'message': f'服务器错误: {str(e)}'
|
||
})
|
||
|
||
|
||
@method_decorator(csrf_exempt, name='dispatch')
|
||
class ProjectDistributionView(View):
|
||
"""项目分布接口"""
|
||
|
||
def get(self, request):
|
||
"""获取项目分布数据"""
|
||
try:
|
||
# 按项目类别统计
|
||
category_stats = Project.objects.values('category').annotate(count=Count('id'))
|
||
|
||
# 格式化数据
|
||
category_data = []
|
||
for stat in category_stats:
|
||
category = stat['category'] or '未分类'
|
||
category_data.append({
|
||
'type': self._get_category_name(category),
|
||
'value': stat['count']
|
||
})
|
||
|
||
return JsonResponse({
|
||
'code': 200,
|
||
'message': '获取项目分布数据成功',
|
||
'data': category_data
|
||
})
|
||
except Exception as e:
|
||
logger.error(f'获取项目分布数据失败: {str(e)}', exc_info=True)
|
||
return JsonResponse({
|
||
'code': 500,
|
||
'message': f'服务器错误: {str(e)}'
|
||
})
|
||
|
||
def _get_category_name(self, category):
|
||
"""获取项目类别名称"""
|
||
category_map = {
|
||
'frontend': '前端项目',
|
||
'backend': '后端项目',
|
||
'mobile': '移动端项目',
|
||
'other': '其他项目'
|
||
}
|
||
return category_map.get(category, '未分类') |