init sdk

Author: aliubymov

Diff for: .gitignore

@@ -0,0 +1,17 @@
+MIT License
+Copyright (c) 2021 Exadel
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

Diff for: .idea/.gitignore

@@ -0,0 +1,3 @@
+# -*- coding: utf-8 -*-
+
+from .core import CompreFace

Diff for: .idea/compreface-python-sdk.iml

@@ -0,0 +1,6 @@
+# -*- coding: utf-8 -*-
+
+from .verify_face_from_image import VerifyFaceFromImageClient
+from .add_example_of_subject import AddExampleOfSubjectClient
+from .delete_example_by_id import DeleteExampleByIdClient
+from .recognize_face_from_image import RecognizeFaceFromImageClient

Diff for: .idea/inspectionProfiles/Project_Default.xml

@@ -0,0 +1,39 @@
+# -*- coding: utf-8 -*-
+
+import os
+import requests
+from requests_toolbelt.multipart.encoder import MultipartEncoder
+
+from ..common import ClientRequest
+
+
+class AddExampleOfSubjectClient(ClientRequest):
+
+    def __init__(self, api_key: str, domain: str, port: str):
+        super().__init__()
+        self.client_url: str = '/api/v1/faces'
+        self.api_key: str = api_key
+        self.url: str = domain + ':' + port + self.client_url
+
+    def get(self) -> dict:
+        url: str = self.url
+        result = requests.get(url, headers={'x-api-key': self.api_key})
+        return result.json()
+
+    def post(self, image_path: str = '', subject: str = '') -> dict:
+        url: str = self.url + '?subject=' + subject
+        name_img: str = os.path.basename(image_path)
+        m = MultipartEncoder(
+            fields={'file': (name_img, open(image_path, 'rb'))}
+        )
+        result = requests.post(url, data=m, headers={'Content-Type': m.content_type,
+                                                     'x-api-key': self.api_key})
+        return result.json()
+
+    def put(self):
+        pass
+
+    def delete(self, subject: str = ''):
+        url: str = self.url + '?subject=' + subject
+        result = requests.delete(url, headers={'x-api-key': self.api_key})
+        return result.json()

Diff for: .idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,28 @@
+# -*- coding: utf-8 -*-
+
+import requests
+
+from ..common import ClientRequest
+
+
+class DeleteExampleByIdClient(ClientRequest):
+
+    def __init__(self, api_key: str, domain: str, port: str):
+        super().__init__()
+        self.client_url: str = '/api/v1/faces/'
+        self.api_key: str = api_key
+        self.url: str = domain + ':' + port + self.client_url
+
+    def get(self):
+        pass
+
+    def post(self):
+        pass
+
+    def put(self):
+        pass
+
+    def delete(self, image_id: str = ''):
+        url: str = self.url + image_id
+        result = requests.delete(url, headers={'x-api-key': self.api_key})
+        return result.json()

Diff for: .idea/misc.xml

@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+
+import os
+import requests
+from requests_toolbelt.multipart.encoder import MultipartEncoder
+
+from ..common import ClientRequest
+
+
+class RecognizeFaceFromImageClient(ClientRequest):
+
+    def __init__(self, api_key: str, domain: str, port: str):
+        super().__init__()
+        self.client_url: str = '/api/v1/faces/recognize'
+        self.api_key: str = api_key
+        self.url: str = domain + ':' + port + self.client_url
+
+    def get(self):
+        pass
+
+    def post(self, image_path: str = '',
+             limit: float = 0,
+             det_prob_threshold: float = 0,
+             prediction_count: int = 0):
+        url: str = self.url + '?limit=' + str(limit) + '&prediction_count=' + str(prediction_count) \
+                   + '&det_prob_threshold=' + \
+                   str(det_prob_threshold)
+        name_img: str = os.path.basename(image_path)
+        m = MultipartEncoder(
+            fields={'file': (name_img, open(image_path, 'rb'))}
+        )
+        result = requests.post(url, data=m, headers={'Content-Type': m.content_type,
+                                                     'x-api-key': self.api_key})
+        return result.json()
+
+    def put(self):
+        pass
+
+    def delete(self):
+        pass

Diff for: .idea/modules.xml

@@ -0,0 +1,24 @@
+import requests
+import os
+from requests_toolbelt.multipart.encoder import MultipartEncoder
+
+url_view = "http://localhost:8000/api/v1/faces"
+api_key = "7dacfc8e-1bb1-4fcf-a9b1-76e4d9d89855"
+headers = {"Content-Type": "application/x-www-form-urlencoded", "x-api-key": api_key}
+headers_api_key = {"x-api-key": api_key}
+path = 'image_path'
+
+
+def add(path_img, subject):
+    url = 'http://localhost:8000/api/v1/faces?subject=' + subject
+    name_img = os.path.basename(path_img)
+    m = MultipartEncoder(
+        fields={'file': (name_img, open(path_img, 'rb'))}
+    )
+
+    r = requests.post(url, data=m, headers={'Content-Type': m.content_type,
+                                            'x-api-key': api_key})
+
+    print(r.status_code)
+    print(r.text)
+    print(r.json())

Diff for: .idea/vcs.xml

@@ -0,0 +1,41 @@
+# -*- coding: utf-8 -*-
+
+import os
+import requests
+from requests_toolbelt.multipart.encoder import MultipartEncoder
+
+from ..common import ClientRequest
+
+
+class VerifyFaceFromImageClient(ClientRequest):
+
+    def __init__(self, api_key: str, domain: str, port: str):
+        super().__init__()
+        self.client_url: str = '/api/v1/faces/'
+        self.api_key: str = api_key
+        self.url: str = domain + ':' + port + self.client_url
+
+    def get(self):
+        pass
+
+    def post(self,
+             image_path: str = '',
+             image_id: str = '',
+             limit: float = 0,
+             det_prob_threshold: float = 0,):
+        url: str = self.url + image_id + '/verify' +'?limit=' + str(limit) + \
+                   '&det_prob_threshold=' + \
+                   str(det_prob_threshold)
+        name_img: str = os.path.basename(image_path)
+        m = MultipartEncoder(
+            fields={'file': (name_img, open(image_path, 'rb'))}
+        )
+        result = requests.post(url, data=m, headers={'Content-Type': m.content_type,
+                                                     'x-api-key': self.api_key})
+        return result.json()
+
+    def put(self):
+        pass
+
+    def delete(self):
+        pass

Diff for: LICENSE.txt

@@ -0,0 +1,4 @@
+# -*- coding: utf-8 -*-
+
+from .client import ClientRequest
+from .service import Service

Diff for: compreface/__init__.py

@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+
+from abc import ABC, abstractmethod
+
+
+class ClientRequest(ABC):
+    """The best class of all requests"""
+
+    @abstractmethod
+    def __init__(self):
+        pass
+
+    @abstractmethod
+    def get(self):
+        pass
+
+    @abstractmethod
+    def post(self):
+        pass
+
+    @abstractmethod
+    def put(self):
+        pass
+
+    @abstractmethod
+    def delete(self):
+        pass

Diff for: compreface/__pycache__/__init__.cpython-38.pyc

@@ -0,0 +1,19 @@
+# -*- coding: utf-8 -*-
+
+from abc import ABC, abstractmethod
+
+
+class Service(ABC):
+    """The best class of all services"""
+
+    @abstractmethod
+    def __init__(self, api_key: str):
+        self._api_key = api_key
+
+    @property
+    def api_key(self):
+        return self._api_key
+
+    @abstractmethod
+    def get_available_functions(self):
+        pass

Diff for: compreface/client/__init__.py

@@ -0,0 +1,3 @@
+# -*- coding: utf-8 -*-
+
+from .model import CompreFace

Diff for: compreface/client/__pycache__/__init__.cpython-38.pyc

@@ -0,0 +1,68 @@
+# -*- coding: utf-8 -*-
+
+from typing import Optional
+from ..service import (
+    RecognitionService,
+    VerificationService,
+    DetectionService
+)
+
+
+class CompreFace(object):
+    """
+    Main class
+    """
+
+    def __init__(self, domain: str, port: str):
+        self._domain: str = domain
+        self._port: str = port
+        self.recognition: Optional[RecognitionService] = None
+        self.verification: Optional[VerificationService] = None
+        self.detection: Optional[DetectionService] = None
+
+    @property
+    def domain(self):
+        return self._domain
+
+    @domain.setter
+    def domain(self, domain: str):
+        self._domain = domain
+
+    @property
+    def port(self):
+        return self._port
+
+    @port.setter
+    def port(self, port: str):
+        self._port = port
+
+    def init_face_recognition(self, api_key: str) -> RecognitionService:
+        """
+        Init Face Recognition Service
+        :param api_key:
+        :return:
+        """
+        self.recognition = RecognitionService(api_key=api_key,
+                                              domain=self.domain,
+                                              port=self.port)
+        return self.recognition
+
+    def init_face_verification(self, api_key: str) -> VerificationService:
+        """
+        Init Face Verification Service
+        :param api_key:
+        :return:
+        """
+        self.verification = VerificationService(api_key=api_key,
+                                              domain=self.domain,
+                                              port=self.port)
+        return self.verification
+
+    def init_face_detection(self, api_key: str) -> DetectionService:
+        """
+        Init Face Detection Service
+        :param api_key:
+        :return:
+        """
+        self.detection = DetectionService(api_key=api_key)
+        return self.detection

Diff for: compreface/client/__pycache__/add_example_of_subject.cpython-38.pyc

@@ -0,0 +1,5 @@
+# -*- coding: utf-8 -*-
+
+from .detection_service import DetectionService
+from .verification_service import VerificationService
+from .recognition_service import RecognitionService