cc-4227: changed upgrade method. added test stubs for it
This commit is contained in:
parent
37aa879ed0
commit
b4c014dbcc
python_apps/media-monitor2
|
@ -0,0 +1,5 @@
|
|||
import unittest
|
||||
import upgrade2dot2
|
||||
|
||||
class TestUpgrade(unittest.TestCase):
|
||||
pass
|
|
@ -1,37 +0,0 @@
|
|||
import os
|
||||
from configobj import ConfigObj
|
||||
import traceback
|
||||
|
||||
upgrades_config = {
|
||||
'/etc/airtime/media-monitor.cfg' :
|
||||
{
|
||||
'check_filesystem_events' : 5,
|
||||
'check_airtime_events' : 30,
|
||||
'touch_interval' : 5,
|
||||
'chunking_number' : 450,
|
||||
'request_max_wait' : 3.0,
|
||||
'rmq_event_wait' : 0.1,
|
||||
'logpath' : '/var/log/airtime/media-monitor/media-monitor.log',
|
||||
'index_path' : '/var/tmp/airtime/media-monitor/last_index',
|
||||
},
|
||||
'/etc/airtime/api_client.cfg' :
|
||||
{
|
||||
'reload_metadata_group' :
|
||||
'reload-metadata-group/format/json/api_key/%%api_key%%',
|
||||
}
|
||||
}
|
||||
|
||||
def upgrade(upgrade_data):
|
||||
for f, values in upgrade_data:
|
||||
if not os.path.exists(f):
|
||||
print("Cannot upgrade '%s'. Skipping this file" % f)
|
||||
continue
|
||||
try:
|
||||
cfg = ConfigObj(f)
|
||||
for k,v in values:
|
||||
if k not in cfg: cfg[k] = v
|
||||
except Exception:
|
||||
print("Error upgrading")
|
||||
print( traceback.format_exc() )
|
||||
|
||||
if __name__ == "__main__": upgrade(upgrades_config)
|
|
@ -0,0 +1,28 @@
|
|||
import os
|
||||
from configobj import ConfigObj
|
||||
import traceback
|
||||
|
||||
upgrades_config = {
|
||||
'/etc/airtime/media-monitor.cfg' : '../media-monitor/media-monitor.cfg',
|
||||
'/etc/airtime/api_client.cfg' : '../api_clients/api_client.cfg',
|
||||
}
|
||||
|
||||
def upgrade(upgrade_data):
|
||||
"""
|
||||
Must be ran as sudo. will do upgrade of configuration files by filling in
|
||||
missing values according to upgrade_data
|
||||
"""
|
||||
for source, destination in upgrade_data:
|
||||
if not os.path.exists(source):
|
||||
print("Cannot upgrade '%s'. Skipping this file" % source)
|
||||
continue
|
||||
try:
|
||||
cfg_source, cfg_dest = ConfigObj(source), ConfigObj(destination)
|
||||
for key, val in cfg_source:
|
||||
if key not in cfg_dest: cfg_dest[key] = val
|
||||
cfg_dest.write()
|
||||
except Exception:
|
||||
print("Error upgrading")
|
||||
print( traceback.format_exc() )
|
||||
|
||||
if __name__ == "__main__": upgrade(upgrades_config)
|
Loading…
Reference in New Issue