'''
本模块用于定义云端读写的相关操作，包括上传、下载、删除等
'''
import oss2
import os
import json
from oss2.credentials import EnvironmentVariableCredentialsProvider
# 云端存储的文件名称都是shixun_id.json,并且要保证都是有答案的.
# 创建Bucket对象，所有Object相关的接口都可以通过Bucket对象来进行
# 判断文件是否存在


# 这些key和secret都是我自己的，如果需要使用，需要自己去阿里云申请捏(云工开物神中神)
# 就一个小40G的OSS存储，里面也没啥重要数据，就是用来存储答案的，hack了也没啥用
access_key_id='LTAI5t927vdUFZa9NRnWfrL3'
access_key_secret='FbXoJUqe545eZhWFvADvGcFwatsGAx'
bucket_name="tasks-jsons"
endpoint="oss-cn-shenzhen.aliyuncs.com"

bucket = oss2.Bucket(oss2.Auth(access_key_id, access_key_secret), endpoint, bucket_name)
def is_exist(name:str) -> bool:
    return bucket.object_exists(name)

# 如果文件存在，则下载到本地并命名为shixun_id_answer
def download(name):
    # 将云端文件下载到本地并命名为shixun_id.json
    bucket.get_object_to_file(name, name)
# 上传函数，用于获得答案后上传到云端，此步骤在获得答案后调用
# (to do)如果用户将答案认证为正确，则将本地json中的verified参数改为True后再上传并覆盖云端文件
def upload(name):
    # 上传前先检验文件中的每个键对应的值的answer键是否存在，以及其对应的值是否为空，如果为空，则不上传
    with open(name,'r',encoding='utf-8') as f:
        data = json.load(f)
    for i,j in data.items():
        if 'answer' not in j.keys() or j['answer'] == '':
            # 断言查询结果并抛出异常
            assert False,'答案为空，不予上传'
    # 再检查文件是否存在，如果不存在，则上传
    if not is_exist(name):
        bucket.put_object_from_file(name, name)

def delete(name):
    # 先检查文件是否存在，如果存在，则删除
    if is_exist(name):
        bucket.delete_object(name)
if __name__ == '__main__':
    print('测试用')
    # 测试用
    # print(is_exist('18503.json'))
    #download('18503.json')
    #检查云端所有文件
    print([i for i in bucket.list_objects()])