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

Add auto-github-timeout to prevent duplicate auto-githubs

Browse source
This commit is contained in:
jesopo 2019-06-16 16:57:22 +01:00
parent b4d61ecd55
commit 002e1333a4
1 changed files with 37 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

54
modules/github/__init__.py
View file

@ -113,6 +113,9 @@ CHECK_RUN_FAILURES = ["failure", "cancelled", "timed_out", "action_required"]
@utils.export("channelset", {"setting": "auto-github",
"help": "Enable/disable automatically getting github issue/PR info",
"validate": utils.bool_or_none, "example": "on"})
@utils.export("channelset", {"setting": "auto-github-timeout",
"help": "Set amount of seconds between auto-github duplicates",
"validate": utils.int_or_none, "example": "300"})
class Module(ModuleManager.BaseModule):
def _parse_ref(self, channel, ref):
repo, _, number = ref.rpartition("#")
@ -235,6 +238,20 @@ class Module(ModuleManager.BaseModule):
else:
event["stderr"].write("Issue/PR not found")
def _cache_ref(self, ref):
return "auto-github-%s" % ref.lower()
def _auto_github_timeout(self, channel, ref):
timeout = channel.get_setting("auto-github-timeout", None)
if not timeout == None:
cache = self._cache_ref(ref)
if not self.bot.cache.has_item(cache):
self.bot.cache.temporary_cache(cache, timeout)
return True
else:
return False
else:
return True
@utils.hook("command.regex")
def url_regex(self, event):
"""
@ -245,14 +262,15 @@ class Module(ModuleManager.BaseModule):
event.eat()
ref = "%s/%s#%s" % (event["match"].group(1),
event["match"].group(2), event["match"].group(4))
try:
result = self._get_info(event["target"], ref)
except utils.EventError:
return
if result:
if event["target"].get_setting("github-hide-prefix", False):
event["stdout"].hide_prefix()
event["stdout"].write(result)
if self._auto_github_timeout(event["target"], ref):
try:
result = self._get_info(event["target"], ref)
except utils.EventError:
return
if result:
if event["target"].get_setting("github-hide-prefix", False):
event["stdout"].hide_prefix()
event["stdout"].write(result)
@utils.hook("command.regex")
def ref_regex(self, event):
@ -262,15 +280,17 @@ class Module(ModuleManager.BaseModule):
"""
if event["target"].get_setting("auto-github", False):
event.eat()
try:
result = self._get_info(event["target"],
event["match"].group(0))
except utils.EventError:
return
if result:
if event["target"].get_setting("github-hide-prefix", False):
event["stdout"].hide_prefix()
event["stdout"].write(result)
ref = event["match"].group(0)
if self._auto_github_timeout(event["target"], ref):
try:
result = self._get_info(event["target"],
event["match"].group(0))
except utils.EventError:
return
if result:
if event["target"].get_setting("github-hide-prefix", False):
event["stdout"].hide_prefix()
event["stdout"].write(result)
@utils.hook("received.command.ghwebhook", min_args=1, channel_only=True)
def github_webhook(self, event):