From 7d0cc5e63030eacf72019e81bb1ad009f2f6735d Mon Sep 17 00:00:00 2001
From: dettlaff <dettlaff@riseup.net>
Date: Sat, 18 May 2024 06:32:56 +0400
Subject: [PATCH] feat: add type hints

---
 selfprivacy_api/services/flake_service_manager.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/selfprivacy_api/services/flake_service_manager.py b/selfprivacy_api/services/flake_service_manager.py
index cb638ba..6538c8e 100644
--- a/selfprivacy_api/services/flake_service_manager.py
+++ b/selfprivacy_api/services/flake_service_manager.py
@@ -1,10 +1,11 @@
 import re
+from typing import Dict, Tuple, Optional
 
 FLAKE_CONFIG_PATH = "/etc/nixos/sp-modules/flake.nix"
 
 
 class FlakeServiceManager:
-    def __enter__(self):
+    def __enter__(self) -> "FlakeServiceManager":
         self.services = {}
 
         with open(FLAKE_CONFIG_PATH, "r") as file:
@@ -15,7 +16,9 @@ class FlakeServiceManager:
 
         return self
 
-    def _extract_services(self, input_string: str):
+    def _extract_services(
+        self, input_string: str
+    ) -> Tuple[Optional[str], Optional[str]]:
         pattern = r"inputs\.(\w+)\.url\s*=\s*(\S+);"
         match = re.search(pattern, input_string)
 
@@ -26,7 +29,7 @@ class FlakeServiceManager:
         else:
             return None, None
 
-    def __exit__(self, exc_type, exc_value, traceback):
+    def __exit__(self, exc_type, exc_value, traceback) -> None:
         with open(FLAKE_CONFIG_PATH, "w") as file:
             file.write(
                 """