fix: align media upload and gosedo records
All checks were successful
All checks were successful
This commit is contained in:
@@ -6,6 +6,7 @@ from apps.parsers.serializers import (
|
||||
)
|
||||
from apps.parsers.source_registry import PARSER_SOURCES, ParserSourceDescriptor
|
||||
from apps.parsers.views import (
|
||||
MEDIA_NEWS_UPLOAD_FILE_PARAM,
|
||||
RESULT_DETAIL_PARAMS,
|
||||
RESULT_LIST_PARAMS,
|
||||
UPLOAD_FILE_PARAM,
|
||||
@@ -123,7 +124,11 @@ def _upload_view(descriptor: ParserSourceDescriptor):
|
||||
f"Ручная загрузка файла для источника: {descriptor.data_scope}. "
|
||||
"Файл обрабатывается через Celery."
|
||||
),
|
||||
manual_parameters=[UPLOAD_FILE_PARAM],
|
||||
manual_parameters=[
|
||||
MEDIA_NEWS_UPLOAD_FILE_PARAM
|
||||
if descriptor.key == "media_news"
|
||||
else UPLOAD_FILE_PARAM
|
||||
],
|
||||
consumes=["multipart/form-data"],
|
||||
tags=[tag],
|
||||
responses={202: ParserRunResponseSerializer, 400: "Ошибка валидации"},
|
||||
|
||||
@@ -19,6 +19,7 @@ from apps.core.serializers import BackgroundJobListSerializer
|
||||
from apps.core.services import BackgroundJobService
|
||||
from apps.parsers import tasks
|
||||
from apps.parsers.fns_upload import FNSUploadService
|
||||
from apps.parsers.media_news import MEDIA_NEWS_MAX_BYTES
|
||||
from apps.parsers.models import (
|
||||
VACANCY_RECORD_SOURCES,
|
||||
FinancialReport,
|
||||
@@ -374,6 +375,13 @@ UPLOAD_FILE_PARAM = openapi.Parameter(
|
||||
type=openapi.TYPE_FILE,
|
||||
required=True,
|
||||
)
|
||||
MEDIA_NEWS_UPLOAD_FILE_PARAM = openapi.Parameter(
|
||||
"file",
|
||||
openapi.IN_FORM,
|
||||
description="XLSX-файл новостей СМИ размером не более 25 МиБ",
|
||||
type=openapi.TYPE_FILE,
|
||||
required=True,
|
||||
)
|
||||
RESULT_LIST_PARAMS = [
|
||||
PAGE_PARAM,
|
||||
PAGE_SIZE_PARAM,
|
||||
@@ -2793,19 +2801,30 @@ class ParserUploadView(APIView):
|
||||
)
|
||||
serializer = ParserUploadRequestSerializer(data=request.data)
|
||||
serializer.is_valid(raise_exception=True)
|
||||
if source_key == "media_news" and not serializer.validated_data[
|
||||
"file"
|
||||
].name.lower().endswith(".xlsx"):
|
||||
return api_error_response(
|
||||
[
|
||||
{
|
||||
"code": "invalid_file_type",
|
||||
"message": "Для новостей требуется XLSX",
|
||||
}
|
||||
],
|
||||
status_code=status.HTTP_400_BAD_REQUEST,
|
||||
)
|
||||
uploaded_file = serializer.validated_data["file"]
|
||||
if source_key == "media_news":
|
||||
if not uploaded_file.name.lower().endswith(".xlsx"):
|
||||
return api_error_response(
|
||||
[
|
||||
{
|
||||
"code": "invalid_file_type",
|
||||
"field": "file",
|
||||
"message": "Для новостей требуется XLSX",
|
||||
}
|
||||
],
|
||||
status_code=status.HTTP_400_BAD_REQUEST,
|
||||
)
|
||||
if uploaded_file.size > MEDIA_NEWS_MAX_BYTES:
|
||||
return api_error_response(
|
||||
[
|
||||
{
|
||||
"code": "file_too_large",
|
||||
"field": "file",
|
||||
"message": "Размер XLSX для новостей не должен превышать 25 МиБ",
|
||||
}
|
||||
],
|
||||
status_code=status.HTTP_400_BAD_REQUEST,
|
||||
)
|
||||
file_path = _save_uploaded_parser_file(uploaded_file)
|
||||
run_serializer = ParserRunRequestSerializer(data={"file_path": file_path})
|
||||
run_serializer.is_valid(raise_exception=True)
|
||||
|
||||
Reference in New Issue
Block a user