mirror of https://github.com/midoks/mdserver-web
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
119 lines
4.1 KiB
119 lines
4.1 KiB
# coding: utf-8
|
|
#-----------------------------
|
|
# 网站备份工具
|
|
#-----------------------------
|
|
|
|
import sys
|
|
import os
|
|
|
|
if sys.platform != 'darwin':
|
|
os.chdir('/www/server/mdserver-web')
|
|
|
|
|
|
chdir = os.getcwd()
|
|
sys.path.append(chdir + '/class/core')
|
|
|
|
# reload(sys)
|
|
# sys.setdefaultencoding('utf-8')
|
|
|
|
|
|
import mw
|
|
import db
|
|
import time
|
|
|
|
|
|
class backupTools:
|
|
|
|
def backupDatabase(self, name, count):
|
|
db_path = getServerDir() + '/mysql'
|
|
db_name = 'mysql'
|
|
name = mw.M('databases').dbPos(db_path, 'mysql').where(
|
|
'name=?', (name,)).getField('name')
|
|
startTime = time.time()
|
|
if not name:
|
|
endDate = time.strftime('%Y/%m/%d %X', time.localtime())
|
|
log = "数据库[" + name + "]不存在!"
|
|
print("★[" + endDate + "] " + log)
|
|
print(
|
|
"----------------------------------------------------------------------------")
|
|
return
|
|
|
|
backup_path = mw.getRootDir() + '/backup/database'
|
|
if not os.path.exists(backup_path):
|
|
mw.execShell("mkdir -p " + backup_path)
|
|
|
|
filename = backup_path + "/db_" + name + "_" + \
|
|
time.strftime('%Y%m%d_%H%M%S', time.localtime()) + ".sql.gz"
|
|
|
|
import re
|
|
mysql_root = mw.M('config').dbPos(db_path, db_name).where(
|
|
"id=?", (1,)).getField('mysql_root')
|
|
|
|
mycnf = mw.readFile(db_path + '/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:
|
|
mw.writeFile(db_path + '/etc/my.cnf', mycnf)
|
|
|
|
mw.execShell(
|
|
db_path + "/bin/mysqldump --opt --default-character-set=utf8 " + name + " | gzip > " + filename)
|
|
|
|
if not os.path.exists(filename):
|
|
endDate = time.strftime('%Y/%m/%d %X', time.localtime())
|
|
log = "数据库[" + name + "]备份失败!"
|
|
print("★[" + endDate + "] " + log)
|
|
print(
|
|
"----------------------------------------------------------------------------")
|
|
return
|
|
|
|
mycnf = mw.readFile(db_path + '/etc/my.cnf')
|
|
mycnf = mycnf.replace(subStr, sea)
|
|
if len(mycnf) > 100:
|
|
mw.writeFile(db_path + '/etc/my.cnf', mycnf)
|
|
|
|
endDate = time.strftime('%Y/%m/%d %X', time.localtime())
|
|
outTime = time.time() - startTime
|
|
pid = mw.M('databases').dbPos(db_path, db_name).where(
|
|
'name=?', (name,)).getField('id')
|
|
|
|
mw.M('backup').add('type,name,pid,filename,addtime,size', (1, os.path.basename(
|
|
filename), pid, filename, endDate, os.path.getsize(filename)))
|
|
log = "数据库[" + name + "]备份成功,用时[" + str(round(outTime, 2)) + u"]秒"
|
|
mw.writeLog('计划任务', log)
|
|
print("★[" + endDate + "] " + log)
|
|
print("|---保留最新的[" + count + "]份备份")
|
|
print("|---文件名:" + filename)
|
|
|
|
# 清理多余备份
|
|
backups = mw.M('backup').where(
|
|
'type=? and pid=?', ('1', pid)).field('id,filename').select()
|
|
|
|
num = len(backups) - int(count)
|
|
if num > 0:
|
|
for backup in backups:
|
|
mw.execShell("rm -f " + backup['filename'])
|
|
mw.M('backup').where('id=?', (backup['id'],)).delete()
|
|
num -= 1
|
|
print("|---已清理过期备份文件：" + backup['filename'])
|
|
if num < 1:
|
|
break
|
|
|
|
def backupDatabaseAll(self, save):
|
|
db_path = getServerDir() + '/mysql'
|
|
db_name = 'mysql'
|
|
databases = mw.M('databases').dbPos(
|
|
db_path, db_name).field('name').select()
|
|
for database in databases:
|
|
self.backupDatabase(database['name'], save)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
backup = backupTools()
|
|
type = sys.argv[1]
|
|
elif type == 'database':
|
|
if sys.argv[2] == 'ALL':
|
|
backup.backupDatabaseAll(sys.argv[3])
|
|
else:
|
|
backup.backupDatabase(sys.argv[2], sys.argv[3])
|
|
|