Store configuration to application specific file. Use json as input to config class

Author: vekkt0r

Diff for: config.py

@@ -1,9 +1,6 @@
-import json
-
 class HarmonyConfig(object):
-    def __init__(self, config_file):
-        with open(config_file) as f:
-            self.json = json.loads(f.read())
+    def __init__(self, config):
+        self.json = config
 
     def get_activities(self):
         return self._build_kv_menu('activity')

Diff for: test_config.py

@@ -1,9 +1,11 @@
 import unittest
 import config
+import json
 
 class TestConfig(unittest.TestCase):
     def setUp(self):
-        self.cfg = config.HarmonyConfig('harmony.json')
+        with open('harmony.json') as f:
+            self.cfg = config.HarmonyConfig(json.loads(f.read()))
 
     def test_get_activities(self):
         a = self.cfg.get_activities()

Diff for: ui.py

@@ -7,20 +7,17 @@
 import rumps
 import control
 import config
+import json
 
 class Harmenubar(rumps.App):
     def __init__(self):
         super(Harmenubar,self).__init__('Harmenubar',icon='resources/icon.png')
-        print 'Logging in to Logitech'
-        self.session_token = control.login_to_logitech('[email protected]', 'ankeborg', '192.168.0.158')
-        print 'Connecting client'
-        client = control.get_client('192.168.0.158', self.session_token)
-        print 'Getting current activity'
-        self.activity = client.get_current_activity() # 5337127
         self.activity = 5337127
-        client.disconnect(send_close=True)
+        self.auth()
+
+        cfg = self.get_and_store_config()
 
-        self.cfg = config.HarmonyConfig('harmony.json')
+        self.cfg = config.HarmonyConfig(cfg)
         self.activities = self.cfg.get_activities()
         self.devices = self.cfg.get_devices()
         activity_menu = self.build_activity_menu()
@@ -35,6 +32,24 @@ def __init__(self):
         ]
         self.update_current_activity(self.activity)
 
+    def auth(self):
+        print 'Logging in to Logitech'
+        self.session_token = control.login_to_logitech('[email protected]', 'ankeborg', '192.168.0.158')
+        print 'Connecting client'
+        client = control.get_client('192.168.0.158', self.session_token)
+        print 'Getting current activity'
+        self.activity = client.get_current_activity() # 5337127
+        client.disconnect(send_close=True)
+
+    def get_and_store_config(self):
+        client = control.get_client('192.168.0.158', self.session_token)
+        print 'Getting configuration'
+        cfg = client.get_config()
+        client.disconnect(send_close=True)
+        with self.open('config.json','w') as f:
+            f.write(json.dumps(cfg))
+        return cfg
+
     def build_activity_menu(self):
         menu = []
         for key,value in self.activities.iteritems():