fixed typing issues

ly4k · Feb 21, 2022 · aa8bbfa · aa8bbfa
1 parent 447208d
commit aa8bbfa
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/certipy/find.py b/certipy/find.py
@@ -530,7 +530,7 @@ def find(self):
 
     def security_to_bloodhound_aces(
         self, security: ActiveDirectorySecurity, ca: bool = False
-    ) -> list:
+    ) -> List:
         aces = []
 
         owner = self.lookup_sid(security.owner)

diff --git a/certipy/shadow.py b/certipy/shadow.py
@@ -51,7 +51,7 @@ def connection(self) -> LDAPConnection:
 
         return self._connection
 
-    def get_key_credentials(self, target_dn: str, user: LDAPEntry) -> list[bytes]:
+    def get_key_credentials(self, target_dn: str, user: LDAPEntry) -> List[bytes]:
         results = self.connection.search(
             search_base=target_dn,
             search_filter="(objectClass=*)",

diff --git a/certipy/template.py b/certipy/template.py
@@ -1,7 +1,7 @@
 import argparse
 import json
 import logging
-from typing import Callable, Tuple
+from typing import Callable, Dict, Tuple
 
 import ldap3
 from ldap3.protocol.microsoft import security_descriptor_control
@@ -128,7 +128,7 @@ def get_configuration(self, template) -> LDAPEntry:
 
         return template
 
-    def json_to_configuration(self, configuration_json: str) -> dict:
+    def json_to_configuration(self, configuration_json: str) -> Dict:
         output = {}
         for key, value in configuration_json.items():
             if key in PROTECTED_ATTRIBUTES:
@@ -141,7 +141,7 @@ def json_to_configuration(self, configuration_json: str) -> dict:
 
         return output
 
-    def load_configuration(self, configuration: str) -> dict:
+    def load_configuration(self, configuration: str) -> Dict:
         with open(configuration, "r") as f:
             configuration_json = json.load(f)