-
Notifications
You must be signed in to change notification settings - Fork 2
/
s3compatible_main.py
200 lines (176 loc) · 7.08 KB
/
s3compatible_main.py
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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
#!/usr/bin/python
#coding: utf-8
# -------------------------------------------------------------------
# 宝塔Linux面板
# -------------------------------------------------------------------
# Copyright (c) 2015-2099 宝塔软件(http://bt.cn) All rights reserved.
# -------------------------------------------------------------------
# Author: zhwen <[email protected]>
# Maintainer:hezhihong<[email protected]>
# -------------------------------------------------------------------
# -------------------------------------------------------------------
# AWSS3存储
# -------------------------------------------------------------------
from __future__ import absolute_import, print_function
import sys
from s3lib.osclient.itools import switch_environment
from s3lib.client.s3compatible import COSClient
if sys.version_info[0] == 2:
reload(sys)
sys.setdefaultencoding('utf-8')
switch_environment()
import public
# 腾讯云oss 的类
class s3compatible_main:
__client = None
__before_error_msg="ERROR: 检测到有*号，输入信息为加密信息或者信息输入不正确！请检查[" \
"SecretId/SecretKey/Bucket/Endpoint]设置是否正确!"
def __init__(self):
self.get_lib()
@property
def client(self):
if self.__client:
return self.__client
self.__client = COSClient()
return self.__client
def get_config(self, get):
return self.client.get_decrypt_config()
def set_config(self, get):
try:
secret_id = get.secret_id.strip()
secret_key = get.secret_key.strip()
bucket_name = get.Bucket.strip()
endpoint = get.Endpoint.strip()
backup_path = get.backup_path.strip()
if not backup_path:
backup_path = 'bt_backup/'
# 验证前端输入
values = [secret_id,
secret_key,
bucket_name,
endpoint,
backup_path]
for v in values:
if not v:
return public.returnMsg(False, 'API资料校验失败，请核实!')
if secret_id.find('*') !=-1 or secret_key.find('*') !=-1 or bucket_name.find('*') !=-1:return public.returnMsg(False, self.__before_error_msg)
conf = secret_id + '|' + \
secret_key + '|' + \
bucket_name + '|' + \
endpoint + '|' + \
backup_path
if self.client.set_config(conf):
if self.client.get_list():
return public.returnMsg(True, '设置成功!')
return public.returnMsg(False, 'API资料校验失败!')
except Exception as e:
return public.returnMsg(False, 'API资料校验失败，请核实!' + str(e))
# 上传文件
def upload_file(self, filename):
return self.client.resumable_upload(filename)
# 创建目录
# def create_dir(self, get):
# path = get.path + get.dirname;
# if self.client.create_dir(path):
# return public.returnMsg(True, '目录{}创建成功!'.format(path));
# else:
# return public.returnMsg(False, "创建失败！")
# 取回文件列表
def get_list(self, get):
return self.client.get_list(get.path)
# 删除文件
def delete_file(self, get):
try:
filename = get.filename
path = get.path
if path != "/":
if path[0] == "/":
path = path[1:]
if path[-1] != "/":
file_name = path + "/" + filename
else:
file_name = path + filename
if file_name[-1] == "/":
return public.returnMsg(False, "暂时不支持目录删除！")
if file_name[:1] == "/":
file_name = file_name[1:]
if self.client.delete_object(file_name):
return public.returnMsg(True, '删除成功')
return public.returnMsg(False, '文件{}删除失败, path:{}'.format(file_name,
get.path))
except:
return public.get_error_info()
def download_file(self, get):
# 连接OSS服务器
print('开始下载文件')
self.client.download_file(get.object_name,get.local_file)
print('下载成功')
def get_lib(self):
import json
info = {
"name": "S3兼容对象存储",
"type": "计划任务",
"ps": "将网站或数据库打包备份到腾讯云COS对象存储空间",
"status": 'false',
"opt": "s3compatible",
"module": "boto3",
"script": "s3compatible",
"help": "https://www.nodeloc.com",
"SecretId": "SecretId|请输入SecretId|AWS S3的SecretId",
"SecretKey": "SecretKey|请输入SecretKey|AWS S3 的SecretKey",
"region": "存储地区|请输入对象存储地区|例如 ap-chengdu",
"Bucket": "存储名称|请输入绑定的存储名称",
"check": ["/usr/lib/python2.6/site-packages/boto3/__init__.py",
"/usr/lib/python2.7/site-packages/boto3/__init__.py",
"/www/server/panel/pyenv/lib/python3.7/site-packages/boto3/__init__.py"]
}
lib = '/www/server/panel/data/libList.conf'
lib_dic = json.loads(public.readFile(lib))
for i in lib_dic:
if info['name'] in i['name']:
return True
else:
pass
lib_dic.append(info)
public.writeFile(lib, json.dumps(lib_dic))
return lib_dic
if __name__ == "__main__":
import json
import panelBackup
new_version = True if hasattr(panelBackup, "_VERSION") \
and panelBackup._VERSION >= 1.2 else False
client = COSClient()
if not new_version:
data = None
type = sys.argv[1]
if type == 'site':
if sys.argv[2] == 'ALL':
client.backupSiteAll(sys.argv[3])
else:
client.backupSite(sys.argv[2], sys.argv[3])
exit()
elif type == 'database':
print("数据库备份")
if sys.argv[2] == 'ALL':
client.backupDatabaseAll(sys.argv[3])
else:
client.backupDatabase(sys.argv[2], sys.argv[3])
exit()
elif type == 'path':
client.backupPath(sys.argv[2], sys.argv[3])
elif type == 'upload':
data = client.upload_file(sys.argv[2])
elif type == 'download':
data = client.generate_download_url(sys.argv[2])
elif type == 'get':
data = client.get_object_info(sys.argv[2])
elif type == 'list':
data = client.get_list("/")
elif type == 'delete_file':
data = client.delete_file(sys.argv[2])
elif type == 'lib':
data = client.get_lib()
else:
data = 'ERROR: 参数不正确!'
else:
client.execute_by_comandline(sys.argv)