mfa send verify

Author: the-veloper

smsapi/sms/api.py

@@ -3,7 +3,7 @@
 from smsapi.exception import EndpointException, SendException
 from smsapi.models import ResultCollection, SendResult, RemoveMessageResult
 from smsapi.sms import response_format_param
-from smsapi.sms.model import SmsSendResult
+from smsapi.sms.model import SmsSendResult, SmsMFASendResult
 from smsapi.utils import join_params
 
 sms_parameters = [
@@ -34,6 +34,19 @@
     'test'
 ]
 
+sms_mfa_parameters = [
+    'phone_number',
+    'from',
+    'from_',
+    'content',
+    'fast'
+]
+
+sms_mfa_verify_parameters = [
+    'phone_number',
+    'code'
+]
+
 fast_force_params = {'fast': 1}
 flash_force_params = {'flash': 1}
 
@@ -55,6 +68,8 @@ def parameters_transformer(_, parameters):
 class Sms(Api):
 
     path = 'sms.do'
+    path_mfa = 'mfa/codes'
+    path_mfa_verify = 'mfa/codes/verifications'
 
     send = bind_api_endpoint(
         method='POST',
@@ -66,6 +81,26 @@ class Sms(Api):
         parameters_transformer=parameters_transformer
     )
 
+    send_mfa = bind_api_endpoint(
+        method='POST',
+        path=path_mfa,
+        mapping=(SendResult, SmsMFASendResult),
+        accept_parameters=sms_mfa_parameters,
+        force_parameters=response_format_param,
+        exception_class=SendException,
+        parameters_transformer=parameters_transformer
+    )
+
+    verify_mfa = bind_api_endpoint(
+        method='POST',
+        path=path_mfa_verify,
+        mapping=(),
+        accept_parameters=sms_mfa_verify_parameters,
+        force_parameters=response_format_param,
+        exception_class=EndpointException,
+        parameters_transformer=parameters_transformer
+    )
+
     send_fast = bind_api_endpoint(
         method='GET',
         path=path,

smsapi/sms/model.py

@@ -26,3 +26,26 @@ def parse(cls, json_response, model):
             collection.append(m)
 
         return cls(count, collection, message=message, length=length, parts=parts)
+
+
+class SmsMFASendResult(ResultCollection):
+
+    def __init__(self, id, code, phone_number, from_):
+        super(SmsMFASendResult, self).__init__(1, [{'id': id, 'code': code, 'phone_number': phone_number, 'from': from_}])
+
+    @classmethod
+    def parse(cls, json_response, model):
+        id = json_response.get('id')
+        code = json_response.get('code')
+        phone_number = json_response.get('phone_number')
+        from_ = json_response.get('from')
+
+        sms_list = [json_response]
+
+        collection = []
+
+        for sms in sms_list:
+            m = model.from_dict(sms)
+            collection.append(m)
+
+        return cls(1, collection, id=id, code=code, phone_number=phone_number, from_=from_)