diff --git a/msal_extensions/windows.py b/msal_extensions/windows.py index 528d51f..0e529f1 100644 --- a/msal_extensions/windows.py +++ b/msal_extensions/windows.py @@ -157,7 +157,7 @@ def add(self, event, **kwargs): super(WindowsTokenCache, self).add(event, **kwargs) self._write() - def update_rt(self, rt_item, new_rt): + def modify(self, credential_type, old_entry, new_key_value_pairs=None): with CrossPlatLock(self._lock_location): if self._needs_refresh(): try: @@ -165,18 +165,10 @@ def update_rt(self, rt_item, new_rt): except IOError as exp: if exp.errno != errno.ENOENT: raise exp - super(WindowsTokenCache, self).update_rt(rt_item, new_rt) - self._write() - - def remove_rt(self, rt_item): - with CrossPlatLock(self._lock_location): - if self._needs_refresh(): - try: - self._read() - except IOError as exp: - if exp.errno != errno.ENOENT: - raise exp - super(WindowsTokenCache, self).remove_rt(rt_item) + super(WindowsTokenCache, self).modify( + credential_type, + old_entry, + new_key_value_pairs=new_key_value_pairs) self._write() def find(self, credential_type, **kwargs): # pylint: disable=arguments-differ diff --git a/setup.py b/setup.py index f69bc89..ce287c8 100644 --- a/setup.py +++ b/setup.py @@ -16,7 +16,7 @@ 'Development Status :: 2 - Pre-Alpha', ], install_requires=[ - 'msal~=0.3', + 'msal~=0.4', 'portalocker~=1.0', ], tests_require=['pytest'],