mirror of
https://github.com/ElvishArtisan/rivendell.git
synced 2025-05-21 07:13:06 +02:00
adca39a4cf
* Added code to display the metadata of the proposed backup to restore in the warning message in rdautorest(8). Signed-off-by: Fred Gleason <fredg@paravelsystems.com>
138 lines
4.2 KiB
Python
Executable File
138 lines
4.2 KiB
Python
Executable File
#!%PYTHON_BANGPATH%
|
|
|
|
# rdautorest.py
|
|
#
|
|
# Unattended restore script for Rivendell
|
|
#
|
|
# (C) Copyright 2024 Fred Gleason <fredg@paravelsystems.com>
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License version 2 as
|
|
# published by the Free Software Foundation.
|
|
#
|
|
# 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 General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public
|
|
# License along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
|
#
|
|
|
|
import configparser
|
|
import os
|
|
from pathlib import Path
|
|
import sys
|
|
import syslog
|
|
|
|
USAGE='USAGE: rdautorest.py [--yes] <backup-mountpoint>'
|
|
|
|
yes=False
|
|
|
|
if(len(sys.argv)==2):
|
|
mountpoint=sys.argv[1]
|
|
else:
|
|
if(len(sys.argv)==3):
|
|
if(sys.argv[1]=='--yes'):
|
|
yes=True
|
|
mountpoint=sys.argv[2]
|
|
else:
|
|
if(sys.argv[2]=='--yes'):
|
|
yes=True
|
|
mountpoint=sys.argv[1]
|
|
else:
|
|
print(USAGE)
|
|
exit(1)
|
|
else:
|
|
print(USAGE)
|
|
exit(1)
|
|
|
|
#
|
|
# Load rd.conf(5)
|
|
#
|
|
rd_config=configparser.ConfigParser(interpolation=None)
|
|
rd_config.read_file(open('/etc/rd.conf'))
|
|
|
|
#
|
|
# Open the syslog
|
|
#
|
|
syslog.openlog('rdautorest.py',logoption=syslog.LOG_PID|syslog.LOG_PERROR,facility=int(rd_config.get('Identity','SyslogFacility',fallback=syslog.LOG_USER)))
|
|
|
|
#
|
|
# Mount backup device
|
|
#
|
|
Path(mountpoint).mkdir(parents=True,exist_ok=True)
|
|
result=os.WEXITSTATUS(os.system(command='mount '+mountpoint+" 2> /dev/null"))
|
|
if((result!=0)and(result!=64)):
|
|
syslog.syslog(syslog.LOG_ERR,'unable to mount backup drive')
|
|
exit(1)
|
|
os.system(command='sleep 5')
|
|
|
|
#
|
|
# Print warning message
|
|
#
|
|
if(not yes):
|
|
try:
|
|
with open(mountpoint+'/INFO.txt','r') as f:
|
|
print('WARNING!')
|
|
print('This operation will COMPLETELY OVERWRITE the existing')
|
|
print('Rivendell data on this system, replacing it with the')
|
|
print('contents of the following data backup:')
|
|
print()
|
|
print(f.read())
|
|
f.close()
|
|
print('This operation cannot be undone!')
|
|
except FileNotFoundError:
|
|
print('WARNING!')
|
|
print('This operation will COMPLETELY OVERWRITE the existing')
|
|
print('Rivendell data on this system, replacing it with the')
|
|
print('contents of the specified data backup. This operation')
|
|
print('cannot be undone!')
|
|
print('')
|
|
resp=input('Are you sure you want to proceed (y/N)?')
|
|
if((resp.upper()!='Y')and(resp.upper()!='YES')):
|
|
os.system(command='umount '+mountpoint)
|
|
os.rmdir(mountpoint)
|
|
exit(0)
|
|
|
|
syslog.syslog(syslog.LOG_INFO,'Starting Rivendell restore from "'+mountpoint+'"')
|
|
|
|
#
|
|
# Stop Rivendell service
|
|
#
|
|
result=os.system(command='/bin/systemctl stop rivendell')
|
|
|
|
#
|
|
# Create new database
|
|
#
|
|
sql='drop\ database\ '+rd_config.get('mySQL','Database')+'\;'
|
|
cmd='echo '+sql+' | mysql -h '+rd_config.get('mySQL','Hostname')+' -u '+rd_config.get('mySQL','Loginname')+' -p'+rd_config.get('mySQL','Password')+' '+rd_config.get('mySQL','Database')
|
|
os.system(command=cmd)
|
|
|
|
sql='create\ database\ '+rd_config.get('mySQL','Database')+'\;'
|
|
cmd='echo '+sql+' | mysql -h '+rd_config.get('mySQL','Hostname')+' -u '+rd_config.get('mySQL','Loginname')+' -p'+rd_config.get('mySQL','Password')
|
|
os.system(command=cmd)
|
|
|
|
cmd='gzip -cd '+mountpoint+'/db.sql.gz | sed \'1{/999999.*sandbox/d}\' | mysql -h '+rd_config.get('mySQL','Hostname')+' -u '+rd_config.get('mySQL','Loginname')+' -p'+rd_config.get('mySQL','Password')+' '+rd_config.get('mySQL','Database')
|
|
os.system(command=cmd)
|
|
|
|
#
|
|
# Copy Audio Store
|
|
#
|
|
cmd='rsync -av --delete '+mountpoint+'/snd/ /var/snd '
|
|
os.system(command=cmd)
|
|
|
|
#
|
|
# Start Rivendell service
|
|
#
|
|
result=os.system(command='/bin/systemctl restart rivendell')
|
|
|
|
#
|
|
# Unmount backup device
|
|
#
|
|
os.system(command='umount '+mountpoint)
|
|
os.rmdir(mountpoint)
|
|
|
|
syslog.syslog(syslog.LOG_INFO,'Completed Rivendell restore from "'+mountpoint+'"')
|