From 21fc193a1319ad3135b3f48a622bf457478e4940 Mon Sep 17 00:00:00 2001 From: Zixiao Wang Date: Wed, 11 Jun 2025 15:53:02 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E8=80=81=E6=8E=A5=E5=8F=A3?= =?UTF-8?q?=E7=9A=84=E9=80=82=E9=85=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- datasets/views.py | 283 ++++++++++++++++++++++++---------------------- 1 file changed, 148 insertions(+), 135 deletions(-) diff --git a/datasets/views.py b/datasets/views.py index 90c5ec3..9839659 100644 --- a/datasets/views.py +++ b/datasets/views.py @@ -292,25 +292,160 @@ def get_match_list(request): -####### 老接口 +@csrf_exempt # 仅在开发时使用，生产环境中请使用更安全的方式 +def create_project(request): + project_name = request.POST.get('name') + username = request.POST.get('user') + task_type = request.POST.get('taskType') + description = request.POST.get('description') + try: + user = User.objects.get(username=username) + except User.DoesNotExist: + return JsonResponse({'success': False, 'message': '用户不存在'}) + project = Project(name=project_name, user=user.username, task_type=task_type, description=description) + project.save() + return JsonResponse({'success': True, 'message': '项目已创建'}) + +def get_user_projects(request): + username = request.GET.get('user') + try: + user = User.objects.get(username=username) + except User.DoesNotExist: + return JsonResponse({'projects': []}) + projects = Project.objects.filter(user=user.username) + return JsonResponse({'projects': list(projects.values())}) + +@csrf_exempt # 仅在开发时使用，生产环境中请使用更安全的方式 +def delete_project(request): + username = request.GET.get('user') + project_name = request.GET.get('projectName') + try: + user = User.objects.get(username=username) + except User.DoesNotExist: + return JsonResponse({'success': False}) + project_deleted, _ = Project.objects.filter(user=user.username, name=project_name).delete() + if project_deleted > 0: + return JsonResponse({'success': True}) + else: + return JsonResponse({'success': False}) + +@csrf_exempt # 仅在开发时使用，生产环境中请使用更安全的方式 +def delete_dataset(request): + username = request.GET.get('user') + dataset_name = request.GET.get('datasetName') + try: + user = User.objects.get(username=username) + except User.DoesNotExist: + return JsonResponse({'success': False}) + project_deleted, _ = Dataset.objects.filter(user=user, name=dataset_name).delete() + + delete_object_name = user.username + '/' + dataset_name + + ## 删除minio中的所有数据 + future1 = executor.submit(delete_from_minio, delete_object_name) + + ## 删除临时文件夹的所有数据 + curr_temp_dir = os.path.join(os.path.join(TEMP_ROOT_DIR, user.username), dataset_name) + + future2 = executor.submit(delete_temp_dir, curr_temp_dir) + if project_deleted > 0: + return JsonResponse({'success': True}) + else: + return JsonResponse({'success': False}) + +@csrf_exempt # 仅在开发时使用，生产环境中请使用更安全的方式 +def get_project(request): + username = request.GET.get('user') + project_name = request.GET.get('projectName') + try: + user = User.objects.get(username=username) + except User.DoesNotExist: + return JsonResponse({'success': False, 'project': None}) + project = Project.objects.filter(user=user.username, name=project_name) + if project.exists(): + return JsonResponse({'success': True, 'project': list(project.values())[0]}) + else: + return JsonResponse({'success': False, 'project': None}) + +@csrf_exempt # 仅在开发时使用，生产环境中请使用更安全的方式 +def get_dataset(request): + username = request.GET.get('user') + dataset_name = request.GET.get('datasetName') + try: + user = User.objects.get(username=username) + except User.DoesNotExist: + return JsonResponse({'success': False, 'dataset': None}) + dataset = Dataset.objects.filter(user=user, name=dataset_name) + if dataset.exists(): + return JsonResponse({'success': True, 'dataset': list(dataset.values())[0]}) + else: + return JsonResponse({'success': False, 'dataset': None}) + +@csrf_exempt # 仅在开发时使用，生产环境中请使用更安全的方式 +def get_minio_links(request): + username = request.GET.get('user') + dataset_name = request.GET.get('datasetName') + next_image = request.GET.get('nextImage') + page_size = request.GET.get('pageSize') + if next_image != '': + next_image = os.path.relpath(next_image, f"http://{LOCAL_IP}:9000/{BUCKET_NAME}") + task_type = request.GET.get('taskType') + try: + user = User.objects.get(username=username) + except User.DoesNotExist: + return JsonResponse({'success': False, 'links': []}) + ## 生成object_name路径，用于访问minio + object_name = user.username + '/' + dataset_name + '/' + task_type + '/images' + http_links = get_dataset_link(object_name, next_image, page_size) + return JsonResponse({'success': True, 'links': http_links}) + +@csrf_exempt +def get_dataset_is_upload(request): + if request.method == 'GET': + username = request.GET.get('user') + dataset_name = request.GET.get('datasetName') + + # 检查用户和数据集名称是否提供 + if not username or not dataset_name: + return JsonResponse({'success': False, 'message': '用户或数据集名称未提供'}) + + # 查询数据库获取数据集 + try: + user = User.objects.get(username=username) + except User.DoesNotExist: + return JsonResponse({'success': False, 'message': '用户未找到'}) + dataset = Dataset.objects.filter(user=user, name=dataset_name).first() + + if dataset is not None: + return JsonResponse({'success': True, 'is_upload': dataset.is_upload}) + else: + return JsonResponse({'success': False, 'message': '数据集未找到'}) + + return JsonResponse({'success': False, 'message': '请求方法错误'}) + @csrf_exempt def upload(request): if request.method == 'POST': # 检查文件是否存在 if 'file' in request.FILES: dataset_name = request.POST.get('name') - user = request.POST.get('user') + username = request.POST.get('user') task_type = request.POST.get('taskType') size = request.POST.get('size') description = request.POST.get('description') uploaded_file = request.FILES.get('file') + try: + user = User.objects.get(username=username) + except User.DoesNotExist: + return JsonResponse({'success': False, 'message': f'用户 {username} 不存在'}) + ## 检查数据库判断该数据集名称是否重复 if Dataset.objects.filter(name=dataset_name, user=user).exists(): return JsonResponse({'success': False, 'message': f'数据集 {dataset_name} 名称已存在'}) ## 创建解压数据集的临时目录 - temp_dir = TEMP_ROOT_DIR + '/' + user + '/' + dataset_name + temp_dir = TEMP_ROOT_DIR + '/' + user.username + '/' + dataset_name # 如果目录已存在且不为空，则删除该目录 if os.path.exists(temp_dir) and os.listdir(temp_dir): @@ -366,7 +501,7 @@ def upload(request): print(f"数据集 {dataset_name} 结构检查失败") ## 删除临时文件夹的所有数据 - curr_temp_dir = os.path.join(os.path.join(TEMP_ROOT_DIR, user), dataset_name) + curr_temp_dir = os.path.join(os.path.join(TEMP_ROOT_DIR, user.username), dataset_name) delete_temp_dir(curr_temp_dir) return JsonResponse({'success': False, 'message': f'数据集 {dataset_name} 结构不符合要求: {message}'}) @@ -376,7 +511,7 @@ def upload(request): user=user, # 这里需要替换为实际的用户信息 task_type=task_type, size=size, - number=0, + # number=0, description=description, categories = categories ) @@ -392,139 +527,17 @@ def upload(request): return JsonResponse({'success': False, 'message': '请求错误'}) +@csrf_exempt def get_user_datasets(request): - user = request.GET.get('user') + username = request.GET.get('user') + if not username: + return JsonResponse({'code': 400, 'message': '缺少user参数', 'data': {}}) + try: + user = User.objects.get(username=username) + except User.DoesNotExist: + return JsonResponse({'code': 400, 'message': '用户不存在', 'data': {}}) datasets = Dataset.objects.filter(user=user) return JsonResponse({'datasets': list(datasets.values())}) -@csrf_exempt # 仅在开发时使用，生产环境中请使用更安全的方式 -def create_project(request): - project_name = request.POST.get('name') - user = request.POST.get('user') - task_type = request.POST.get('taskType') - description = request.POST.get('description') - project = Project(name=project_name, user=user, task_type=task_type, description=description) - project.save() - return JsonResponse({'success': True, 'message': '项目已创建'}) - -def get_user_projects(request): - user = request.GET.get('user') - projects = Project.objects.filter(user=user) - return JsonResponse({'projects': list(projects.values())}) - -@csrf_exempt # 仅在开发时使用，生产环境中请使用更安全的方式 -def delete_project(request): - user = request.GET.get('user') - project_name = request.GET.get('projectName') - project_deleted, _ = Project.objects.filter(user=user, name=project_name).delete() - if project_deleted > 0: - return JsonResponse({'success': True}) - else: - return JsonResponse({'success': False}) - -@csrf_exempt # 仅在开发时使用，生产环境中请使用更安全的方式 -def delete_dataset(request): - user = request.GET.get('user') - dataset_name = request.GET.get('datasetName') - project_deleted, _ = Dataset.objects.filter(user=user, name=dataset_name).delete() - - - delete_object_name = user + '/' + dataset_name - - ## 删除minio中的所有数据 - future1 = executor.submit(delete_from_minio, delete_object_name) - - ## 删除临时文件夹的所有数据 - curr_temp_dir = os.path.join(os.path.join(TEMP_ROOT_DIR, user), dataset_name) - - future2 = executor.submit(delete_temp_dir, curr_temp_dir) - if project_deleted > 0: - return JsonResponse({'success': True}) - else: - return JsonResponse({'success': False}) - -@csrf_exempt # 仅在开发时使用，生产环境中请使用更安全的方式 -def get_project(request): - user = request.GET.get('user') - project_name = request.GET.get('projectName') - project = Project.objects.filter(user=user, name=project_name) - if project is not None: - return JsonResponse({'success': True, 'project': list(project.values())[0]}) - else: - return JsonResponse({'success': False, 'project': None}) - -@csrf_exempt # 仅在开发时使用，生产环境中请使用更安全的方式 -def get_dataset(request): - user = request.GET.get('user') - dataset_name = request.GET.get('datasetName') - dataset = Dataset.objects.filter(user=user, name=dataset_name) - if dataset is not None: - return JsonResponse({'success': True, 'dataset': list(dataset.values())[0]}) - else: - return JsonResponse({'success': False, 'dataset': None}) - -@csrf_exempt # 仅在开发时使用，生产环境中请使用更安全的方式 -def get_minio_links(request): - user = request.GET.get('user') - dataset_name = request.GET.get('datasetName') - next_image = request.GET.get('nextImage') - page_size = request.GET.get('pageSize') - if next_image != '': - next_image = os.path.relpath(next_image, f"http://{LOCAL_IP}:9000/{BUCKET_NAME}") - task_type = request.GET.get('taskType') - - - ## 生成object_name路径，用于访问minio - object_name = user + '/' + dataset_name + '/' + task_type + '/images' - http_links = get_dataset_link(object_name, next_image, page_size) - return JsonResponse({'success': True, 'links': http_links}) - -@csrf_exempt -def get_dataset_is_upload(request): - if request.method == 'GET': - user = request.GET.get('user') - dataset_name = request.GET.get('datasetName') - - # 检查用户和数据集名称是否提供 - if not user or not dataset_name: - return JsonResponse({'success': False, 'message': '用户或数据集名称未提供'}) - - # 查询数据库获取数据集 - dataset = Dataset.objects.filter(user=user, name=dataset_name).first() - - if dataset is not None: - return JsonResponse({'success': True, 'is_upload': dataset.is_upload}) - else: - return JsonResponse({'success': False, 'message': '数据集未找到'}) - - return JsonResponse({'success': False, 'message': '请求方法错误'}) - - - - """根据传入的product_id和creator_id插入一条CreatorProductMatch记录，统一返回格式""" - if request.method == 'POST': - try: - data = json.loads(request.body.decode('utf-8')) - except Exception: - return JsonResponse({'code': 400, 'message': '请求体不是有效的JSON', 'data': {}}, status=400) - product_id = data.get('product_id') - creator_id = data.get('creator_id') - if not product_id or not creator_id: - return JsonResponse({'code': 400, 'message': '缺少product_id或creator_id参数', 'data': {}}, status=400) - try: - product = Product.objects.get(id=product_id) - except Product.DoesNotExist: - return JsonResponse({'code': 400, 'message': '产品不存在', 'data': {}}, status=400) - try: - creator = CreatorProfile.objects.get(id=creator_id) - except CreatorProfile.DoesNotExist: - return JsonResponse({'code': 400, 'message': '达人不存在', 'data': {}}, status=400) - # 检查是否已存在 - if CreatorProductMatch.objects.filter(product=product, creator=creator).exists(): - return JsonResponse({'code': 400, 'message': '该达人和产品的匹配已存在', 'data': {}}, status=400) - match = CreatorProductMatch.objects.create(product=product, creator=creator) - return JsonResponse({'code': 200, 'message': '匹配关系创建成功', 'data': {'match_id': match.id}}) - return JsonResponse({'code': 400, 'message': '请求方法错误', 'data': {}}, status=400) -