nemunaire/nemubot
1
0
Fork 0
Code Releases Activity

WatchWebsite module: can alert when a site is up or down

Browse Source
This commit is contained in:
Némunaire 2012-12-07 21:18:21 +01:00
parent eada0c71e7
commit ab557dc8e8
2 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/networking.py
View File

@ -48,8 +48,7 @@ def cmd_isup(msg):
res.append(Response(msg.sender, "%s n'est pas une URL valide" % url, channel=msg.channel))
return res
else:
return Response(msg.sender, "Indiquer une URL a verifier !", channel=msg.channel)
return Response(msg.sender, "Indiquer une URL à vérifier !", channel=msg.channel)
def traceURL(url, timeout=5, stack=None):
"""Follow redirections and return the redirections stack"""

15
modules/watchWebsite/__init__.py
View File

@ -40,9 +40,9 @@ def getPageContent(url):
print_debug("Get page %s" % url)
try:
raw = urlopen(url, timeout=15)
return raw.read().decode()
except socket.timeout:
return None
return raw.read().decode()
def start_watching(site):
o = urlparse(site["url"], "http")
@ -85,7 +85,7 @@ def del_site(msg):
return Response(msg.sender, "je ne surveillais pas cette URL pour vous.",
channel=msg.channel, nick=msg.nick)
def add_site(msg):
def add_site(msg, diffType="diff"):
if len(msg.cmds) <= 1:
return Response(msg.sender, "quel site dois-je surveiller ?",
msg.channel, msg.nick)
@ -102,7 +102,7 @@ def add_site(msg):
if url not in DATAS.index:
watch = ModuleState("watch")
watch["type"] = "diff"
watch["type"] = diffType
watch["url"] = url
watch["time"] = 123
DATAS.addChild(watch)
@ -125,6 +125,15 @@ def format_response(site, link='%s', title='%s', categ='%s'):
def alert_change(content, site):
"""Alert when a change is detected"""
if site["type"] == "updown":
if site["lastcontent"] is None:
site["lastcontent"] = content is not None
if content is None != site.getInt("lastcontent"):
format_response(site, link=site["url"])
site["lastcontent"] = content is not None
return
if content is None:
start_watching(site)
return