|
| 1 | +from http import HTTPStatus |
| 2 | + |
| 3 | +from rest_framework import status |
| 4 | +from rest_framework.parsers import MultiPartParser |
| 5 | +from rest_framework.request import Request |
| 6 | +from rest_framework.test import APIRequestFactory |
| 7 | + |
| 8 | +from sds_gateway.api_methods.views.capture_endpoints import CaptureViewSet |
| 9 | +from sds_gateway.api_methods.views.file_endpoints import CheckFileContentsExistView |
| 10 | +from sds_gateway.api_methods.views.file_endpoints import FileViewSet |
| 11 | + |
| 12 | + |
| 13 | +def upload_file_helper_simple(request, file_data): |
| 14 | + """Upload a single file using FileViewSet.create. |
| 15 | +
|
| 16 | + file_data should contain all required fields: name, directory, file, |
| 17 | + media_type, etc. Returns ([response], []) for success, ([], [error]) for |
| 18 | + error, and handles 409 as a warning. |
| 19 | + """ |
| 20 | + factory = APIRequestFactory() |
| 21 | + django_request = factory.post( |
| 22 | + request.path, |
| 23 | + file_data, |
| 24 | + format="multipart", |
| 25 | + ) |
| 26 | + django_request.user = request.user |
| 27 | + drf_request = Request(django_request, parsers=[MultiPartParser()]) |
| 28 | + drf_request.user = request.user |
| 29 | + view = FileViewSet() |
| 30 | + view.request = drf_request |
| 31 | + view.action = "create" |
| 32 | + view.format_kwarg = None |
| 33 | + view.args = () |
| 34 | + view.kwargs = {} |
| 35 | + try: |
| 36 | + response = view.create(drf_request) |
| 37 | + except (ValueError, TypeError, AttributeError, KeyError) as e: |
| 38 | + return [], [f"Data validation error: {e}"] |
| 39 | + else: |
| 40 | + responses = [] |
| 41 | + errors = [] |
| 42 | + |
| 43 | + if not hasattr(response, "status_code"): |
| 44 | + errors.append(getattr(response, "data", str(response))) |
| 45 | + else: |
| 46 | + http_status = HTTPStatus(response.status_code) |
| 47 | + response_data = getattr(response, "data", str(response)) |
| 48 | + |
| 49 | + if http_status.is_success: |
| 50 | + responses.append(response) |
| 51 | + elif response.status_code == status.HTTP_409_CONFLICT: |
| 52 | + # Already exists, treat as warning |
| 53 | + errors.append(response_data) |
| 54 | + elif http_status.is_server_error: |
| 55 | + # Handle 500 and other server errors |
| 56 | + errors.append("Internal server error") |
| 57 | + elif http_status.is_client_error: |
| 58 | + # Handle 4xx client errors |
| 59 | + errors.append(f"Client error ({response.status_code}): {response_data}") |
| 60 | + else: |
| 61 | + # Handle any other status codes |
| 62 | + errors.append(response_data) |
| 63 | + |
| 64 | + return responses, errors |
| 65 | + |
| 66 | + |
| 67 | +# TODO: Use this helper method when implementing the file upload mode multiplexer. |
| 68 | +def check_file_contents_exist_helper(request, check_data): |
| 69 | + """Call the post method of CheckFileContentsExistView with the given data. |
| 70 | +
|
| 71 | + check_data should contain the required fields: directory, name, sum_blake3, |
| 72 | + etc. |
| 73 | + """ |
| 74 | + factory = APIRequestFactory() |
| 75 | + django_request = factory.post( |
| 76 | + request.path, # or a specific path for the check endpoint |
| 77 | + check_data, |
| 78 | + format="multipart", |
| 79 | + ) |
| 80 | + django_request.user = request.user |
| 81 | + drf_request = Request(django_request, parsers=[MultiPartParser()]) |
| 82 | + drf_request.user = request.user |
| 83 | + view = CheckFileContentsExistView() |
| 84 | + view.request = drf_request |
| 85 | + view.action = None |
| 86 | + view.format_kwarg = None |
| 87 | + view.args = () |
| 88 | + view.kwargs = {} |
| 89 | + return view.post(drf_request) |
| 90 | + |
| 91 | + |
| 92 | +def create_capture_helper_simple(request, capture_data): |
| 93 | + """Create a capture using CaptureViewSet.create. |
| 94 | +
|
| 95 | + capture_data should contain all required fields for capture creation: |
| 96 | + owner, top_level_dir, capture_type, channel, index_name, etc. |
| 97 | + Returns ([response], []) for success, ([], [error]) for error, and handles |
| 98 | + 409 as a warning. |
| 99 | + """ |
| 100 | + factory = APIRequestFactory() |
| 101 | + django_request = factory.post( |
| 102 | + request.path, |
| 103 | + capture_data, |
| 104 | + format="multipart", |
| 105 | + ) |
| 106 | + django_request.user = request.user |
| 107 | + drf_request = Request(django_request, parsers=[MultiPartParser()]) |
| 108 | + drf_request.user = request.user |
| 109 | + view = CaptureViewSet() |
| 110 | + view.request = drf_request |
| 111 | + view.action = "create" |
| 112 | + view.format_kwarg = None |
| 113 | + view.args = () |
| 114 | + view.kwargs = {} |
| 115 | + # Set the context for the serializer |
| 116 | + view.get_serializer_context = lambda: {"request_user": request.user} |
| 117 | + try: |
| 118 | + response = view.create(drf_request) |
| 119 | + except (ValueError, TypeError, AttributeError, KeyError) as e: |
| 120 | + return [], [f"Data validation error: {e}"] |
| 121 | + else: |
| 122 | + responses = [] |
| 123 | + errors = [] |
| 124 | + |
| 125 | + if not hasattr(response, "status_code"): |
| 126 | + errors.append(getattr(response, "data", str(response))) |
| 127 | + else: |
| 128 | + http_status = HTTPStatus(response.status_code) |
| 129 | + response_data = getattr(response, "data", str(response)) |
| 130 | + |
| 131 | + if http_status.is_success: |
| 132 | + responses.append(response) |
| 133 | + elif response.status_code == status.HTTP_409_CONFLICT: |
| 134 | + # Already exists, treat as warning |
| 135 | + errors.append(response_data) |
| 136 | + elif http_status.is_server_error: |
| 137 | + # Handle 500 and other server errors |
| 138 | + errors.append(f"Server error ({response.status_code}): {response_data}") |
| 139 | + elif http_status.is_client_error: |
| 140 | + # Handle 4xx client errors |
| 141 | + errors.append(f"Client error ({response.status_code}): {response_data}") |
| 142 | + else: |
| 143 | + # Handle any other status codes |
| 144 | + errors.append(response_data) |
| 145 | + |
| 146 | + return responses, errors |
0 commit comments