diff --git a/cope2n-api/fwd_api/api/ctel_view.py b/cope2n-api/fwd_api/api/ctel_view.py index ceeb64e..dd8a5fc 100755 --- a/cope2n-api/fwd_api/api/ctel_view.py +++ b/cope2n-api/fwd_api/api/ctel_view.py @@ -122,6 +122,9 @@ class CtelViewSet(viewsets.ViewSet): 'redemption_ID': { 'type': 'string' }, + 'is_test_request': { + 'type': 'boolean', + }, }, 'required': {'imei_files'} } @@ -144,7 +147,7 @@ class CtelViewSet(viewsets.ViewSet): "invoice": invoice_file_objs } total_page = len(files.keys()) - + is_test_request = validated_data.get("is_test_request", False) rq_id = provider_code + "_" + datetime.now().strftime("%Y%m%d%H%M%S") + "_" + uuid.uuid4().hex p_type = validated_data['type'] new_request: SubscriptionRequest = SubscriptionRequest(pages=total_page, @@ -152,7 +155,8 @@ class CtelViewSet(viewsets.ViewSet): process_type=p_type, status=1, request_id=rq_id, provider_code=provider_code, subscription=sub, - redemption_id=validated_data["redemption_ID"]) + redemption_id=validated_data["redemption_ID"], + is_test_request=is_test_request) new_request.save() count = 0 compact_files = [] @@ -194,6 +198,9 @@ class CtelViewSet(viewsets.ViewSet): 'redemption_ID': { 'type': 'string' }, + 'is_test_request': { + 'type': 'boolean', + }, }, 'required': {'imei_files'} } @@ -216,7 +223,7 @@ class CtelViewSet(viewsets.ViewSet): "invoice": invoice_file_objs } rq_id = provider_code + "_" + datetime.now().strftime("%Y%m%d%H%M%S") + "_" + uuid.uuid4().hex - + is_test_request = validated_data.get("is_test_request", False) count = 0 doc_files_with_type = [] for doc_type, doc_files in files.items(): @@ -236,7 +243,9 @@ class CtelViewSet(viewsets.ViewSet): pages_left=total_page, process_type=p_type, status=1, request_id=rq_id, provider_code=provider_code, - subscription=sub) + subscription=sub, + redemption_id=validated_data["redemption_ID"], + is_test_request=is_test_request) new_request.save() # Run file processing in a pool of 2 threads. TODO: Convert to Celery worker when possible