Skip to content

Commit d18bdbc

Browse files
author
Kevin Hellemun
committed
Added method to refresh user context data. (#79)
1 parent 414a72a commit d18bdbc

File tree

1 file changed

+13
-2
lines changed

1 file changed

+13
-2
lines changed

bunq/sdk/context.py

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -428,8 +428,15 @@ def __init__(self, user_id):
428428
self._user_company = None
429429
self._primary_monetary_account = None
430430

431-
user_object = endpoint.User.list().value[0].get_referenced_object()
432-
self._set_user(user_object)
431+
self._set_user(self.__get_user_object())
432+
433+
@staticmethod
434+
def __get_user_object():
435+
"""
436+
:rtype: core.BunqModel
437+
"""
438+
439+
return endpoint.User.list().value[0].get_referenced_object()
433440

434441
def _set_user(self, user):
435442
if isinstance(user, endpoint.UserPerson):
@@ -478,6 +485,10 @@ def is_both_user_type_set(self):
478485

479486
return self._user_company is not None and self._user_person is not None
480487

488+
def refresh_user_context(self):
489+
self._set_user(self.__get_user_object())
490+
self.init_main_monetary_account()
491+
481492
@property
482493
def user_company(self):
483494
"""

0 commit comments

Comments
 (0)