fix some CAP related type hints
This commit is contained in:
parent
f9f637e084
commit
5204ac3300
2 changed files with 8 additions and 5 deletions
src
|
@ -288,7 +288,8 @@ class Server(IRCObject.Object):
|
||||||
return bool(self.available_capability(capability))
|
return bool(self.available_capability(capability))
|
||||||
def has_capability_str(self, capability: str) -> bool:
|
def has_capability_str(self, capability: str) -> bool:
|
||||||
return capability in self.agreed_capabilities
|
return capability in self.agreed_capabilities
|
||||||
def available_capability(self, capability: utils.irc.Capability) -> str:
|
def available_capability(self, capability: utils.irc.Capability
|
||||||
|
) -> typing.Optional[str]:
|
||||||
return capability.available(self.agreed_capabilities)
|
return capability.available(self.agreed_capabilities)
|
||||||
|
|
||||||
def waiting_for_capabilities(self) -> bool:
|
def waiting_for_capabilities(self) -> bool:
|
||||||
|
|
|
@ -270,10 +270,12 @@ class IRCBatch(object):
|
||||||
return self._lines
|
return self._lines
|
||||||
|
|
||||||
class Capability(object):
|
class Capability(object):
|
||||||
def __init__(self, name, draft_name=None):
|
def __init__(self, name: typing.Optional[str], draft_name: str=None):
|
||||||
self._caps = set([name, draft_name])
|
self._caps = set([name, draft_name])
|
||||||
self._on_ack_callbacks = []
|
self._on_ack_callbacks = [
|
||||||
def available(self, capabilities: typing.Iterable[str]) -> str:
|
] # type: typing.List[typing.Callable[[], None]]
|
||||||
|
def available(self, capabilities: typing.Iterable[str]
|
||||||
|
) -> typing.Optional[str]:
|
||||||
match = list(set(capabilities)&self._caps)
|
match = list(set(capabilities)&self._caps)
|
||||||
return match[0] if match else None
|
return match[0] if match else None
|
||||||
|
|
||||||
|
@ -289,7 +291,7 @@ class Capability(object):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
class MessageTag(object):
|
class MessageTag(object):
|
||||||
def __init__(self, name: str, draft_name: str=None):
|
def __init__(self, name: typing.Optional[str], draft_name: str=None):
|
||||||
self._names = set([name, draft_name])
|
self._names = set([name, draft_name])
|
||||||
def get_value(self, tags: typing.Dict[str, str]) -> typing.Optional[str]:
|
def get_value(self, tags: typing.Dict[str, str]) -> typing.Optional[str]:
|
||||||
key = list(set(tags.keys())&self._names)
|
key = list(set(tags.keys())&self._names)
|
||||||
|
|
Loading…
Add table
Reference in a new issue