summaryrefslogtreecommitdiff
path: root/poezio/bookmarks.py
diff options
context:
space:
mode:
authorMaxime “pep” Buquet <pep@bouah.net>2020-05-31 01:20:41 +0200
committerMaxime “pep” Buquet <pep@bouah.net>2020-05-31 01:20:41 +0200
commit91831e7903bac287be9e5403e56b07691151f1ab (patch)
treefc7e4f05916c311cc91dd7d018e7c237ce60ea19 /poezio/bookmarks.py
parentf670e3ce3b8d69a9fde71301e156f936aaee6fd8 (diff)
downloadpoezio-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>
Diffstat (limited to 'poezio/bookmarks.py')
-rw-r--r--poezio/bookmarks.py8
1 files changed, 5 insertions, 3 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