Skip to content

Commit

Permalink
feature/python-sdk-refactor Updated bunq_context according to new MR …
Browse files Browse the repository at this point in the history
…comments. Other classes to follow.
  • Loading branch information
angelomelonas committed Oct 14, 2019
1 parent 2ed259e commit 186f64c
Showing 1 changed file with 4 additions and 18 deletions.
22 changes: 4 additions & 18 deletions bunq/sdk/context/bunq_context.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,3 @@
import bunq
from bunq.sdk.context.api_context import ApiContext
from bunq.sdk.context.user_context import UserContext
from bunq.sdk.exception.bunq_exception import BunqException
Expand All @@ -20,47 +19,34 @@ def __init__(self):
raise TypeError(self._ERROR_CLASS_SHOULD_NOT_BE_INITIALIZED)

@classmethod
def load_api_context(cls, api_context: bunq.sdk.context.api_context.ApiContext) -> None:
def load_api_context(cls, api_context: ApiContext) -> None:
"""
:param api_context:
:type api_context: bunq.sdk.context.api_context.ApiContext
"""
cls._api_context = api_context
cls._user_context = UserContext(api_context.session_context.user_id)
cls._user_context.init_main_monetary_account()

@classmethod
def api_context(cls) -> bunq.sdk.context.api_context.ApiContext:
"""
:return:
:rtype: bunq.sdk.context.api_context.ApiContext
"""
def api_context(cls) -> ApiContext:
if cls._api_context is not None:
return cls._api_context

raise BunqException(cls._ERROR_API_CONTEXT_HAS_NOT_BEEN_LOADED)

@classmethod
def user_context(cls) -> bunq.sdk.context.user_context.UserContext:
"""
:return:
:rtype: bunq.sdk.context.user_context.UserContext
"""

def user_context(cls) -> UserContext:
if cls._user_context is not None:
return cls._user_context

raise BunqException(cls._ERROR_USER_CONTEXT_HAS_NOT_BEEN_LOADED)

@classmethod
def update_api_context(cls, api_context: bunq.sdk.context.api_context.ApiContext) -> None:
def update_api_context(cls, api_context: ApiContext) -> None:
"""
:param api_context:
:type api_context: bunq.sdk.context.api_context.ApiContext
"""

cls._api_context = api_context

0 comments on commit 186f64c

Please sign in to comment.