42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
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() |