python2-udfs-api/UdfsDevTools.py at master · UlordChain/python2-udfs-api · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
# -*- coding: utf-8 -*-
import base64
import hashlib
import hmac
import os
import time
from hashlib import sha1


class UdfsDevTools():

    def GetFileMd5(self,filePath):
        if not os.path.isfile(filePath):
            return
        myhash = hashlib.md5()
        f = open(filePath, 'rb')
        while True:
            b = f.read(8096)
            if not b:
                break
            myhash.update(b)
        f.close()
        return myhash.hexdigest()

    #expireSec:指定毫秒后过期时间
    #secretKey:秘钥,用于udfs访问,从udfs官方获取
    #uosAccount:uos账号
    #filePath:需要上传的文件路径
    #udfsCallbackUrl:回调url,udfs上传完毕后回调该url,将返回（文件id，哈希值，文件size），该参数可为空
    def getUploadUrl(self,expireSec, uosAccount,secretKey,  filePath, udfsCallbackUrl):
        fileName = os.path.basename(filePath);
        fileSize = str(os.path.getsize(filePath))
        md5 = self.GetFileMd5(filePath);
        expireTime = int(time.time())
        expireTime += expireSec;
        expireTime=str(expireTime)
        if udfsCallbackUrl == None or udfsCallbackUrl == "":
            json = "{\"ver\":0,\"expired\":" + expireTime + ",\"callback_url\":\"\",\"callback_body\":\"\",\"ext\":{\"file_name\":\"" + fileName + "\",\"size\":" + fileSize + ",\"md5\":\"" + md5 + "\"}}"
        else:
            json = "{\"ver\":0,\"expired\":" + expireTime + ",\"callback_url\":\"" + udfsCallbackUrl + "\",\"callback_body\":\"{\\\"file_size\\\":\\\"$(size)\\\",\\\"hash\\\":\\\"$(hash)\\\",\\\"file_id\\\":0}\",\"ext\":{\"file_name\":\"" + fileName + "\",\"size\":" + fileSize + ",\"md5\":\"" + md5 + "\"}}"
        try :
            textByte = bytes(json)
            encodedPolicy = base64.urlsafe_b64encode(textByte)
            my_sign = hmac.new(secretKey, encodedPolicy, sha1).digest()
            encodeSign = base64.urlsafe_b64encode(my_sign)
        except Exception as ex:
            raise Exception("构建上传链接出错" + ex.toString())
        token = uosAccount + ":" + encodeSign + ":" + encodedPolicy
        uploadloadUrl = 'http://api.udfs.one:15001/api/v0/add' + "?token=" + token
        return uploadloadUrl

    #uosAccount:uos账号
    #secretKey:秘钥,用于udfs访问,从udfs官方获取
    #expireSec:指定毫秒后过期时间
    #hashvalue:要下载的对象哈希值
    def getDownloadUrl(self,uosAccount,secretKey,expireSec,hashvalue):
        expireTime = int(time.time())
        expireTime += expireSec;
        json = "{\"ver\":0,\"expired\": "+str(expireTime)+",\"ext\":{}}"
        try :
            textByte = bytes(json)
            encodedPolicy = base64.urlsafe_b64encode(textByte)
            my_sign = hmac.new(secretKey, encodedPolicy, sha1).digest()
            encodeSign = base64.urlsafe_b64encode(my_sign)
        except Exception as ex:
            raise Exception("构建下载链接出错" + ex.toString())
        token = uosAccount + ":" + encodeSign + ":" + encodedPolicy
        downloadUrl = 'http://api.udfs.one:15001/api/v0/cat/' + hashvalue + "?token=" + token
        return downloadUrl;

    def deletefile(self,uosAccount,secretKey,expireSec,hashvalue):
        expireTime = int(time.time())
        expireTime += expireSec;
        json = "{\"ver\":0,\"expired\": "+str(expireTime)+",\"ext\":{}}"
        try :
            textByte = bytes(json)
            encodedPolicy = base64.urlsafe_b64encode(textByte)
            my_sign = hmac.new(secretKey, encodedPolicy, sha1).digest()
            encodeSign = base64.urlsafe_b64encode(my_sign)
        except Exception as ex:
            raise Exception("构建下载链接出错" + ex.toString())
        token = uosAccount + ":" + encodeSign + ":" + encodedPolicy
        deleteUrl = 'http://api.udfs.one:15001/api/v0/delete/' + hashvalue + "?token=" + token
        return deleteUrl;