finished settings manager and socket handler

backend/app/classes/SettingsManager.py

@@ -1,15 +1,23 @@
 import os
 import json
-import cscore
-import networktables
 from .Singleton import Singleton
-from .CamerasHandler import CamerasHandler
+from handlers.CamerasHandler import CamerasHandler
 from .Exceptions import PipelineAlreadyExistsException
 
 
 class SettingsManager(metaclass=Singleton):
     cams = {}
-    settings = {}
+    general_settings = {}
+
+    default_pipeline = {
+        "exposure": 50,
+        "brightness": 50
+    }
+    default_general_settings = {
+        "team_number": 1577,
+        "curr_camera": "cam1",
+        "curr_pipeline": "pipeline1"
+    }
 
     def __init__(self):
         self.settings_path = os.path.join(os.getcwd(), "settings")
@@ -17,20 +25,19 @@ class SettingsManager(metaclass=Singleton):
         self._init_general_settings()
         self._init_cameras()
 
+        if not self.cams[self.general_settings["curr_camera"]]:
+            self.general_settings["curr_camera"] = list(self.cams.keys())[0]
+
     def _init_general_settings(self):
         try:
             with open(os.path.join(self.settings_path, 'settings.json')) as setting_file:
-                self.settings = json.load(setting_file)
+                self.general_settings = json.load(setting_file)
         except FileNotFoundError:
-            self.settings = {
-                "team_number": 1577,
-                "curr_camera": "cam1",
-                "curr_pipeline": "pipeline1"
-            }
+            self.general_settings = self.default_general_settings.copy()
 
     def _init_cameras(self):
         cameras = CamerasHandler.get_cameras()
-        
+
         for cam in cameras:
             if os.path.exists(os.path.join(self.cams_path, cam.name)):
                 self.cams[cam.name] = {}
@@ -50,25 +57,42 @@ class SettingsManager(metaclass=Singleton):
     # Access methods
 
     def get_curr_pipeline(self):
-        return self.cams[self.settings["curr_camera"]]["pipelines"][self.settings["curr_pipeline"]]
+        return self.cams[self.general_settings["curr_camera"]]["pipelines"][self.general_settings["curr_pipeline"]]
 
     def get_curr_cam(self):
-        return self.cams[self.settings["curr_camera"]]
+        return self.cams[self.general_settings["curr_camera"]]
 
     def set_curr_camera(self, cam_name):
         if cam_name in self.cams:
-            self.settings["curr_camera"] = cam_name
+            self.general_settings["curr_camera"] = cam_name
 
     def set_curr_pipeline(self, pipe_name):
-        if pipe_name in self.cams:
-            self.settings["curr_pipeline"] = pipe_name
+        if pipe_name in self.get_curr_cam()["pipelines"]:
+            self.general_settings["curr_pipeline"] = pipe_name
+
+    def change_pipeline_values(self, dic, cam_name=None, pipe_name=None):
+
+        if not cam_name:
+            cam_name = self.general_settings["curr_camera"]
+
+        if not pipe_name:
+            pipe_name = self.general_settings["curr_pipeline"]
+
+        for key in dic:
+            if self.default_pipeline[key]:
+                self.cams[cam_name]["pipelines"][pipe_name][key] = dic[key]
+
+    def change_general_settings_values(self, dic):
+        for key in dic:
+            if self.default_general_settings[key]:
+                self.general_settings[key] = dic[key]
 
     # Creators
 
     def create_new_pipeline(self, pipe_name=None, cam_name=None):
 
         if not cam_name:
-            cam_name = self.settings["curr_camera"]
+            cam_name = self.general_settings["curr_camera"]
 
         if not pipe_name:
             suffix = 0
@@ -80,10 +104,7 @@ class SettingsManager(metaclass=Singleton):
         elif self.cams[cam_name]["pipelines"][pipe_name]:
             raise PipelineAlreadyExistsException(pipe_name)
 
-        self.cams[cam_name]["pipelines"][pipe_name] = {
-            "exposure": 50,
-            "brightness": 50
-        }
+        self.cams[cam_name]["pipelines"][pipe_name] = self.default_pipeline.copy()
 
     def create_new_cam(self, cam_name):
         self.cams[cam_name] = {}
@@ -92,7 +113,11 @@ class SettingsManager(metaclass=Singleton):
 
     # Savers
 
-    def save_cameras(self):
+    def save_settings(self):
+        self._save_cameras()
+        self._save_general_settings()
+
+    def _save_cameras(self):
         for cam in self.cams:
             for pipeline in self.cams[cam]:
                 path = os.path.join(self.cams_path, cam)
@@ -103,6 +128,6 @@ class SettingsManager(metaclass=Singleton):
                 with open(os.path.join(path, pipeline + '.json'), 'w+') as pipeline_file:
                     json.dump(self.cams[cam][pipeline], pipeline_file)
 
-    def save_settings(self):
+    def _save_general_settings(self):
         with open(os.path.join(self.settings_path, 'settings.json'), 'w+') as setting_file:
-            json.dump(self.settings, setting_file)
+            json.dump(self.general_settings, setting_file)