Rivendellaudio/scripts/init_template_host.py
Fred Gleason 7cc4e2df57 2018-07-07 Fred Gleason <fredg@paravelsystems.com>
* Fixed a bug in rddbmgr(8) that caused the 'AUDIO_CARDS',
	'AUDIO_INPUTS' and 'AUDIO_OUTPUTS' table to be created with the
	incorrect engine type when using '--modify'.
	* Updated the 'init_template_host.py' script to work with
	schema version 287.
2018-07-06 11:29:25 -04:00

115 lines
3.0 KiB
Python
Executable File

#!/usr/bin/python
# init_template_host.py
#
# Initialize a template-only host profile so it can be used as a
# template. (Requires the 'mysql-connector-python' package).
#
# (C) Copyright 2018 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.
#
# Usage: init_template_host.py
from __future__ import print_function
import argparse
import ConfigParser
import mysql.connector
import sys
def eprint(*args,**kwargs):
print(*args,file=sys.stderr,**kwargs)
def GetDbCredentials():
config=ConfigParser.ConfigParser()
config.readfp(open('/etc/rd.conf'))
return (config.get('mySQL','Loginname'),config.get('mySQL','Password'),
config.get('mySQL','Hostname'),config.get('mySQL','Database'))
def OpenDb():
creds=GetDbCredentials()
return mysql.connector.connect(user=creds[0],password=creds[1],
host=creds[2],database=creds[3],buffered=True)
#
# Read arguments
#
parser=argparse.ArgumentParser(description='Initialize a Rivendell host entry for use as a template')
parser.add_argument("--host",required=True,help="name of host entry to initialize")
parser.add_argument("--device",required=True,help='add a virtual device (CARD:INPUTS:OUTPUTS)')
args = parser.parse_args()
device = args.device.split(":")
if(len(device)!=3):
eprint("init_template_host.py: invalid argument to --device")
sys.exit(1)
cardnum=int(device[0])
if((cardnum<0) or (cardnum>7)):
eprint("init_template_host.py: invalid card number")
sys.exit(1)
inputs=int(device[1])
if((inputs<0) or (inputs>8)):
eprint("init_template_host.py: invalid input quantity")
sys.exit(1)
outputs=int(device[2])
if((outputs<0) or (outputs>8)):
eprint("init_template_host.py: invalid output quantity")
sys.exit(1)
#
# Open the database
#
db=OpenDb()
#
# Sanity checks
#
sql='select NAME from STATIONS where NAME="'+args.host+'"';
q=db.cursor()
q.execute(sql)
if(len(q.fetchall()) != 1):
eprint('init_template.py: no such host\n')
sys.exit(1)
q.close()
#
# Write host records
#
sql='update AUDIO_CARDS set '
sql+='DRIVER=2,'
sql+='NAME="Dummy Template Card",'
sql+='INPUTS='+str(inputs)+','
sql+='OUTPUTS='+str(outputs)
sql+=' where STATION_NAME="'+args.host+'" && CARD_NUMBER='+str(cardnum)
q=db.cursor()
q.execute(sql)
q.close()
sql='update STATIONS set '
sql+='STATION_SCANNED="Y" where '
sql+='NAME="'+args.host+'"'
q=db.cursor()
q.execute(sql)
q.close()
db.close()