Updated core models. (#75)

Author: Kevin Hellemun

bunq/sdk/model/core.py

@@ -1,5 +1,7 @@
 from bunq.sdk import client
 from bunq.sdk.json import converter
+from bunq.sdk.exception import BunqException
+from bunq.sdk import context
 
 
 class AnchoredObjectInterface:
@@ -135,6 +137,35 @@ def _from_json_list(cls, response_raw, wrapper=None):
         return client.BunqResponse(array_deserialized, response_raw.headers,
                                    pagination)
 
+    @classmethod
+    def _get_api_context(cls):
+        """
+        :rtype: context.ApiContext
+        """
+
+        return context.BunqContext.api_context()
+
+    @classmethod
+    def _determine_user_id(cls):
+        """
+        :rtype: int
+        """
+
+        return context.BunqContext.user_context().user_id
+
+    @classmethod
+    def _determine_monetary_account_id(cls, monetary_account_id=None):
+        """
+        :type monetary_account_id: int
+
+        :rtype: int
+        """
+
+        if monetary_account_id is None:
+            return context.BunqContext.user_context().primary_monetary_account.id_
+
+        return monetary_account_id
+
 
 class Id(BunqModel):
     """
@@ -366,6 +397,9 @@ class SessionServer(BunqModel):
     # Field constants
     FIELD_SECRET = "secret"
 
+    # Error constants
+    _ERROR_ALL_FIELD_IS_NULL = 'All fields are null'
+
     def __init__(self):
         self._id_ = None
         self._token = None
@@ -442,3 +476,16 @@ def is_all_field_none(self):
             return False
 
         return True
+
+    def get_referenced_object(self):
+        """
+        :rtype: BunqModel
+        """
+
+        if self._user_person is not None:
+            return self._user_person
+
+        if self._user_company is not None:
+            return self._user_company
+
+        raise BunqException(self._ERROR_ALL_FIELD_IS_NULL)

bunq/sdk/model/device_server_internal.py

@@ -0,0 +1,60 @@
+from bunq.sdk.model.generated.endpoint import DeviceServer
+from bunq.sdk.model.generated.endpoint import BunqResponseInt
+from bunq.sdk import client
+from bunq.sdk.json import converter
+from bunq.sdk.exception import BunqException
+
+
+class DeviceServerInternal(DeviceServer):
+    _ERROR_API_CONTEXT_IS_NULL = 'ApiContext should not be None,' \
+                                 ' use the generated class instead.'
+
+    @classmethod
+    def create(cls, description, secret, permitted_ips=None,
+               custom_headers=None, api_context=None):
+        """
+        Create a new DeviceServer providing the installation token in the header
+        and signing the request with the private part of the key you used to
+        create the installation. The API Key that you are using will be bound to
+        the IP address of the DeviceServer which you have
+        created.<br/><br/>Using a Wildcard API Key gives you the freedom to make
+        API calls even if the IP address has changed after the POST
+        device-server.<br/><br/>Find out more at this link <a
+        href="https://bunq.com/en/apikey-dynamic-ip"
+        target="_blank">https://bunq.com/en/apikey-dynamic-ip</a>.
+
+        :param description: The description of the DeviceServer. This is only
+        for your own reference when reading the DeviceServer again.
+        :type description: str
+        :param secret: The API key. You can request an API key in the bunq app.
+        :type secret: str
+        :param permitted_ips: An array of IPs (v4 or v6) this DeviceServer will
+        be able to do calls from. These will be linked to the API key.
+        :type permitted_ips: list[str]
+        :type custom_headers: dict[str, str]|None
+        :type api_context: context.ApiContext
+
+        :rtype: BunqResponseInt
+        """
+
+        if api_context is None:
+            raise BunqException(cls._ERROR_API_CONTEXT_IS_NULL)
+
+        if custom_headers is None:
+            custom_headers = {}
+
+        request_map = {
+            cls.FIELD_DESCRIPTION: description,
+            cls.FIELD_SECRET: secret,
+            cls.FIELD_PERMITTED_IPS: permitted_ips
+        }
+
+        api_client = client.ApiClient(api_context)
+        request_bytes = converter.class_to_json(request_map).encode()
+        endpoint_url = cls._ENDPOINT_URL_CREATE
+        response_raw = api_client.post(endpoint_url, request_bytes,
+                                       custom_headers)
+
+        return BunqResponseInt.cast_from_bunq_response(
+            cls._process_for_id(response_raw)
+        )