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

Switch to using utils.irc.Capability in sts.py

Browse source
This commit is contained in:
jesopo 2019-05-15 10:31:55 +01:00
parent a80eda614d
commit 99ce2ac6dd
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
modules/sts.py
View file

@ -1,6 +1,8 @@
import time import time
from src import ModuleManager, utils from src import ModuleManager, utils
CAP = utils.irc.Capability("sts", "draft/sts")
class Module(ModuleManager.BaseModule): class Module(ModuleManager.BaseModule):
def _get_policy(self, server): def _get_policy(self, server):
return server.get_setting("sts-policy", None) return server.get_setting("sts-policy", None)
@ -26,14 +28,12 @@ class Module(ModuleManager.BaseModule):
port = int(info["port"]) port = int(info["port"])
self.set_policy(server, port, duration) self.set_policy(server, port, duration)
def _get_sts(self, capabilities):
return capabilities.get("sts", capabilities.get("draft/sts", None))
@utils.hook("received.cap.ls") @utils.hook("received.cap.ls")
def on_cap_ls(self, event): def on_cap_ls(self, event):
sts = self._get_sts(event["capabilities"]) sts = CAP.available(event["capabilities"])
if sts: if sts:
info = utils.parse.keyvalue(sts, delimiter=",") info = utils.parse.keyvalue(event["capabilities"][sts],
delimiter=",")
if not event["server"].connection_params.tls: if not event["server"].connection_params.tls:
self.set_policy(event["server"], int(info["port"]), None) self.set_policy(event["server"], int(info["port"]), None)
event["server"].disconnect() event["server"].disconnect()