cc-4227: changed upgrade method. added test stubs for it

This commit is contained in:
Rudi Grinberg 2012-08-20 11:19:57 -04:00
parent 37aa879ed0
commit b4c014dbcc
3 changed files with 33 additions and 37 deletions

View File

@ -0,0 +1,5 @@
import unittest
import upgrade2dot2
class TestUpgrade(unittest.TestCase):
pass

View File

@ -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)

View File

@ -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)