refactoring

1 files changed, 43 insertions, 0 deletions

diff --git a/kvmd/tools.py b/kvmd/tools.py
new file mode 100644
index 00000000..1259893c
--- /dev/null
+++ b/kvmd/tools.py
@@ -0,0 +1,43 @@
+# ========================================================================== #
+#                                                                            #
+#    KVMD - The main Pi-KVM daemon.                                          #
+#                                                                            #
+#    Copyright (C) 2018  Maxim Devaev <[email protected]>                    #
+#                                                                            #
+#    This program is free software: you can redistribute it and/or modify    #
+#    it under the terms of the GNU General Public License as published by    #
+#    the Free Software Foundation, either version 3 of the License, or       #
+#    (at your option) any later version.                                     #
+#                                                                            #
+#    This program is distributed in the hope that it will be useful,         #
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of          #
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the           #
+#    GNU General Public License for more details.                            #
+#                                                                            #
+#    You should have received a copy of the GNU General Public License       #
+#    along with this program.  If not, see <https://www.gnu.org/licenses/>.  #
+#                                                                            #
+# ========================================================================== #
+
+
+import functools
+
+from typing import Dict
+from typing import Hashable
+
+
+# =====
+def merge(dest: Dict, src: Dict) -> None:
+    for key in src:
+        if key in dest:
+            if isinstance(dest[key], dict) and isinstance(src[key], dict):
+                merge(dest[key], src[key])
+                continue
+        dest[key] = src[key]
+
+
+def rget(dct: Dict, *keys: Hashable) -> Dict:
+    result = functools.reduce((lambda nxt, key: nxt.get(key, {})), keys, dct)
+    if not isinstance(result, dict):
+        raise TypeError(f"Not a dict as result: {result!r} from {dct!r} at {list(keys)}")
+    return result