Upload a document

# This sample code is for reference only. Do not use it directly in a production environment.
import requests

def upload_file(pre_signed_url, file_path):
    try:
        # Set the request header.
        headers = {
            "x-bailian-extra": "Replace this with the value of the x-bailian-extra field from data.param.headers returned by the request an upload lease API call in the previous step.",
            "Content-Type": "Replace this with the value of the Content-Type field from data.param.headers returned by the request an upload lease API call in the previous step."
        }

        # Read and upload the document.
        with open(file_path, 'rb') as file:
            # The request method for uploading the document must be the same as the value of the method field in data.param returned by the request an upload lease API call in the previous step.
            response = requests.put(pre_signed_url, data=file, headers=headers)

        # Check the response status code.
        if response.status_code == 200:
            print("File uploaded successfully.")
        else:
            print(f"Failed to upload the file. ResponseCode: {response.status_code}")

    except Exception as e:
        print(f"An error occurred: {str(e)}")

if __name__ == "__main__":

    pre_signed_url_or_http_url = "Replace this with the value of the url field in data.param returned by the request an upload lease API call in the previous step."

    # The document source is local. Upload the local document to OSS.
    file_path = "Replace this with the actual local path of the document you want to upload."
    upload_file(pre_signed_url_or_http_url, file_path)

import hashlib
import os
import requests
from http import HTTPStatus

'''
API name: Upload a document
API paths: https://dashscope.aliyuncs.com/api/v2/apps/zhiwen-file/apply_upload_lease
        https://dashscope.aliyuncs.com/api/v2/apps/zhiwen-file/submit_parse_file
Environment requirements: Python >= 3.7
'''

apply_lease_url = "https://dashscope.aliyuncs.com/api/v2/apps/zhiwen-file/apply_upload_lease"
submit_file_url = "https://dashscope.aliyuncs.com/api/v2/apps/zhiwen-file/submit_parse_file"
request_headers = {
    "Authorization": os.environ.get("DASHSCOPE_API_KEY"),  # If you have not configured the environment variable, replace this with your API-KEY.
    "Content-Type": "application/json"
}

file_path = "./test.txt"
file_name = os.path.basename(file_path)
file_size = os.path.getsize(file_path)
with open(file_path, 'rb') as f:
    md5 = hashlib.md5()
    while True:
        data = f.read(4096)  # Read 4 KB of data at a time.
        if not data:
            break
        md5.update(data)
file_md5 = md5.hexdigest()

# 1. Request a lease.
response = requests.post(apply_lease_url,
                         json={
                             "fileName": file_name,
                             "sizeBytes": file_size,
                             "md5": file_md5
                         },
                         headers=request_headers)
lease_data = {}
if response.status_code == HTTPStatus.OK:
    print("1. Lease applied successfully.")
    print(response.json())
    lease_data = response.json()['data']
else:
    print(f'response={response.json()}')
    print(f'code={response.status_code}')

# 2. Upload the document.
upload_headers = {
    "X-bailian-extra": lease_data['param']['headers']['x-bailian-extra'],
    "Content-Type": lease_data['param']['headers']['Content-Type']
}
with open(file_path, 'rb') as file:
    # The request method for uploading the document must be the same as the value of the Method field in Data.Param returned by the ApplyFileUploadLease API call in the previous step.
    response = requests.put(lease_data['param']['url'], data=file, headers=upload_headers)
if response.status_code == HTTPStatus.OK:
    print("2. File uploaded successfully.")
else:
    print(f"Failed to upload the file. ResponseCode: {response.status_code}")

# 3. Submit for parsing.
response = requests.post(submit_file_url,
                         json={
                             'leaseId': lease_data['lease_id']
                         },

                         headers=request_headers)
if response.status_code == HTTPStatus.OK:
    print("3. File submit parse successfully.")
    print(response.json())
else:
    print(f'response={response.json()}')
    print(f'code={response.status_code}')