views.py 3.08 KB
from django.shortcuts import render
from webargs import fields, validate
from webargs.djangoparser import use_args, parser
from common.mixins import GenericView
from common import response
from .models import UploadDocRecords

# Create your views here.


# restframework将request.body封装至request.data, webargs从request.data中获取参数
@parser.location_loader("data")
def load_data(request, schema):
    return request.data


application_data_args = {'applicationId': fields.Str(required=True, validate=validate.Length(max=64))}

applicant_data_args = {
    'mainApplicantName': fields.Str(required=True, validate=validate.Length(max=16)),
    'coApplicantName': fields.Str(required=True, validate=validate.Length(max=16)),
    'guarantor1Name': fields.Str(required=True, validate=validate.Length(max=16)),
    'guarantor2Name': fields.Str(required=True, validate=validate.Length(max=16)),
}

document_args = {
    'documentName': fields.Str(required=True, validate=validate.Length(max=255)),
    'documentScheme': fields.Str(required=True, validate=validate.Length(max=64)),
    'businessType': fields.Str(required=True, validate=validate.Length(max=64)),
    'uploadFinishTime': fields.DateTime(required=True),
    'dataSource': fields.Str(required=True, validate=validate.Length(max=64)),
    'metadataVersionId': fields.Str(required=True, validate=validate.Length(max=64)),
}

doc_upload_args = {
    'applicationData': fields.Nested(application_data_args, required=True),
    'applicantData': fields.Nested(applicant_data_args, required=True),
    'document': fields.Nested(document_args, required=True),
}


class DocView(GenericView):
    permission_classes = []

    # 创建模型
    @use_args(doc_upload_args, location='data')
    def post(self, request, args):
        application_data = args.get('applicationData')
        applicant_data = args.get('applicantData')
        document = args.get('document')
        UploadDocRecords.objects.create(
            application_id=application_data.get('applicationId'),
            main_applicant=applicant_data.get('mainApplicantName'),
            co_applicant=applicant_data.get('coApplicantName'),
            guarantor_1=applicant_data.get('guarantor1Name'),
            guarantor_2=applicant_data.get('guarantor2Name'),
            document_name=document.get('documentName'),
            document_scheme=document.get('documentScheme'),
            business_type=document.get('businessType'),
            data_source=document.get('dataSource'),
            metadata_version_id=document.get('metadataVersionId'),
            upload_finish_time=document.get('uploadFinishTime'),
        )
        self.running_log.info('[doc upload success] [args={0}]'.format(args))
        return response.ok()

    post.openapi_doc = '''
    tags: [doc]
    summary: POS系统上传文件信息
    consumes: [application/json]
    produces: [application/json]
    parameters:
    - in: body
      name: body
      required: true
      schema:
        $ref: "#/definitions/Doc"

    responses:
      200:
        description: ok
        schema:
          $ref: '#/definitions/ApiResponse'
    '''