diff --git a/vban_cmd/util.py b/vban_cmd/util.py index 924c49c..0a9f601 100644 --- a/vban_cmd/util.py +++ b/vban_cmd/util.py @@ -7,9 +7,8 @@ def cache_bool(func, param): def wrapper(*args, **kwargs): self, *rem = args - cmd = f"{self.identifier}.{param}" - if cmd in self._remote.cache: - return self._remote.cache.pop(cmd) == 1 + if self._cmd(param) in self._remote.cache: + return self._remote.cache.pop(self._cmd(param)) == 1 if self._remote.sync: self._remote.clear_dirty() return func(*args, **kwargs) @@ -22,9 +21,8 @@ def cache_string(func, param): def wrapper(*args, **kwargs): self, *rem = args - cmd = f"{self.identifier}.{param}" - if cmd in self._remote.cache: - return self._remote.cache.pop(cmd) + if self._cmd(param) in self._remote.cache: + return self._remote.cache.pop(self._cmd(param)) if self._remote.sync: self._remote.clear_dirty() return func(*args, **kwargs)