# -*- coding=utf-8
from qcloud_cos import CosConfig
from qcloud_cos import CosS3Client
import sys
import os
import logging

# Normally, the log level uses INFO. When troubleshooting is required, it can be changed to DEBUG, at which time the SDK will print communication information with the server.
logging.basicConfig(level=logging.INFO, stream=sys.stdout)

# 1. Set user attributes, including secret_id, secret_key, region, etc. Appid has been removed from CosConfig. Please include the Appid in the Bucket parameter. The Bucket is composed of BucketName-Appid.
secret_id = os.environ['COS_SECRET_ID']     # User's SecretId. It is recommended to use a sub-account key, as authorization follows the principle of least privilege to reduce usage risks. For obtaining sub-account keys, refer to https://www.tencentcloud.com/document/product/598/37140?from_cn_redirect=1
secret_key = os.environ['COS_SECRET_KEY']   # User's SecretKey. It is recommended to use a sub-account key, as authorization follows the principle of least privilege to reduce usage risks. For obtaining sub-account keys, refer to https://www.tencentcloud.com/document/product/598/37140?from_cn_redirect=1
region = 'ap-beijing'      # Replace with the user's region. The region to which the created bucket belongs can be viewed in the console at https://console.tencentcloud.com/cos5/bucket
                           # For a list of all regions supported by COS, refer to https://www.tencentcloud.com/document/product/436/6224?from_cn_redirect=1
token = None               # If using permanent credentials, the token does not need to be entered. If using temporary credentials, it must be provided. For guidance on generating and using temporary credentials, refer to https://www.tencentcloud.com/document/product/436/14048?from_cn_redirect=1
scheme = 'https'           # Specifies the protocol (http/https) used to access COS. The default is https, and this field is optional.

config = CosConfig(Region=region, SecretId=secret_id, SecretKey=secret_key, Token=token, Scheme=scheme)
client = CosS3Client(config)

# Method 1: Explicitly set StorageClass to DEFAULT
file_name = 'test.txt'
with open('test.txt', 'rb') as fp:
    response = client.put_object(
        Bucket='examplebucket-1250000000',  # The Bucket is composed of BucketName-APPID
        Body=fp,
        Key=file_name,
        StorageClass='DEFAULT',
        ContentType='text/html; charset=utf-8'
    )
    print(response['ETag'])
    
# Method 2: Do not set the StorageClass parameter
file_name = 'test.txt'
with open('test.txt', 'rb') as fp:
    response = client.put_object(
        Bucket='examplebucket-1250000000',  # The Bucket is composed of BucketName-APPID
        Body=fp,
        Key=file_name,
        ContentType='text/html; charset=utf-8'
    )
    print(response['ETag'])