Get application user with web3 signature

Author: kompotkot

bugout/app.py

@@ -64,22 +64,31 @@ def create_user(
     def get_user(
         self,
         token: Union[str, uuid.UUID],
+        application_id: Optional[Union[str, uuid.UUID]] = None,
         timeout: float = REQUESTS_TIMEOUT,
         auth_type: str = data.AuthType.bearer.name,
     ) -> data.BugoutUser:
         self.user.timeout = timeout
-        return self.user.get_user(token=token, auth_type=data.AuthType[auth_type])
+        return self.user.get_user(
+            token=token,
+            application_id=application_id,
+            auth_type=data.AuthType[auth_type],
+        )
 
     def get_user_by_id(
         self,
         token: Union[str, uuid.UUID],
         user_id: Union[str, uuid.UUID],
+        application_id: Optional[Union[str, uuid.UUID]] = None,
         timeout: float = REQUESTS_TIMEOUT,
         auth_type: str = data.AuthType.bearer.name,
     ) -> data.BugoutUser:
         self.user.timeout = timeout
         return self.user.get_user_by_id(
-            token=token, user_id=user_id, auth_type=data.AuthType[auth_type]
+            token=token,
+            user_id=user_id,
+            application_id=application_id,
+            auth_type=data.AuthType[auth_type],
         )
 
     def find_user(

bugout/settings.py

@@ -12,3 +12,8 @@
     raise Exception(
         f"Could not parse BUGOUT_REQUESTS_TIMEOUT environment variable as int: {REQUESTS_TIMEOUT_RAW}"
     )
+
+# Web3 signature
+BUGOUT_APPLICATION_ID_HEADER = os.environ.get(
+    "BUGOUT_APPLICATION_ID_HEADER", "x-bugout-application-id"
+)

bugout/user.py

@@ -4,7 +4,7 @@
 from .calls import make_request
 from .data import AuthType, BugoutToken, BugoutUser, BugoutUserTokens, Method, TokenType
 from .exceptions import InvalidUrlSpec, TokenInvalidParameters
-from .settings import REQUESTS_TIMEOUT
+from .settings import REQUESTS_TIMEOUT, BUGOUT_APPLICATION_ID_HEADER
 
 
 class User:
@@ -52,25 +52,33 @@ def create_user(
         return BugoutUser(**result)
 
     def get_user(
-        self, token: Union[str, uuid.UUID], auth_type: AuthType = AuthType.bearer
+        self,
+        token: Union[str, uuid.UUID],
+        application_id: Optional[Union[str, uuid.UUID]] = None,
+        auth_type: AuthType = AuthType.bearer,
     ) -> BugoutUser:
         get_user_path = "user"
         headers = {
             "Authorization": f"{auth_type.value} {token}",
         }
+        if auth_type == AuthType.web3 and application_id is not None:
+            headers[BUGOUT_APPLICATION_ID_HEADER] = str(application_id)
         result = self._call(method=Method.get, path=get_user_path, headers=headers)
         return BugoutUser(**result)
 
     def get_user_by_id(
         self,
         token: Union[str, uuid.UUID],
         user_id: Union[str, uuid.UUID],
+        application_id: Optional[Union[str, uuid.UUID]] = None,
         auth_type: AuthType = AuthType.bearer,
     ) -> BugoutUser:
         get_user_by_id_path = f"user/{user_id}"
         headers = {
             "Authorization": f"{auth_type.value} {token}",
         }
+        if auth_type == AuthType.web3 and application_id is not None:
+            headers[BUGOUT_APPLICATION_ID_HEADER] = str(application_id)
         result = self._call(
             method=Method.get, path=get_user_by_id_path, headers=headers
         )