diff options
| author | Maxime “pep” Buquet <pep@bouah.net> | 2020-05-31 01:20:41 +0200 |
|---|---|---|
| committer | Maxime “pep” Buquet <pep@bouah.net> | 2020-05-31 01:20:41 +0200 |
| commit | 91831e7903bac287be9e5403e56b07691151f1ab (patch) | |
| tree | fc7e4f05916c311cc91dd7d018e7c237ce60ea19 | |
| parent | f670e3ce3b8d69a9fde71301e156f936aaee6fd8 (diff) | |
| download | poezio-91831e7903bac287be9e5403e56b07691151f1ab.tar.gz poezio-91831e7903bac287be9e5403e56b07691151f1ab.tar.bz2 poezio-91831e7903bac287be9e5403e56b07691151f1ab.tar.xz poezio-91831e7903bac287be9e5403e56b07691151f1ab.zip | |
Bookmarks: type bookmark method (local/remote)
Signed-off-by: Maxime “pep” Buquet <pep@bouah.net>
| -rw-r--r-- | poezio/bookmarks.py | 8 | ||||
| -rw-r--r-- | poezio/core/commands.py | 6 |
2 files changed, 8 insertions, 6 deletions
diff --git a/poezio/bookmarks.py b/poezio/bookmarks.py index d842d2dd..46f6dfab 100644 --- a/poezio/bookmarks.py +++ b/poezio/bookmarks.py @@ -30,7 +30,7 @@ Adding a remote bookmark: import functools import logging -from typing import Optional, List, Union +from typing import Optional, List, Literal, Union from slixmpp import InvalidJID, JID from slixmpp.plugins.xep_0048 import Bookmarks, Conference, URL @@ -39,6 +39,8 @@ from poezio.config import config log = logging.getLogger(__name__) +Method = Union[Literal['local'], Literal['remote']] + class Bookmark: def __init__(self, jid: Union[JID, str], @@ -46,7 +48,7 @@ class Bookmark: autojoin=False, nick: Optional[str] = None, password: Optional[str] = None, - method='local') -> None: + method: Method = 'local') -> None: try: if isinstance(jid, JID): self._jid = jid @@ -82,7 +84,7 @@ class Bookmark: return self._method @method.setter - def method(self, value: str): + def method(self, value: Method): if value not in ('local', 'remote'): log.debug('Could not set bookmark storing method: %s', value) return diff --git a/poezio/core/commands.py b/poezio/core/commands.py index 6fc800ae..98acef32 100644 --- a/poezio/core/commands.py +++ b/poezio/core/commands.py @@ -18,7 +18,7 @@ from poezio import common from poezio import pep from poezio import tabs from poezio import multiuserchat as muc -from poezio.bookmarks import Bookmark +from poezio.bookmarks import Bookmark, Method as BookmarkMethod from poezio.common import safeJID from poezio.config import config, DEFAULT_CONFIG, options as config_opts from poezio.contact import Contact, Resource @@ -444,7 +444,7 @@ class CommandCore: self._add_bookmark(jid, autojoin, password, 'remote') - def _add_bookmark(self, jid, autojoin, password, method): + def _add_bookmark(self, jid: str, autojoin: bool, password: str, method: BookmarkMethod) -> None: nick = None if not jid: tab = self.core.tabs.current_tab @@ -478,7 +478,7 @@ class CommandCore: self.core.bookmarks.save_remote(self.core.xmpp, self.core.handler.on_bookmark_result) - def _add_wildcard_bookmarks(self, method): + def _add_wildcard_bookmarks(self, method: BookmarkMethod): new_bookmarks = [] for tab in self.core.get_tabs(tabs.MucTab): bookmark = self.core.bookmarks[tab.jid.bare] |