Merge branch 'devel' of dev.sourcefabric.org:airtime into devel
This commit is contained in:
commit
cad4ad0eb4
|
@ -1,51 +0,0 @@
|
||||||
from configobj import ConfigObj
|
|
||||||
import unittest
|
|
||||||
import os
|
|
||||||
import upgrade2dot2
|
|
||||||
|
|
||||||
|
|
||||||
def create_cfg(cfg):
|
|
||||||
config = ConfigObj()
|
|
||||||
config.filename = cfg['path']
|
|
||||||
for k,v in cfg['data'].iteritems(): config[k] = v
|
|
||||||
return config
|
|
||||||
|
|
||||||
class TestUpgrade(unittest.TestCase):
|
|
||||||
|
|
||||||
def setUp(self):
|
|
||||||
self.source = 'ttt1.cfg'
|
|
||||||
self.dest = 'ttt2.cfg'
|
|
||||||
|
|
||||||
def test_upgrade(self):
|
|
||||||
cf = {
|
|
||||||
'source' : {
|
|
||||||
'path' : self.source,
|
|
||||||
'data' : {
|
|
||||||
'key1' : 'val1',
|
|
||||||
'key2' : 'val2',
|
|
||||||
'key3' : 5,
|
|
||||||
'key4' : 10,},
|
|
||||||
},
|
|
||||||
'dest' : {
|
|
||||||
'path' : self.dest,
|
|
||||||
'data' : {
|
|
||||||
'key1' : 'NEW_VAL',
|
|
||||||
'key3' : 25, }
|
|
||||||
}
|
|
||||||
}
|
|
||||||
config1, config2 = create_cfg(cf['source']), create_cfg(cf['dest'])
|
|
||||||
for c in [config1,config2]: c.write()
|
|
||||||
self.assertTrue( os.path.exists(cf['source']['path']) )
|
|
||||||
self.assertTrue( os.path.exists(cf['dest']['path']) )
|
|
||||||
# Finished preparing
|
|
||||||
upgrade2dot2.upgrade(cf['source']['path'], cf['dest']['path'] )
|
|
||||||
c1, c2 = ConfigObj(cf['source']['path']), ConfigObj(cf['dest']['path'])
|
|
||||||
self.assertEqual( c2['key2'], 'val2')
|
|
||||||
self.assertEqual( c2['key4'], '10')
|
|
||||||
self.assertEqual( c2['key3'], '25')
|
|
||||||
|
|
||||||
def tearDown(self):
|
|
||||||
for clean in [ self.source, self.dest ]:
|
|
||||||
os.unlink(clean)
|
|
||||||
|
|
||||||
if __name__ == '__main__': unittest.main()
|
|
|
@ -1,20 +0,0 @@
|
||||||
import os
|
|
||||||
from configobj import ConfigObj
|
|
||||||
import traceback
|
|
||||||
|
|
||||||
def upgrade(source, destination):
|
|
||||||
"""
|
|
||||||
Must be ran as sudo. will do upgrade of configuration files by filling in
|
|
||||||
missing values according to upgrade_data
|
|
||||||
"""
|
|
||||||
if not os.path.exists(source):
|
|
||||||
print("Cannot upgrade '%s'. Skipping this file" % source)
|
|
||||||
return
|
|
||||||
try:
|
|
||||||
cfg_source, cfg_dest = ConfigObj(source), ConfigObj(destination)
|
|
||||||
for key, val in cfg_source.iteritems():
|
|
||||||
if key not in cfg_dest: cfg_dest[key] = val
|
|
||||||
cfg_dest.write()
|
|
||||||
except Exception:
|
|
||||||
print("Error upgrading")
|
|
||||||
print( traceback.format_exc() )
|
|
Loading…
Reference in New Issue