from flask import current_app
import requests

def upload_file(rb_file, file_name):
    if file_name is not None and "/" in file_name:
        file_name = file_name.split("/")[-1]
    cos_client = current_app.extensions["cos_client"]
    cos_client.put_object(
        Bucket=current_app.config["TENCENT_COS"]["bucket"],
        Body=rb_file,
        Key=file_name,
    )
    return f"https://{current_app.config['TENCENT_COS']['bucket']}.cos.{current_app.config['TENCENT_COS']['region']}.myqcloud.com/{file_name}"


def delete_file(file_name):
    if file_name is not None and "/" in file_name:
        file_name = file_name.split("/")[-1]
    cos_client = current_app.extensions["cos_client"]
    cos_client.delete_object(
        Bucket=current_app.config["TENCENT_COS"]["bucket"],
        Key=file_name,
    )
    return True

def get_text_file(file_name):
    response = requests.get(file_name)
    
    if response.status_code == 200:
        print(file_name)
        # print(f"get_text_file: {response.text}")

        return response.text
    else:
        return None
        
    return response.text
    # if file_name is not None and "/" in file_name:
    #     file_name = file_name.split("/")[-1]
    # cos_client = current_app.extensions["cos_client"]
    # print(f"get_text_file: {file_name}")
    # return cos_client.get_object(Bucket=current_app.config["TENCENT_COS"]["bucket"], Key=file_name)["Body"].read()