宝塔备份插件魔改-谷歌云网盘 2.4
魔改目的
不魔改的话，所有vps的数据备份到一个目录下比较乱，而他这个插件不支持定义备份路径
效果
每个vps，有自己的备份目录
适用版本：
谷歌云网盘 2.4
魔改代码
文件路径/www/server/panel/plugin/gdrive/gdrive_main.py
需要自己修改一下变量:__vps_name,希望有朋友可以再修改为配置项目
新代码
#!/usr/bin/python
# coding: utf-8
# +------------------------------------------------------------------
# | 2.4
# -----------------------------
from __future__ import print_function
from genericpath import isfile
import sys, os, json
# import pickle
import os.path
import google.oauth2.credentials
import google_auth_oauthlib.flow
from googleapiclient.discovery import build
from google_auth_oauthlib.flow import InstalledAppFlow
from google.auth.transport.requests import Request
from googleapiclient.http import MediaFileUpload
from googleapiclient.http import MediaIoBaseDownload
if sys.version_info[0] == 2:
reload(sys)
sys.setdefaultencoding('utf-8')
os.chdir('/www/server/panel')
sys.path.append("class/")
import public, db, time, re
class gdrive_main():
__credentials = "/root/credentials.json"
__setup_path = "/www/server/panel/plugin/gdrive"
__config_path = os.path.join(__setup_path,'config.conf')
__vps_name = "arm2_bt_backup"
__backup_dir_name = "arm2_backup"
__creds = None
__exclude = ""
__scpos = ['https://www.googleapis.com/auth/drive.file']
_panel_path= public.get_panel_path()
__aes_status=os.path.join(__setup_path,'aes_status')
__a_pass=os.path.join(_panel_path,'data/a_pass.pl')
_title = 'Google Drive'
_name = 'Google Drive'
config_separator = "|"
_DEFAULT_AUTH_PROMPT_MESSAGE = (
'Please visit this URL to authorize this application: {url}')
"""str: The message to display when prompting the user for
authorization."""
_DEFAULT_AUTH_CODE_MESSAGE = (
'Enter the authorization code: ')
"""str: The message to display when prompting the user for the
authorization code. Used only by the console strategy."""
_DEFAULT_WEB_SUCCESS_MESSAGE = (
'The authentication flow has completed, you may close this window.')
def __init__(self):
self.set_libList()
self.set_creds()
self.get_exclode()
def create_auth_url(self):
flow = google_auth_oauthlib.flow.Flow.from_client_secrets_file(
'/www/server/panel/plugin/gdrive/credentials.json',
scopes=self.__scpos)
flow.redirect_uri = 'https://drive.aapanel.com'
auth_url, state = flow.authorization_url(
access_type='offline',
prompt='consent',
include_granted_scopes='false')
if os.path.isfile('/tmp/auth_url'):
tmp_exec ="rm -f "+'/tmp/auth_url'
public.ExecShell(tmp_exec)
public.writeFile("/tmp/auth_url",str(auth_url))
return auth_url
def set_libList(self):
libList = public.readFile("/www/server/panel/data/libList.conf")
if libList:
libList = json.loads(libList)
for i in libList:
if "gdrive" in i.values():
return
d = {
"name": "Google Drive",
"type": "Cron job",
"ps": "Back up your website or database to Google Cloud Storage.",
"status": False,
"opt": "gdrive",
"module": "os",
"script": "gdrive",
"help": "http://forum.aapanel.com",
"key": "",
"secret": "",
"bucket": "",
"domain": "",
"check": ["/www/server/panel/plugin/gdrive/gdrive_main.py",
"/www/server/panel/script/backup_gdrive.py"]
}
d1 = {
"name": "谷歌硬盘",
"type": "计划任务",
"ps": "将网站或数据库打包备份到谷歌硬盘.",
"status": False,
"opt": "gdrive",
"module": "os",
"script": "gdrive",
"help": "http://www.bt.cn/bbs",
"key": "",
"secret": "",
"bucket": "",
"domain": "",
"check": ["/www/server/panel/plugin/gdrive/gdrive_main.py",
"/www/server/panel/script/backup_gdrive.py"]
}
language = public.readFile("/www/server/panel/config/config.json")
if "English" in language:
data = d
else:
data = d1
libList.append(data)
public.writeFile("/www/server/panel/data/libList.conf", json.dumps(libList))
return libList
# 设置creds
def set_creds(self):
tmp_data={}
default_conf = {"credentials": {"token": "", "id_token": "", "refresh_token":\
"", "token_uri": "https://oauth2.googleapis.com/token", "client_id": "226011946234-d3e1vashgag64utjedu1ljt9u39ncrpq.apps.googleusercontent.com",\
"client_secret": "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", "scopes": ["https://www.googleapis.com/auth/drive.file"]}}
if os.path.isfile(self.__config_path):
decrypt_key=''
try:
decrypt_key = public.readFile(self.__a_pass)
tmp_data =public.readFile(self.__config_path)
decrypt_data=public.aes_decrypt(tmp_data,decrypt_key)
decrypt_data_list =decrypt_data.split(self.config_separator)
default_conf['credentials']['token']=decrypt_data_list[0]
default_conf['credentials']['refresh_token']=decrypt_data_list[1]
tmp_data = default_conf['credentials']
except:
if os.path.isfile(self.__aes_status):return False
elif os.path.isfile(self.__setup_path + '/token.json'):
test_path = self.__setup_path + '/token.json'
tmp_data=public.readFile(self.__setup_path + '/token.json')
tmp_data = json.loads(tmp_data)['credentials']
# if isinstance(tmp_data,str):tmp_data = json.load(tmp_data)['credentials']
if not tmp_data:return False
try:
self.__creds = google.oauth2.credentials.Credentials(
tmp_data['token'],
tmp_data['refresh_token'],
tmp_data['id_token'],
tmp_data['token_uri'],
tmp_data['client_id'],
tmp_data['client_secret'],
tmp_data['scopes'])
except:
return False
if not self._check_connect():
return False
else:
return True
# 获取token
def get_token(self,get):
import requests
try:
respone = requests.get("https://www.google.com",timeout=2)
except:
return public.returnMsg(False, "连接谷歌失败")
if respone.status_code != 200:
return public.returnMsg(False, "连接谷歌失败")
if not self.set_creds():
return public.returnMsg(False,"验证失败，请根据页面1 2 3 步骤完成验证")
if not os.path.isfile(self.__setup_path + '/token.json') and not os.path.isfile(self.__config_path):
return public.returnMsg(False,"验证失败，请根据页面1 2 3 步骤完成验证")
return public.returnMsg(True,"验证成功")
# 导出配置
def export_conf(self,get):
file_name = "{}/token.json".format(self.__setup_path)
if not os.path.exists(file_name):
return public.returnMsg(False, '没有找到文件')
return public.returnMsg(True, file_name)
# 导入数据
def import_data(self,get):
"""
重构版弃用此方法
"""
import files
f = files.files()
get.f_path = self.__setup_path
result = f.upload(get)
print(get)
return result
# 获取auth_url
def get_auth_url(self,get):
self.create_auth_url()
if os.path.exists("/tmp/auth_url"):
return public.readFile("/tmp/auth_url")
def set_auth_url(self,get):
public.writeFile(self.__aes_status,'True')
if not os.path.isfile(self.__a_pass):
public.writeFile(self.__a_pass,'VE508prf'+public.GetRandomString(10))
encrypt_key = public.readFile(self.__a_pass)
flow = google_auth_oauthlib.flow.Flow.from_client_secrets_file(
'/www/server/panel/plugin/gdrive/credentials.json',
scopes=self.__scpos,
state=get.url.split('state=')[1].split('&code=')[0])
flow.redirect_uri = 'https://drive.aapanel.com'
flow.fetch_token(authorization_response=get.url)
credentials = flow.credentials
add_str = credentials.token+self.config_separator+credentials.refresh_token
decrypt_data = public.aes_encrypt(add_str,encrypt_key)
public.writeFile(self.__config_path,decrypt_data)
public.ExecShell('\cp -rp {} /root/gdrive_token.json'.format(self.__config_path))
public.ExecShell('ln -sf {} /root/gdrive_token.json'.format(self.__config_path))
if not self.set_creds():
return public.returnMsg(False, "验证失败，请根据页面1 2 3 步骤完成验证")
return public.returnMsg(True, "验证成功")
# 检查连接
def check_connect(self,get):
if os.path.isfile(self.__config_path) or os.path.isfile(self.__setup_path + '/token.json'):
self.set_creds()
else:
print("Failed to get Google token, please verify before use")
return public.returnMsg(True, "Failed to get Google token, please verify before use")
service = build('drive', 'v3', credentials=self.__creds)
results = service.files().list(
pageSize=10, fields="nextPageToken, files(id, name)").execute()
try:
results.get('files', [])
return public.returnMsg(False, "验证失败，请根据页面1 2 3 步骤完成验证")
return public.returnMsg(True, "验证成功")
except:
return public.returnMsg(False, "验证失败，请根据页面1 2 3 步骤完成验证")
#检查gdrive连接
def _check_connect(self):
try:
service = build('drive', 'v3', credentials=self.__creds)
results = service.files().list(
pageSize=10, fields="nextPageToken, files(id, name)").execute()
results.get('files', [])
except:
return False
return True
def _get_filename(self,filename):
l = filename.split("/")
return l[-1]
def _create_folder_cycle(self,filepath):
l = filepath.split("/")
fid_list = []
for i in l:
if not i:
continue
fid = self.__get_folder_id(i)
if fid:
fid_list.append(fid)
continue
if not fid_list:
fid_list.append("")
fid_list.append(self.create_folder(i,fid_list[-1]))
return fid_list[-1]
def build_object_name(self, data_type,file_name):
import re
prefix_dict = {
"site": "web",
"database": "db",
"path": "path",
}
file_regx = prefix_dict.get(data_type) + "_(.+)_20\d+_\d+(?:\.|_)"
sub_search = re.search(file_regx.lower(), file_name)
sub_path_name = ""
if sub_search:
sub_path_name = sub_search.groups()[0]
sub_path_name += '/'
# 构建OS存储路径
object_name = '{}/{}/{}'.format(self.__vps_name,data_type,sub_path_name)
if object_name[:1] == "/":
object_name = object_name[1:]
print("构建OS存储路径: "+object_name)
return object_name
# 上传文件
def upload_file(self,get=None,data_type=None):
if isinstance(get,str):
filename = get
get = getObject
get.filepath = self.build_object_name(data_type,filename)
get.path = ''
get.filename = filename
filename = self._get_filename(get.filename)
if(self.__vps_name not in get.filepath):
get.filepath = get.filepath.replace("bt_backup",self.__vps_name)
parents = self._create_folder_cycle(get.filepath)
drive_service = build('drive', 'v3', credentials=self.__creds)
file_metadata = {'name': filename, 'parents': [parents]}
media = MediaFileUpload(get.filename, resumable=True)
file = drive_service.files().create(body=file_metadata, media_body=media, fields='id').execute()
print('上传成功 ,File ID: %s' % file.get('id'))
print("远程路径:" , get.filepath)
return True
def _get_file_id(self,filename):
service = build('drive', 'v3', credentials=self.__creds)
results = service.files().list(pageSize=10, q="name='{}'".format(filename),fields="nextPageToken, files(id, name)").execute()
items = results.get('files', [])
if not items:
return []
else:
for item in items:
return item["id"]
def delete_file(self,filename=None,data_type=None):
file_id = self._get_file_id(filename)
self._delete_file(file_id)
def _delete_file(self,file_id):
try:
drive_service = build('drive', 'v3', credentials=self.__creds)
drive_service.files().delete(fileId=file_id).execute()
except:
pass
print("delete ok")
# 获取目录id
def __get_folder_id(self, floder_name):
service = build('drive', 'v3', credentials=self.__creds)
results = service.files().list(pageSize=10, q="name='{}' and mimeType='application/vnd.google-apps.folder'".format(floder_name),fields="nextPageToken, files(id, name)").execute()
items = results.get('files', [])
if not items:
return []
else:
for item in items:
return item["id"]
# 创建目录
def create_folder(self,folder_name,parents=""):
print("folder_name: {}\nparents: {}".format(folder_name,parents))
service = build('drive', 'v3', credentials=self.__creds)
file_metadata = {
'name': folder_name,
'mimeType': 'application/vnd.google-apps.folder'
}
if parents:
file_metadata['parents'] = [parents]
folder = service.files().create(body=file_metadata,fields='id').execute()
print('Create Folder ID: %s' % folder.get('id'))
return folder.get('id')
# 获取数据库编码
def get_database_character(self,db_name):
try:
import panelMysql
tmp = panelMysql.panelMysql().query("show create database `%s`" % db_name.strip())
c_type = str(re.findall("SET\s+([\w\d-]+)\s",tmp[0][1])[0])
c_types = ['utf8','utf-8','gbk','big5','utf8mb4']
if not c_type.lower() in c_types: return 'utf8'
return c_type
except:
return 'utf8'
def get_exclode(self,exclude=[]):
if not exclude:
tmp_exclude = os.getenv('BT_EXCLUDE')
if tmp_exclude:
exclude = tmp_exclude.split(',')
if not exclude: return ""
for ex in exclude:
self.__exclude += " --exclude=\"" + ex + "\""
self.__exclude += " "
return self.__exclude
def download_file(self,filename):
file_id = self._get_file_id(filename)
service = build('drive', 'v3', credentials=self.__creds)
request = service.files().get_media(fileId=file_id)
with open('/tmp/{}'.format(filename),'wb') as fh:
downloader = MediaIoBaseDownload(fh, request)
done = False
while done is False:
status, done = downloader.next_chunk()
print("Download %d%%." % int(status.progress() * 100))
# 备份网站
def backup_site(self, name, count, exclude=[]):
sql = db.Sql()
path = sql.table('sites').where('name=?', (name,)).getField('path')
startTime = time.time()
if not path:
endDate = time.strftime('%Y/%m/%d %X', time.localtime())
log = "The site [" + name + "] does not exist!"
print("★[" + endDate + "] " + log)
print("----------------------------------------------------------------------------")
return
backup_path = sql.table('config').where("id=?", (1,)).getField('backup_path') + '/site'
if not os.path.exists(backup_path): public.ExecShell("mkdir -p " + backup_path)
filename = backup_path + "/Web_" + name + "_" + time.strftime('%Y%m%d_%H%M%S',
time.localtime()) + '_' + public.GetRandomString(
8) + '.tar.gz'
public.ExecShell("cd " + os.path.dirname(path) + " && tar zcvf '" + filename + "' '" + os.path.basename(path) + "'" + self.__exclude + " > /dev/null")
endDate = time.strftime('%Y/%m/%d %X', time.localtime())
if not os.path.exists(filename):
log = "Website [" + name + "] backup failed!"
print("★[" + endDate + "] " + log)
print("----------------------------------------------------------------------------")
return
# 上传文件
get = getObject()
get.filename = filename
get.filepath = self.__vps_name+'/sites/' + name
print("Start upload")
self.upload_file(get)
outTime = time.time() - startTime
pid = sql.table('sites').where('name=?', (name,)).getField('id')
download = get.filepath + '/' + os.path.basename(filename)
sql.table('backup').add('type,name,pid,filename,addtime,size',('0', download, pid, 'gdrive', endDate, os.path.getsize(filename)))
log = "The website [" + name + "] has been successfully backed up to Google Drive, using [" + str(round(outTime, 2)) + "] seconds"
public.WriteLog('TYPE_CRON', log)
print("★[" + endDate + "] " + log)
print("|---Keep the latest [" + count + "] backups")
print("|---File name:" + os.path.basename(filename))
backups = sql.table('backup').where(
'type=? and pid=? and '
'filename LIKE \'%{}%\''.format(self._name),
('0', pid)).field('id,name,filename').select();
num = len(backups) - int(count)
if num > 0:
for backup in backups:
_base_file_name = backup["name"]
_local_file_name = os.path.join(backup_path,
_base_file_name)
if os.path.isfile(_local_file_name):
public.ExecShell("rm -f " + _local_file_name);
self.echo_info("已清理本地备份文件:" + _local_file_name)
_file_name = backup["filename"]
if _file_name.find(self.CONFIG_SEPARATOR) != -1:
os_file_name = _file_name.split(self.CONFIG_SEPARATOR)[0]
else:
os_file_name = _file_name
self.delete_object(os_file_name)
sql.table('backup').where('id=?', (backup['id'],)).delete();
num -= 1;
self.echo_info("已清理{}过期备份文件：".format(self._title) +
os_file_name)
if num < 1: break;
return None
# 备份数据库
def backup_database(self, name, count):
sql = db.Sql()
path = sql.table('databases').where('name=?', (name,)).getField('id')
startTime = time.time()
if not path:
endDate = time.strftime('%Y/%m/%d %X', time.localtime())
log = "The database [" + name + "] does not exist!"
print("★[" + endDate + "] " + log)
print("----------------------------------------------------------------------------")
return;
backup_path = sql.table('config').where("id=?", (1,)).getField('backup_path') + '/database'
if not os.path.exists(backup_path): public.ExecShell("mkdir -p " + backup_path);
filename = backup_path + "/Db_" + name + "_" + time.strftime('%Y%m%d_%H%M%S',
time.localtime()) + '_' + public.GetRandomString(
8) + ".sql.gz"
import re
mysql_root = '"'+sql.table('config').where("id=?", (1,)).getField('mysql_root')+'"'
mycnf = public.readFile('/etc/my.cnf')
rep = "\[mysqldump\]\nuser=root"
sea = "[mysqldump]\n"
subStr = sea + "user=root\npassword=" + mysql_root + "\n"
mycnf = mycnf.replace(sea, subStr)
if len(mycnf) > 100:
public.writeFile('/etc/my.cnf', mycnf)
sh = "/www/server/mysql/bin/mysqldump --default-character-set="+ self.get_database_character(name) +" --force --opt " + name + " | gzip > " + filename
public.ExecShell(sh)
if not os.path.exists(filename):
endDate = time.strftime('%Y/%m/%d %X', time.localtime())
log = "Database [" + name + "] backup failed!"
print("★[" + endDate + "] " + log)
print("---------------------------------------------------------------------------")
return
mycnf = public.readFile('/etc/my.cnf')
mycnf = mycnf.replace(subStr, sea)
if len(mycnf) > 100:
public.writeFile('/etc/my.cnf', mycnf)
# 上传
get = getObject()
get.filename = filename
get.filepath = self.__vps_name+'/database/' + name
print("开始上传文件：",get.filename,get.filepath)
self.upload_file(get)
endDate = time.strftime('%Y/%m/%d %X', time.localtime())
outTime = time.time() - startTime
pid = sql.table('databases').where('name=?', (name,)).getField('id')
download = get.filepath + '/' + os.path.basename(filename)
sql.table('backup').add('type,name,pid,filename,addtime,size',(1, download, pid, 'gdrive', endDate, os.path.getsize(filename)))
log = "The database [" + name + "] has been successfully backed up to Google Drive, using [" + str(round(outTime, 2)) + "] seconds"
public.WriteLog('TYPE_CRON', log)
print("★[" + endDate + "] " + log)
print("|---Keep the latest [" + count + "] backups")
print("|---File name:" + os.path.basename(filename))
# 清理本地文件
public.ExecShell("rm -f " + filename)
# 清理多余备份
backups = sql.table('backup').where('type=? and pid=? and filename=?', ('1', pid, 'gdrive')).field(
'id,name,filename').select()
num = len(backups) - int(count)
if num > 0:
for backup in backups:
if os.path.exists(backup['filename']):
public.ExecShell("rm -f " + backup['filename'])
name = self._get_filename(backup['name'].split("/")[-1])
file_id = self._get_file_id(name)
try:
self._delete_file(file_id)
except Exception as e:
print(e)
sql.table('backup').where('id=?', (backup['id'],)).delete()
num -= 1
print("|---The expired backup file has been cleaned up：" + backup['name'])
if num < 1: break
return None
# 备份指定目录
def backup_path(self, path, count,exclude=[]):
sql = db.Sql()
startTime = time.time()
if path[-1:] == '/': path = path[:-1]
name = os.path.basename(path)
backup_path = sql.table('config').where("id=?", (1,)).getField('backup_path') + '/path'
if not os.path.exists(backup_path): os.makedirs(backup_path);
filename = backup_path + "/Path_" + name + "_" + time.strftime('%Y%m%d_%H%M%S',
time.localtime()) + '.tar.gz'
os.system("cd " + os.path.dirname(path) + " && tar zcvf '" + filename + "' '" + os.path.basename(path) + "'" + self.__exclude + " > /dev/null")
get = getObject()
get.filename = filename
get.filepath = self.__vps_name+'/path/' + name
self.upload_file(get)
endDate = time.strftime('%Y/%m/%d %X', time.localtime())
if not os.path.exists(filename):
log = "Directory [" + path + "] backup failed"
print("★[" + endDate + "] " + log)
print("----------------------------------------------------------------------------")
return
outTime = time.time() - startTime
download = get.filepath + '/' + os.path.basename(filename)
sql.table('backup').add('type,name,filename,addtime,size',('2', download, 'gdrive', endDate, os.path.getsize(filename)))
log = "The directory [" + path + "] is successfully backed up, using [" + str(round(outTime, 2)) + "] seconds"
public.WriteLog('TYPE_CRON', log)
print("★[" + endDate + "] " + log)
print("|---Keep the latest [" + count + "] backups")
print("|---File name:" + filename)
# 清理多余备份
backups = sql.table('backup').where('type=? and filename=?',('2',"gdrive")).field('id,name,filename').select()
# 清理本地备份
if os.path.exists(filename): os.remove(filename)
num = len(backups) - int(count)
if num > 0:
for backup in backups:
if os.path.exists(backup['filename']):
public.ExecShell("rm -f " + backup['filename'])
name = self._get_filename(backup['name'].split("/")[-1])
file_id = self._get_file_id(name)
try:
self._delete_file(file_id)
except Exception as e:
print(e)
sql.table('backup').where('id=?', (backup['id'],)).delete()
num -= 1
print("|---The expired backup file has been cleaned up：" + backup['filename'])
if num < 1: break
return None
def backup_all_site(self, save):
sites = public.M('sites').field('name').select()
for site in sites:
self.backup_site(site['name'], save)
def backup_all_databases(self, save):
databases = public.M('databases').field('name').select()
for database in databases:
self.backup_database(database['name'], save)
def get_function_args(self, func):
import sys
if sys.version_info[0] == 3:
import inspect
return inspect.getfullargspec(func).args
else:
return func.__code__.co_varnames
def execute_command(self,args):
client = self
cls_args = self.get_function_args(panelBackup.backup.__init__)
if "cron_info" in cls_args and len(args) == 5:
cron_name = args[4]
cron_info = {
"echo": cron_name
}
backup_tool = panelBackup.backup(cloud_object=client,
cron_info=cron_info)
else:
backup_tool = panelBackup.backup(cloud_object=client)
_type = args[1]
data = None
if _type == 'site':
if args[2] == 'ALL':
backup_tool.backup_site_all(save=int(args[3]), echo_id=args[4])
else:
backup_tool.backup_site(args[2], save=int(args[3]), echo_id=args[4])
exit()
elif _type == 'database':
if args[2] == 'ALL':
backup_tool.backup_database_all(int(args[3]), echo_id=args[4])
else:
backup_tool.backup_database(args[2], save=int(args[3]), echo_id=args[4])
exit()
elif _type == 'path':
backup_tool.backup_path(args[2], save=int(args[3]), echo_id=args[4])
exit()
else:
data = 'ERROR: The parameter is incorrect!';
if data:
print()
print(json.dumps(data))
class getObject:
pass
if __name__ == "__main__":
import json
import panelBackup
data = None
new_version = True if panelBackup._VERSION >= 1.2 else False
q = gdrive_main()
# 检查获取验证
q.set_creds()
type = sys.argv[1]
if not new_version:
if type == 'site':
if sys.argv[2] == 'ALL':
q.backup_all_site( sys.argv[3])
else:
q.backup_site(sys.argv[2], sys.argv[3])
exit()
elif type == 'database':
if sys.argv[2] == 'ALL':
data = q.backup_all_databases(sys.argv[3])
else:
data = q.backup_database(sys.argv[2], sys.argv[3])
exit()
elif type == 'path':
data = q.backup_path(sys.argv[2],sys.argv[3])
else:
data = 'ERROR: The parameter is incorrect!'
else:
q.execute_command(sys.argv)