add SecondFactorCode

Author: hwjeremy

procuret/security/__init__.py

@@ -0,0 +1,70 @@
+"""
+Procuret Python
+Second Factor Code Module
+author: hugh@blinkybeach.com
+"""
+from pydoc import plain
+from procuret.ancillary.session_perspective import Perspective
+from typing import Optional
+from procuret.http.method import HTTPMethod
+from procuret.http.api_request import ApiRequest
+
+
+class SecondFactorCode:
+
+    PATH = '/second-factor-code'
+
+    @staticmethod
+    def create_with_email(
+        email: str,
+        plaintext_secret: str,
+        perspective: Optional[Perspective]
+    ) -> None:
+
+        return SecondFactorCode._create(
+            email=email,
+            agent_id=None,
+            plaintext_secret=plaintext_secret,
+            perspective=perspective
+        )
+
+    @staticmethod
+    def create_with_agent_id(
+        agent_id: int,
+        plaintext_secret: str,
+        perspective: Optional[Perspective]
+    ) -> None:
+
+        return SecondFactorCode._create(
+            email=None,
+            agent_id=agent_id,
+            plaintext_secret=plaintext_secret,
+            perspective=perspective
+        )
+
+    @staticmethod
+    def _create(
+        email: Optional[str],
+        agent_id: Optional[int],
+        plaintext_secret: str,
+        perspective: Optional[Perspective]
+    ) -> None:
+
+        data = {
+            'email': email,
+            'plaintext_secret': plaintext_secret,
+            'agent_id': agent_id,
+            'perspective': (
+                perspective.value if perspective is not None else None
+            )
+        }
+
+        ApiRequest.make(
+            path=SecondFactorCode.PATH,
+            method=HTTPMethod.POST,
+            data=data,
+            session=None,
+            query_parameters=None
+        )
+
+        return None

procuret/security/second_factor_code.py

@@ -71,14 +71,16 @@ def create_with_email(
         email: str,
         plaintext_secret: str,
         perspective: Perspective,
+        code: str,
         lifecycle: Lifecycle = Lifecycle.LONG_LIVED
     ) -> Self:
 
         data = {
             'email': email,
             'secret': plaintext_secret,
             'perspective': perspective.value,
-            'lifecycle': lifecycle.value
+            'lifecycle': lifecycle.value,
+            'code': code
         }
 
         result = ApiRequest.make(

procuret/session.py

@@ -6,6 +6,7 @@
 from procuret.tests.test import Test
 from procuret.session import Session, Perspective
 from typing import Optional
+from procuret.security.second_factor_code import SecondFactorCode
 
 
 class TestWithSession(Test):
@@ -23,8 +24,23 @@ def _tws_load_session(self) -> Session:
             if not isinstance(self.test_perspective, Perspective):
                 raise NotImplementedError('Implement .test_perspective')
 
+            SecondFactorCode.create_with_email(
+                email=self.email,
+                plaintext_secret=self.secret,
+                perspective=self.test_perspective
+            )
+
+            code = input('Second factor code: ')
+
+            if not code.isdigit():
+                raise ValueError('Second factor code must be integer')
+            
+            if not len(code) == 6:
+                raise ValueError('Second factor code must be six digits')
+
             self._tws_cached_session = Session.create_with_email(
                 email=self.email,
+                code=code,
                 plaintext_secret=self.secret,
                 perspective=self.test_perspective
             )