POST -Add Label

• url: The request address.
• headers: The HTTP headers of the request.
• data: The metadata information sent, example format:

  metadata = {
      "app": "appA",
      "queryID": str(uuid.uuid4()),
      "update": True,
      "sendBack": {"type": "response"},
      "labels": {
          "employID": "12340017",
      }
  }

• files: Local files to be operated on.

{   
    "app": "875a2442-2fe6-49a5-abf7-00fbd727e99e",
    "queryID": "8a0aa812-0f80-4d25-8e3f-60583bac8b96",
    "update": True,
    "labels": {
        "test_key": "tt_value",
        "id": "8888"
    },
    "sendBack": {
        "type": "response"
    }
}

{
    "labels": {
        "test_key": "tt_value",
        "id": "8888"
    },
    "app": "875a2442-2fe6-49a5-abf7-00fbd727e99e",
    "update": True,
    "queryID": "8a0aa812-0f80-4d25-8e3f-60583bac8b96",
    "callback_url": "http://172.11.9.1:8000/callback",
    "sendBack": {
        "type": "s3",
        "s3Bucket": "leo",
        "s3Region": "",  # Region name when using S3. If running on EC2 and using AWS IAM role, the following three items can be omitted
        "accessKey": "admin",
        "secretKey": "admin",
        "endpointUrl": "http://172.20.162.160:9000",
    },
    "uploadtype": "file"
}

# -*- coding: utf-8 -*-
from requests.packages.urllib3.exceptions import InsecureRequestWarning
from ucwi_config import UCWIConfig
from ucwi_auth import get_headers
import requests
import json
import os
import uuid
import time
from requests_toolbelt.multipart import decoder

requests.packages.urllib3.disable_warnings(InsecureRequestWarning)

api = "/skg/v1/dlp/label/custom/sync"
url = "{0}{1}".format(UCWIConfig.base_url, api)
file_path="CustomLabelTest.docx.RET.RET"

metadata = {
    "app": "appA",
    "queryID": str(uuid.uuid4()),
    "update": True,
    "sendBack": {"type": "response"},
    "labels": {
        "employID": "12340017",
        "Student-name": "peter",
        "BBB-XXX": "2222222",
        "EEE-XXX": "555555",
    }
}

headers = get_headers()
data = {"metadata": json.dumps(metadata)}
fd = open(file_path, 'rb')
files = {
    "request": fd
}
response = requests.post(url, headers=headers, data=data, files=files, verify=False)
fd.close()

if response.status_code != 200:
    print("Bad request, response code:", response.status_code)
    print(response.text)
else:
    if response.headers['Content-Type'] == 'application/json':
        info = response.json()
        print(info)
    elif response.headers['Content-Type'].startswith('multipart/form-data'):
        multipart_data = decoder.MultipartDecoder.from_response(response)
        for part in multipart_data.parts:
            content = part.content
            content_type = part.headers[b'content-type']
            content_disposition = part.headers[b'content-disposition']

            if content_type == b"application/json":
                info = json.loads(content)
                print("multipart info:{}".format(info))
            elif content_type == b"application/octet-stream":
                cd_info = [x.strip()
                           for x in content_disposition.decode('utf-8').split(';')]
                print("multipart xxx {}".format(cd_info))
                for item in cd_info[1:]:
                    index = item.find('=')
                    key, value = item[:index], item[index + 1:]
                    if key == 'filename':
                        filename = value.strip('"') + ".RET"
                        with open(filename, 'wb') as f:
                            f.write(content)