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

python_apps/media-monitor2/tests/test_upgrade2dot2.py

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

python_apps/media-monitor2/upgrade2.2.py

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

python_apps/media-monitor2/upgrade2dot2.py

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