mirror of https://github.com/midoks/mdserver-web
parent
bbf76c724d
commit
41306ab505
@ -0,0 +1,119 @@ |
||||
# 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 = mw.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 = mw.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]) |
||||
Loading…
Reference in new issue