Add __str__() to Headers

iamdual · Jan 9, 2025 · 3d3e7f7 · 3d3e7f7
1 parent 474c76a
commit 3d3e7f7
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/ua_generator/headers.py b/src/ua_generator/headers.py
@@ -15,7 +15,7 @@ def __init__(self, gen: Generator, ch: ClientHints):
         self.__generator = gen
         self.__client_hints = ch
         self.__is_generated = False
-        self.__headers = {}
+        self.__headers: dict[str, str] = {}
 
     def reset(self):
         self.__is_generated = True
@@ -62,8 +62,14 @@ def accept_ch(self, val: str):
         for hint in requested_hints:
             self.add(hint.strip().lower())
 
-    def get(self):
+    def get(self) -> dict[str, str]:
         if not self.__is_generated:
             self.reset()
 
         return self.__headers
+
+    def __str__(self):
+        text = ""
+        for k, v in self.get().items():
+            text += f"{k}: {v}\n"
+        return text