mirror of
https://github.com/aaPanel/aaPanel.git
synced 2026-08-21 15:12:26 +02:00
93 lines
3.8 KiB
Python
93 lines
3.8 KiB
Python
# coding: utf-8
|
|
# -------------------------------------------------------------------
|
|
# aapanel
|
|
# -------------------------------------------------------------------
|
|
# Copyright (c) 2015-2099 aapanel(http://www.aapanel.com) All rights reserved.
|
|
# -------------------------------------------------------------------
|
|
# Author: miku <miku@bt.cn>
|
|
# -------------------------------------------------------------------
|
|
import os
|
|
import sys
|
|
import warnings
|
|
|
|
if "/www/server/panel/class" not in sys.path:
|
|
sys.path.insert(0, "/www/server/panel/class")
|
|
if "/www/server/panel/class_v2" not in sys.path:
|
|
sys.path.insert(0, "/www/server/panel/class_v2")
|
|
if "/www/server/panel" not in sys.path:
|
|
sys.path.insert(0, "/www/server/panel")
|
|
|
|
import public
|
|
from mod.project.backup_restore.base_util import BaseUtil
|
|
from mod.project.backup_restore.config_manager import ConfigManager
|
|
|
|
warnings.filterwarnings("ignore", category=SyntaxWarning)
|
|
|
|
|
|
class SshModule(BaseUtil, ConfigManager):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.base_path = '/www/backup/backup_restore'
|
|
self.bakcup_task_json = self.base_path + '/backup_task.json'
|
|
self.ssh_path = "/www/server/panel/config/ssh_info"
|
|
|
|
def backup_ssh_data(self, timestamp):
|
|
self.print_log("====================================================", "backup")
|
|
self.print_log(public.lang("Start backing up terminal data"), "backup")
|
|
|
|
ssh_backup_path = self.base_path + "/{timestamp}_backup/ssh".format(timestamp=timestamp)
|
|
if not os.path.exists(ssh_backup_path):
|
|
public.ExecShell('mkdir -p {}'.format(ssh_backup_path))
|
|
print(self.ssh_path)
|
|
public.ExecShell("\cp -rpa {} {}".format(self.ssh_path, ssh_backup_path))
|
|
|
|
ssh_info = {
|
|
'status': 2,
|
|
'msg': None,
|
|
'ssh_info_path': ssh_backup_path,
|
|
}
|
|
|
|
backup_size = self.format_size(self.get_file_size(ssh_backup_path))
|
|
self.print_log(public.lang("Terminal data backup completed. Data size: {}").format(backup_size), 'backup')
|
|
|
|
data_list = self.get_backup_data_list(timestamp)
|
|
data_list['data_list']['ssh'] = ssh_info
|
|
self.update_backup_data_list(timestamp, data_list)
|
|
|
|
def restore_ssh_data(self, timestamp):
|
|
self.print_log("==================================", "restore")
|
|
self.print_log(public.lang("Start restoring terminal data"), "restore")
|
|
|
|
restore_data = self.get_restore_data_list(timestamp)
|
|
ssh_data = restore_data['data_list']['ssh']
|
|
ssh_info_path = ssh_data['ssh_info_path'] + "/ssh_info"
|
|
|
|
restore_data['data_list']['ssh']['restore_status'] = 1
|
|
self.update_restore_data_list(timestamp, restore_data)
|
|
|
|
if not os.path.exists(ssh_info_path):
|
|
self.print_log(public.lang("Restore failed, file does not exist"), "restore")
|
|
return
|
|
if not os.path.exists(self.ssh_path):
|
|
public.ExecShell("mkdir -p {}".format(self.ssh_path))
|
|
public.ExecShell("\cp -rpa {}/* {}".format(ssh_info_path, self.ssh_path))
|
|
self.print_log(public.lang("Terminal data restoration completed"), "restore")
|
|
|
|
restore_data['data_list']['ssh']['restore_status'] = 2
|
|
self.update_restore_data_list(timestamp, restore_data)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# 获取命令行参数
|
|
if len(sys.argv) < 2:
|
|
print("Usage: btpython backup_manager.py <method> <timestamp>")
|
|
sys.exit(1)
|
|
method_name = sys.argv[1] # 方法名
|
|
timestamp = sys.argv[2]
|
|
ssh_manager = SshModule() # 实例化对象
|
|
if hasattr(ssh_manager, method_name): # 检查方法是否存在
|
|
method = getattr(ssh_manager, method_name) # 获取方法
|
|
method(timestamp) # 调用方法
|
|
else:
|
|
print(f"Error: Method '{method_name}' does not exist")
|