Merge pull request #8 from alice-biometrics/feature/add-document-source

Add optional argument source to add_document

alice/onboarding/document_source.py

@@ -0,0 +1,6 @@
+from enum import Enum
+
+
+class DocumentSource(Enum):
+    file = "file"
+    camera = "camera"

alice/onboarding/onboarding.py

@@ -3,6 +3,7 @@
 from meiga import Result, Success, Failure, isSuccess
 
 from alice.config import Config
+from alice.onboarding.document_source import DocumentSource
 from alice.onboarding.onboarding_errors import OnboardingError
 from alice.onboarding.user_info import UserInfo
 from alice.onboarding.device_info import DeviceInfo
@@ -518,6 +519,7 @@ def add_document(
         media_data: bytes,
         side: str,
         manual: bool = False,
+        source: DocumentSource = None,
         fields: dict = None,
         verbose: bool = False,
     ) -> Result[bool, OnboardingError]:
@@ -537,6 +539,8 @@ def add_document(
             Side of the document [front, back or internal]
         manual
             If True defines manual document uploading
+        source
+            Source of the media: camera or file
         fields
             Fields to add regardless of the OCR process
         verbose
@@ -554,6 +558,7 @@ def add_document(
             media_data=media_data,
             side=side,
             manual=manual,
+            source=source,
             fields=fields,
             verbose=verbose,
         )

alice/onboarding/onboarding_client.py

@@ -4,6 +4,7 @@
 from requests import request, Response
 
 from alice.auth.auth import Auth
+from alice.onboarding.document_source import DocumentSource
 from alice.onboarding.tools import timeit, print_intro, print_response, print_token
 
 from alice.onboarding.user_info import UserInfo
@@ -542,6 +543,7 @@ def add_document(
         media_data: bytes,
         side: str,
         manual: bool = False,
+        source: DocumentSource = None,
         fields: dict = None,
         verbose: bool = False,
     ) -> Response:
@@ -562,6 +564,8 @@ def add_document(
             Side of the document [front or back]
         manual
             If True defines manual document uploading
+        source
+            Source of the media: camera or file
         fields
             Fields to add regardless of the OCR process
         verbose
@@ -585,6 +589,9 @@ def add_document(
             "fields": json.dumps(fields),
         }
 
+        if source:
+            data["source"] = source.value
+
         files = {"image": ("image", media_data)}
 
         response = request(