wcp-library 1.3.4__py3-none-any.whl → 1.3.5__py3-none-any.whl

wcp_library/credentials/{credential_manager_asynchronous.py → _credential_manager_asynchronous.py}

@@ -38,6 +38,8 @@ class AsyncCredentialManager(ABC):
             for field in password['GenericFieldInfo']:
                 password_info[field['DisplayName']] = field['Value'].lower() if field['DisplayName'].lower() == 'username' else field['Value']
             password_dict[password['UserName'].lower()] = password_info
+            if password['OTP']:
+                password_dict[password['UserName'].lower()]['OTP'] = password['OTP']
         logger.debug("Credentials retrieved")
         return password_dict
 
@@ -99,14 +101,24 @@ class AsyncCredentialManager(ABC):
         logger.debug(f"Credentials for {username} retrieved")
         return return_credential
 
+    async def get_credential_by_id(self, password_id: int) -> dict:
+        """
+        Get the credentials for a specific Password ID
+
+        :param password_id:
+        :return:
+        """
+
+        logger.debug(f"Getting credentials for ID {password_id}")
+        credential = await self._get_credential(password_id)
+        logger.debug(f"Credentials for ID {password_id} retrieved")
+        return credential
+
     async def update_credential(self, credentials_dict: dict) -> bool:
         """
         Update username and password in PasswordState
 
-        Credentials dictionary must have the following keys:
-            - PasswordID
-            - UserName
-            - Password
+        Credentials dictionary must the same keys as the original dictionary from the get_credentials method
 
         The dictionary can be obtained from the get_credentials method
 
@@ -114,6 +126,9 @@ class AsyncCredentialManager(ABC):
         :return:
         """
 
+        if "OTP" in credentials_dict:
+            credentials_dict.pop("OTP")
+
         logger.debug(f"Updating credentials for {credentials_dict['UserName']}")
         url = (self.password_url / str(self._password_list_id)).with_query("QueryAll")
         async with aiohttp.ClientSession() as session:

wcp_library/credentials/{credential_manager_synchronous.py → _credential_manager_synchronous.py}

@@ -36,6 +36,8 @@ class CredentialManager(ABC):
             for field in password['GenericFieldInfo']:
                 password_info[field['DisplayName']] = field['Value'].lower() if field['DisplayName'].lower() == 'username' else field['Value']
             password_dict[password['UserName'].lower()] = password_info
+            if password['OTP']:
+                password_dict[password['UserName'].lower()]['OTP'] = password['OTP']
         logger.debug("Credentials retrieved")
         return password_dict
 
@@ -110,10 +112,7 @@ class CredentialManager(ABC):
         """
         Update the credentials for a specific username
 
-        Credentials dictionary must have the following keys:
-            - PasswordID
-            - UserName
-            - Password
+        Credentials dictionary must the same keys as the original dictionary from the get_credentials method
 
         The dictionary should be obtained from the get_credentials method and modified accordingly
 
@@ -121,6 +120,9 @@ class CredentialManager(ABC):
         :return: True if successful, False otherwise
         """
 
+        if "OTP" in credentials_dict:
+            credentials_dict.pop("OTP")
+
         logger.debug(f"Updating credentials for {credentials_dict['UserName']}")
         url = (self.password_url / str(self._password_list_id)).with_query("QueryAll")
         passwords = requests.get(str(url), headers=self.headers).json()

wcp_library/credentials/api.py

@@ -0,0 +1,74 @@
+import logging
+
+from wcp_library.credentials._credential_manager_asynchronous import AsyncCredentialManager
+from wcp_library.credentials._credential_manager_synchronous import CredentialManager
+
+logger = logging.getLogger(__name__)
+
+
+class APICredentialManager(CredentialManager):
+    def __init__(self, passwordState_api_key: str):
+        super().__init__(passwordState_api_key, 214)
+
+    def new_credentials(self, credentials_dict: dict) -> bool:
+        """
+        Create a new credential entry
+
+        Credentials dictionary can have the following keys:
+            - Title
+            - UserName
+            - Password
+            - API KEY
+            - Authentication Header
+            - URL
+
+        :param credentials_dict:
+        :return: True if successful, False otherwise
+        """
+
+        data = {
+            "PasswordListID": self._password_list_id,
+            "Title": credentials_dict['UserName'].upper() if "Title" not in credentials_dict else credentials_dict['Title'].upper(),
+            "Notes": credentials_dict['Notes'] if 'Notes' in credentials_dict else None,
+            "UserName": credentials_dict['UserName'],
+            "Password": credentials_dict['Password'],
+            "GenericField1": credentials_dict['API KEY'],
+            "GenericField2": credentials_dict['Authentication Header'],
+            "URL": credentials_dict['URL']
+        }
+
+        return self._publish_new_password(data)
+
+
+class AsyncAPICredentialManager(AsyncCredentialManager):
+    def __init__(self, passwordState_api_key: str):
+        super().__init__(passwordState_api_key, 214)
+
+    async def new_credentials(self, credentials_dict: dict) -> bool:
+        """
+        Create a new credential entry
+
+        Credentials dictionary can have the following keys:
+            - Title
+            - UserName
+            - Password
+            - API KEY
+            - Authentication Header
+            - URL
+
+        :param credentials_dict:
+        :return:
+        """
+
+        data = {
+            "PasswordListID": self._password_list_id,
+            "Title": credentials_dict['UserName'].upper() if "Title" not in credentials_dict else credentials_dict['Title'].upper(),
+            "Notes": credentials_dict['Notes'] if 'Notes' in credentials_dict else None,
+            "UserName": credentials_dict['UserName'],
+            "Password": credentials_dict['Password'],
+            "GenericField1": credentials_dict['API KEY'],
+            "GenericField2": credentials_dict['Authentication Header'],
+            "URL": credentials_dict['URL']
+        }
+
+        return await self._publish_new_password(data)

wcp_library/credentials/ftp.py

@@ -1,7 +1,7 @@
 import logging
 
-from wcp_library.credentials.credential_manager_asynchronous import AsyncCredentialManager
-from wcp_library.credentials.credential_manager_synchronous import CredentialManager
+from wcp_library.credentials._credential_manager_asynchronous import AsyncCredentialManager
+from wcp_library.credentials._credential_manager_synchronous import CredentialManager
 
 logger = logging.getLogger(__name__)
 

wcp_library/credentials/internet.py

@@ -0,0 +1,66 @@
+import logging
+
+from wcp_library.credentials._credential_manager_asynchronous import AsyncCredentialManager
+from wcp_library.credentials._credential_manager_synchronous import CredentialManager
+
+logger = logging.getLogger(__name__)
+
+
+class InternetCredentialManager(CredentialManager):
+    def __init__(self, passwordState_api_key: str):
+        super().__init__(passwordState_api_key, 93)
+
+    def new_credentials(self, credentials_dict: dict) -> bool:
+        """
+        Create a new credential entry
+
+        Credentials dictionary must have the following keys:
+            - Title
+            - UserName
+            - Password
+            - URL
+
+        :param credentials_dict:
+        :return:
+        """
+
+        data = {
+            "PasswordListID": self._password_list_id,
+            "Title": credentials_dict['UserName'].upper() if "Title" not in credentials_dict else credentials_dict['Title'].upper(),
+            "Notes": credentials_dict['Notes'] if 'Notes' in credentials_dict else None,
+            "UserName": credentials_dict['UserName'].lower(),
+            "Password": credentials_dict['Password'],
+            "URL": credentials_dict['URL']
+        }
+
+        return self._publish_new_password(data)
+
+
+class AsyncInternetCredentialManager(AsyncCredentialManager):
+    def __init__(self, passwordState_api_key: str):
+        super().__init__(passwordState_api_key, 93)
+
+    async def new_credentials(self, credentials_dict: dict) -> bool:
+        """
+        Create a new credential entry
+
+        Credentials dictionary must have the following keys:
+            - Title
+            - UserName
+            - Password
+            - URL
+
+        :param credentials_dict:
+        :return:
+        """
+
+        data = {
+            "PasswordListID": self._password_list_id,
+            "Title": credentials_dict['UserName'].upper() if "Title" not in credentials_dict else credentials_dict['Title'].upper(),
+            "Notes": credentials_dict['Notes'] if 'Notes' in credentials_dict else None,
+            "UserName": credentials_dict['UserName'].lower(),
+            "Password": credentials_dict['Password'],
+            "URL": credentials_dict['URL']
+        }
+
+        return await self._publish_new_password(data)

wcp_library/credentials/oracle.py

@@ -2,8 +2,8 @@ import logging
 
 import aiohttp
 
-from wcp_library.credentials.credential_manager_asynchronous import AsyncCredentialManager
-from wcp_library.credentials.credential_manager_synchronous import CredentialManager
+from wcp_library.credentials._credential_manager_asynchronous import AsyncCredentialManager
+from wcp_library.credentials._credential_manager_synchronous import CredentialManager
 
 logger = logging.getLogger(__name__)
 

wcp_library/credentials/postgres.py

@@ -1,7 +1,7 @@
 import logging
 
-from wcp_library.credentials.credential_manager_asynchronous import AsyncCredentialManager
-from wcp_library.credentials.credential_manager_synchronous import CredentialManager
+from wcp_library.credentials._credential_manager_asynchronous import AsyncCredentialManager
+from wcp_library.credentials._credential_manager_synchronous import CredentialManager
 
 logger = logging.getLogger(__name__)
 

wcp_library/selenium_helper.py

@@ -75,3 +75,14 @@ class SeleniumHelper:
         """
 
         WebDriverWait(self.driver, timeout).until(expected_conditions.presence_of_element_located((By.CSS_SELECTOR, css_element)))
+
+    def wait_until_clickable(self, css_element: str, timeout: int = 30) -> None:
+        """
+        Wait until an element is clickable
+
+        :param css_element:
+        :param timeout:
+        :return:
+        """
+
+        WebDriverWait(self.driver, timeout).until(expected_conditions.element_to_be_clickable((By.CSS_SELECTOR, css_element)))

{wcp_library-1.3.4.dist-info → wcp_library-1.3.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wcp-library
-Version: 1.3.4
+Version: 1.3.5
 Summary: Common utilites for internal development at WCP
 Home-page: https://github.com/Whitecap-DNA/WCP-Library
 Author: Mitch-Petersen

{wcp_library-1.3.4.dist-info → wcp_library-1.3.5.dist-info}/RECORD

@@ -1,21 +1,23 @@
 wcp_library/__init__.py,sha256=hwLbcu00uI6L_xjXO9-I0YcODl2WtIOkdNLoDcXv7zk,591
 wcp_library/credentials/__init__.py,sha256=HRmg7mqcATeclIz3lZQjSR4nmK6aY6MK9-QXEYZoFrw,1857
-wcp_library/credentials/credential_manager_asynchronous.py,sha256=avAvolbtE1qPTTaUas_fMX7Oj_fiBUMutAFztRDDKJI,5722
-wcp_library/credentials/credential_manager_synchronous.py,sha256=ouPNLt20FvocuMoFx49mbDL7-Moj_WrlpR7k97mx-i4,5512
-wcp_library/credentials/ftp.py,sha256=O4oSPtCCv_0w6sLITFnY9EpN6-K2XxeibbGB0VnTHJ4,2589
-wcp_library/credentials/oracle.py,sha256=m0WtmSyUdKUfsz1SPkRgc7A080rK6cq7jVoQ0YcWJ50,2867
-wcp_library/credentials/postgres.py,sha256=tCCWdc10lgdu6zbU-Hv0ZxGw9rn6ZvDvFkMZqm9qfBo,2571
+wcp_library/credentials/_credential_manager_asynchronous.py,sha256=XLYZ6XjLU3Cp18cin3BidIBPvjoqqdJ5L7hhueMCfpU,6311
+wcp_library/credentials/_credential_manager_synchronous.py,sha256=5QyIYYPM2HUlApdZl6ELypuuCioi6Z-JHwyJWFrwYBs,5685
+wcp_library/credentials/api.py,sha256=NZfZcN46-ZVB_-DatR0DLBYAtJxulDlKD2cculzbyXo,2625
+wcp_library/credentials/ftp.py,sha256=lSjOlfU68j9PsOCaoWJSdjbR1QmjTxjVmSxOpMdMG7Q,2591
+wcp_library/credentials/internet.py,sha256=z9tihFddTJcX2pt-iHFP_rkCoheLhTdFgn4ksc2sYYE,2239
+wcp_library/credentials/oracle.py,sha256=OV02op95LNUT21cNDM4zDbD63C1ldiq--vYofDpU-zU,2869
+wcp_library/credentials/postgres.py,sha256=B5qNPKYzuzOJZ5S4M3MCN4xUelvZ6AgyR6QJc4z0Cgo,2573
 wcp_library/emailing.py,sha256=xqNly6Tmj-pvwl5bdys3gauZFDd4SuWCQYzGFNemv2Q,2496
 wcp_library/ftp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wcp_library/ftp/ftp.py,sha256=EpyW0J2QIGxP8zVGD4VarA0hi4C2XAPDPF-0j2sRdpI,4350
 wcp_library/ftp/sftp.py,sha256=hykXGLGdxe7DYAxFdTwjPjTEOYuIpSMyK3NOiTQNUK0,4176
 wcp_library/informatica.py,sha256=_g9lzSd-JLsY4e5_g6YcxQtOxP6fkg4sGfvNJaAN7jY,8123
 wcp_library/logging.py,sha256=e6gG7HFgUrMajUZs4Gs0atFfOJJmdmxN0GerfynNWlY,2061
-wcp_library/selenium_helper.py,sha256=rlphTXsUgnbaXZknY5nfQqxFhnc7UmrpzhV3hQ-cv7k,2509
+wcp_library/selenium_helper.py,sha256=2PENUH8bgpyzovbLjP8a4Oyd7c3yOwoyQJt-a8-zq0c,2855
 wcp_library/sql/__init__.py,sha256=s2psmwkq_ZU23iGWvXjJrLu0hD1fB6CDv6RHcK7y828,1917
 wcp_library/sql/oracle.py,sha256=TGiTC5L5UcM5QcHFajgn43NI8HygOGIEAtLmLbVFp2I,15772
 wcp_library/sql/postgres.py,sha256=GElSjLDworrRlzA7DLrbcQlW7BqvhgMbvW89v-yZejk,14533
 wcp_library/time.py,sha256=ktSzhK7SZnGPNXobFexnhFGQAcriLCJQKxnO0fed8fQ,1740
-wcp_library-1.3.4.dist-info/METADATA,sha256=-3zbu2jB2faCUbBvOLm0I3ikaxEOYbNaxlNaGJY0tUU,1552
-wcp_library-1.3.4.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-wcp_library-1.3.4.dist-info/RECORD,,
+wcp_library-1.3.5.dist-info/METADATA,sha256=hnu1wsVA8QLeWwq7wFNuYu_eZH3R3CqCot1_NWQ4D60,1552
+wcp_library-1.3.5.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+wcp_library-1.3.5.dist-info/RECORD,,