Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

fix some CAP related type hints

Browse source
This commit is contained in:
jesopo 2019-06-02 10:14:09 +01:00
parent f9f637e084
commit 5204ac3300
2 changed files with 8 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
src/IRCServer.py
View file

@ -288,7 +288,8 @@ class Server(IRCObject.Object):
return bool(self.available_capability(capability))
def has_capability_str(self, capability: str) -> bool:
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)
def waiting_for_capabilities(self) -> bool:

10
src/utils/irc/__init__.py
View file

@ -270,10 +270,12 @@ class IRCBatch(object):
return self._lines
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._on_ack_callbacks = []
def available(self, capabilities: typing.Iterable[str]) -> str:
self._on_ack_callbacks = [
] # type: typing.List[typing.Callable[[], None]]
def available(self, capabilities: typing.Iterable[str]
) -> typing.Optional[str]:
match = list(set(capabilities)&self._caps)
return match[0] if match else None
@ -289,7 +291,7 @@ class Capability(object):
pass
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])
def get_value(self, tags: typing.Dict[str, str]) -> typing.Optional[str]:
key = list(set(tags.keys())&self._names)