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.
106 lines
4.7 KiB
106 lines
4.7 KiB
# -*- coding: utf-8 -*-
|
|
###############################################################################
|
|
#
|
|
# Cybrosys Technologies Pvt. Ltd.
|
|
#
|
|
# Copyright (C) 2023-TODAY Cybrosys Technologies(<https://www.cybrosys.com>)
|
|
# Author: Aslam AK (odoo@cybrosys.com)
|
|
#
|
|
# You can modify it under the terms of the GNU LESSER
|
|
# GENERAL PUBLIC LICENSE (LGPL v3), Version 3.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU LESSER GENERAL PUBLIC LICENSE (LGPL v3) for more details.
|
|
#
|
|
# You should have received a copy of the GNU LESSER GENERAL PUBLIC LICENSE
|
|
# (LGPL v3) along with this program.
|
|
# If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
###############################################################################
|
|
import ftplib
|
|
import gdown
|
|
import odoo
|
|
import odoo.modules.registry
|
|
import os
|
|
import paramiko
|
|
import requests
|
|
import tempfile
|
|
from odoo import fields, models, _
|
|
from odoo.exceptions import UserError
|
|
from odoo.http import dispatch_rpc
|
|
from odoo.service import db
|
|
from odoo.tools.misc import str2bool
|
|
|
|
|
|
class DataBaseRestore(models.TransientModel):
|
|
"""
|
|
Database Restore Model
|
|
"""
|
|
_name = "database.restore"
|
|
_description = "Restore Backups"
|
|
|
|
db_file = fields.Char(string="File",
|
|
help="Path or URL to the database backup file")
|
|
db_name = fields.Char(string="Database Name",
|
|
help="Name of the database to restore")
|
|
db_master_pwd = fields.Char(string="Database Master Password",
|
|
help="Master password for the database")
|
|
backup_location = fields.Char(string="Backup Location",
|
|
help="Location where the database backup is "
|
|
"stored")
|
|
|
|
def action_restore_database(self, copy=False):
|
|
"""
|
|
Function to restore the database Backups
|
|
"""
|
|
insecure = odoo.tools.config.verify_admin_password('admin')
|
|
if insecure and self.db_master_pwd:
|
|
dispatch_rpc('db', 'change_admin_password',
|
|
["admin", self.db_master_pwd])
|
|
try:
|
|
db.check_super(self.db_master_pwd)
|
|
temp_file = tempfile.NamedTemporaryFile(delete=False)
|
|
if self.backup_location == 'Google Drive':
|
|
gdown.download(self.db_file, temp_file.name, quiet=False)
|
|
elif self.backup_location in ['Dropbox', 'OneDrive']:
|
|
response = requests.get(self.db_file, stream=True)
|
|
temp_file.write(response.content)
|
|
elif self.backup_location == 'FTP Storage':
|
|
for rec in self.env['db.backup.configure'].search([]):
|
|
if rec.backup_destination == 'ftp' and \
|
|
rec.ftp_path == os.path.dirname(self.db_file):
|
|
ftp_server = ftplib.FTP()
|
|
ftp_server.connect(rec.ftp_host, int(rec.ftp_port))
|
|
ftp_server.login(rec.ftp_user, rec.ftp_password)
|
|
ftp_server.retrbinary("RETR " + self.db_file,
|
|
temp_file.write)
|
|
temp_file.seek(0)
|
|
ftp_server.quit()
|
|
elif self.backup_location == 'SFTP Storage':
|
|
for rec in self.env['db.backup.configure'].search([]):
|
|
if rec.backup_destination == 'sftp' and \
|
|
rec.sftp_path == os.path.dirname(self.db_file):
|
|
sftp_client = paramiko.SSHClient()
|
|
sftp_client.set_missing_host_key_policy(
|
|
paramiko.AutoAddPolicy())
|
|
sftp_client.connect(hostname=rec.sftp_host,
|
|
username=rec.sftp_user,
|
|
password=rec.sftp_password,
|
|
port=rec.sftp_port)
|
|
sftp_server = sftp_client.open_sftp()
|
|
sftp_server.getfo(self.db_file, temp_file)
|
|
sftp_server.close()
|
|
sftp_client.close()
|
|
elif self.backup_location == 'Local Storage':
|
|
temp_file.name = self.db_file
|
|
db.restore_db(self.db_name, temp_file.name, str2bool(copy))
|
|
temp_file.close()
|
|
return {
|
|
'type': 'ir.actions.act_url',
|
|
'url': '/web/database/manager'
|
|
}
|
|
except Exception as error:
|
|
raise UserError(
|
|
_("Database restore error: %s" % (str(error) or repr(error))))
|
|
|