Compare commits
23 commits
master
...
refactor_t
| Author | SHA1 | Date | |
|---|---|---|---|
| c2f7606d1e | |||
| 334d342b40 | |||
| 410fa6ded1 | |||
| 559a2a8adc | |||
| 16facd949d | |||
| b7b1a92161 | |||
| 0c4144ce20 | |||
| 20f3e8b215 | |||
| 56b8e8a260 | |||
| 4d0260b5fc | |||
| fb670b0777 | |||
| 226785af19 | |||
| 7eaf39f850 | |||
| 9b68b9e217 | |||
| db58319af5 | |||
| dfc929f3e5 | |||
| 66341461a3 | |||
| fb3c715acd | |||
| 1f82b17219 | |||
| 82e50061b1 | |||
| 6c44df1482 | |||
| ca8434a476 | |||
| 16b7024f62 |
105 changed files with 1950 additions and 4124 deletions
26
.drone.yml
26
.drone.yml
|
|
@ -1,26 +0,0 @@
|
|||
---
|
||||
kind: pipeline
|
||||
type: docker
|
||||
name: default-arm64
|
||||
|
||||
platform:
|
||||
os: linux
|
||||
arch: arm64
|
||||
|
||||
steps:
|
||||
- name: build
|
||||
image: python:3.11-alpine
|
||||
commands:
|
||||
- pip install --no-cache-dir -r requirements.txt
|
||||
- pip install .
|
||||
|
||||
- name: docker
|
||||
image: plugins/docker
|
||||
settings:
|
||||
repo: nemunaire/nemubot
|
||||
auto_tag: true
|
||||
auto_tag_suffix: ${DRONE_STAGE_OS}-${DRONE_STAGE_ARCH}
|
||||
username:
|
||||
from_secret: docker_username
|
||||
password:
|
||||
from_secret: docker_password
|
||||
3
.gitmodules
vendored
Normal file
3
.gitmodules
vendored
Normal file
|
|
@ -0,0 +1,3 @@
|
|||
[submodule "modules/nextstop/external"]
|
||||
path = modules/nextstop/external
|
||||
url = git://github.com/nbr23/NextStop.git
|
||||
|
|
@ -1,9 +1,8 @@
|
|||
language: python
|
||||
python:
|
||||
- 3.3
|
||||
- 3.4
|
||||
- 3.5
|
||||
- 3.6
|
||||
- 3.7
|
||||
- nightly
|
||||
install:
|
||||
- pip install -r requirements.txt
|
||||
|
|
|
|||
21
Dockerfile
21
Dockerfile
|
|
@ -1,21 +0,0 @@
|
|||
FROM python:3.11-alpine
|
||||
|
||||
WORKDIR /usr/src/app
|
||||
|
||||
COPY requirements.txt /usr/src/app/
|
||||
RUN apk add --no-cache bash build-base capstone-dev mandoc-doc man-db w3m youtube-dl aspell aspell-fr py3-matrix-nio && \
|
||||
pip install --no-cache-dir --ignore-installed -r requirements.txt && \
|
||||
pip install bs4 capstone dnspython openai && \
|
||||
apk del build-base capstone-dev && \
|
||||
ln -s /var/lib/nemubot/home /home/nemubot
|
||||
|
||||
VOLUME /var/lib/nemubot
|
||||
|
||||
COPY . /usr/src/app/
|
||||
|
||||
RUN ./setup.py install
|
||||
|
||||
WORKDIR /var/lib/nemubot
|
||||
USER guest
|
||||
ENTRYPOINT [ "python", "-m", "nemubot", "-d", "-P", "", "-M", "/usr/src/app/modules" ]
|
||||
CMD [ "-D", "/var/lib/nemubot" ]
|
||||
|
|
@ -9,8 +9,10 @@ Requirements
|
|||
|
||||
*nemubot* requires at least Python 3.3 to work.
|
||||
|
||||
Connecting to SSL server requires [this patch](http://bugs.python.org/issue27629).
|
||||
|
||||
Some modules (like `cve`, `nextstop` or `laposte`) require the
|
||||
[BeautifulSoup module](https://www.crummy.com/software/BeautifulSoup/),
|
||||
[BeautifulSoup module](http://www.crummy.com/software/BeautifulSoup/),
|
||||
but the core and framework has no dependency.
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -12,7 +12,7 @@ from nemubot.message import Command
|
|||
from nemubot.tools.human import guess
|
||||
from nemubot.tools.xmlparser.node import ModuleState
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
|
||||
# LOADING #############################################################
|
||||
|
|
@ -76,7 +76,7 @@ def get_variable(name, msg=None):
|
|||
elif name in context.data.getNode("variables").index:
|
||||
return context.data.getNode("variables").index[name]["value"]
|
||||
else:
|
||||
return None
|
||||
return ""
|
||||
|
||||
|
||||
def list_variables(user=None):
|
||||
|
|
@ -108,12 +108,12 @@ def set_variable(name, value, creator):
|
|||
context.save()
|
||||
|
||||
|
||||
def replace_variables(cnts, msg):
|
||||
def replace_variables(cnts, msg=None):
|
||||
"""Replace variables contained in the content
|
||||
|
||||
Arguments:
|
||||
cnt -- content where search variables
|
||||
msg -- Message where pick some variables
|
||||
msg -- optional message where pick some variables
|
||||
"""
|
||||
|
||||
unsetCnt = list()
|
||||
|
|
@ -122,12 +122,12 @@ def replace_variables(cnts, msg):
|
|||
resultCnt = list()
|
||||
|
||||
for cnt in cnts:
|
||||
for res, name, default in re.findall("\\$\{(([a-zA-Z0-9:]+)(?:-([^}]+))?)\}", cnt):
|
||||
rv = re.match("([0-9]+)(:([0-9]*))?", name)
|
||||
for res in re.findall("\\$\{(?P<name>[a-zA-Z0-9:]+)\}", cnt):
|
||||
rv = re.match("([0-9]+)(:([0-9]*))?", res)
|
||||
if rv is not None:
|
||||
varI = int(rv.group(1)) - 1
|
||||
if varI >= len(msg.args):
|
||||
cnt = cnt.replace("${%s}" % res, default, 1)
|
||||
if varI > len(msg.args):
|
||||
cnt = cnt.replace("${%s}" % res, "", 1)
|
||||
elif rv.group(2) is not None:
|
||||
if rv.group(3) is not None and len(rv.group(3)):
|
||||
varJ = int(rv.group(3)) - 1
|
||||
|
|
@ -142,10 +142,9 @@ def replace_variables(cnts, msg):
|
|||
cnt = cnt.replace("${%s}" % res, msg.args[varI], 1)
|
||||
unsetCnt.append(varI)
|
||||
else:
|
||||
cnt = cnt.replace("${%s}" % res, get_variable(name) or default, 1)
|
||||
cnt = cnt.replace("${%s}" % res, get_variable(res), 1)
|
||||
resultCnt.append(cnt)
|
||||
|
||||
# Remove used content
|
||||
for u in sorted(set(unsetCnt), reverse=True):
|
||||
msg.args.pop(u)
|
||||
|
||||
|
|
@ -185,7 +184,7 @@ def cmd_listvars(msg):
|
|||
def cmd_set(msg):
|
||||
if len(msg.args) < 2:
|
||||
raise IMException("!set take two args: the key and the value.")
|
||||
set_variable(msg.args[0], " ".join(msg.args[1:]), msg.frm)
|
||||
set_variable(msg.args[0], " ".join(msg.args[1:]), msg.nick)
|
||||
return Response("Variable $%s successfully defined." % msg.args[0],
|
||||
channel=msg.channel)
|
||||
|
||||
|
|
@ -222,13 +221,13 @@ def cmd_alias(msg):
|
|||
|
||||
if alias.cmd in context.data.getNode("aliases").index:
|
||||
return Response("%s corresponds to %s" % (alias.cmd, context.data.getNode("aliases").index[alias.cmd]["origin"]),
|
||||
channel=msg.channel, nick=msg.frm)
|
||||
channel=msg.channel, nick=msg.nick)
|
||||
|
||||
elif len(msg.args) > 1:
|
||||
create_alias(alias.cmd,
|
||||
" ".join(msg.args[1:]),
|
||||
channel=msg.channel,
|
||||
creator=msg.frm)
|
||||
creator=msg.nick)
|
||||
return Response("New alias %s successfully registered." % alias.cmd,
|
||||
channel=msg.channel)
|
||||
|
||||
|
|
@ -260,18 +259,19 @@ def cmd_unalias(msg):
|
|||
|
||||
@hook.add(["pre","Command"])
|
||||
def treat_alias(msg):
|
||||
if context.data.getNode("aliases") is not None and msg.cmd in context.data.getNode("aliases").index:
|
||||
if msg.cmd in context.data.getNode("aliases").index:
|
||||
origin = context.data.getNode("aliases").index[msg.cmd]["origin"]
|
||||
rpl_msg = context.subparse(msg, origin)
|
||||
if isinstance(rpl_msg, Command):
|
||||
rpl_msg.args = replace_variables(rpl_msg.args, msg)
|
||||
rpl_msg.args += msg.args
|
||||
rpl_msg.kwargs.update(msg.kwargs)
|
||||
rpl_cmd = context.subparse(msg, origin)
|
||||
if isinstance(rpl_cmd, Command):
|
||||
rpl_cmd.args = replace_variables(rpl_cmd.args, msg)
|
||||
rpl_cmd.args += msg.args
|
||||
rpl_cmd.kwargs.update(msg.kwargs)
|
||||
elif len(msg.args) or len(msg.kwargs):
|
||||
raise IMException("This kind of alias doesn't take any argument (haven't you forgotten the '!'?).")
|
||||
|
||||
# Avoid infinite recursion
|
||||
if not isinstance(rpl_msg, Command) or msg.cmd != rpl_msg.cmd:
|
||||
return rpl_msg
|
||||
if msg.cmd != rpl_cmd.cmd:
|
||||
# Also return origin message, if it can be treated as well
|
||||
return [msg, rpl_cmd]
|
||||
|
||||
return msg
|
||||
|
|
|
|||
|
|
@ -13,7 +13,7 @@ from nemubot.tools.countdown import countdown_format
|
|||
from nemubot.tools.date import extractDate
|
||||
from nemubot.tools.xmlparser.node import ModuleState
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
|
||||
# LOADING #############################################################
|
||||
|
|
@ -27,7 +27,7 @@ def load(context):
|
|||
def findName(msg):
|
||||
if (not len(msg.args) or msg.args[0].lower() == "moi" or
|
||||
msg.args[0].lower() == "me"):
|
||||
name = msg.frm.lower()
|
||||
name = msg.nick.lower()
|
||||
else:
|
||||
name = msg.args[0].lower()
|
||||
|
||||
|
|
@ -77,7 +77,7 @@ def cmd_anniv(msg):
|
|||
else:
|
||||
return Response("désolé, je ne connais pas la date d'anniversaire"
|
||||
" de %s. Quand est-il né ?" % name,
|
||||
msg.channel, msg.frm)
|
||||
msg.channel, msg.nick)
|
||||
|
||||
|
||||
@hook.command("age",
|
||||
|
|
@ -98,7 +98,7 @@ def cmd_age(msg):
|
|||
msg.channel)
|
||||
else:
|
||||
return Response("désolé, je ne connais pas l'âge de %s."
|
||||
" Quand est-il né ?" % name, msg.channel, msg.frm)
|
||||
" Quand est-il né ?" % name, msg.channel, msg.nick)
|
||||
return True
|
||||
|
||||
|
||||
|
|
@ -106,18 +106,18 @@ def cmd_age(msg):
|
|||
|
||||
@hook.ask()
|
||||
def parseask(msg):
|
||||
res = re.match(r"^(\S+)\s*('s|suis|est|is|was|were)?\s+(birthday|geburtstag|née? |nee? le|born on).*$", msg.message, re.I)
|
||||
res = re.match(r"^(\S+)\s*('s|suis|est|is|was|were)?\s+(birthday|geburtstag|née? |nee? le|born on).*$", msg.text, re.I)
|
||||
if res is not None:
|
||||
try:
|
||||
extDate = extractDate(msg.message)
|
||||
extDate = extractDate(msg.text)
|
||||
if extDate is None or extDate.year > datetime.now().year:
|
||||
return Response("la date de naissance ne paraît pas valide...",
|
||||
msg.channel,
|
||||
msg.frm)
|
||||
msg.nick)
|
||||
else:
|
||||
nick = res.group(1)
|
||||
if nick == "my" or nick == "I" or nick == "i" or nick == "je" or nick == "mon" or nick == "ma":
|
||||
nick = msg.frm
|
||||
nick = msg.nick
|
||||
if nick.lower() in context.data.index:
|
||||
context.data.index[nick.lower()]["born"] = extDate
|
||||
else:
|
||||
|
|
@ -129,6 +129,6 @@ def parseask(msg):
|
|||
return Response("ok, c'est noté, %s est né le %s"
|
||||
% (nick, extDate.strftime("%A %d %B %Y Ã %H:%M")),
|
||||
msg.channel,
|
||||
msg.frm)
|
||||
msg.nick)
|
||||
except:
|
||||
raise IMException("la date de naissance ne paraît pas valide.")
|
||||
|
|
|
|||
|
|
@ -4,11 +4,12 @@
|
|||
|
||||
from datetime import datetime, timezone
|
||||
|
||||
from nemubot import context
|
||||
from nemubot.event import ModuleEvent
|
||||
from nemubot.hooks import hook
|
||||
from nemubot.tools.countdown import countdown_format
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
|
||||
# GLOBALS #############################################################
|
||||
|
|
|
|||
|
|
@ -9,7 +9,7 @@ from nemubot.exception import IMException
|
|||
from nemubot.hooks import hook
|
||||
from nemubot.tools import web
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
|
||||
# LOADING #############################################################
|
||||
|
|
|
|||
|
|
@ -7,7 +7,7 @@ from nemubot.exception import IMException
|
|||
from nemubot.hooks import hook
|
||||
from nemubot.message import Command, DirectAsk, Text
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
|
||||
# MODULE CORE #########################################################
|
||||
|
|
|
|||
|
|
@ -11,7 +11,7 @@ from nemubot.hooks import hook
|
|||
from nemubot.tools import web
|
||||
from nemubot.tools.web import striphtml
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
|
||||
# GLOBALS #############################################################
|
||||
|
|
@ -36,7 +36,7 @@ for k, v in s:
|
|||
# MODULE CORE #########################################################
|
||||
|
||||
def get_conjug(verb, stringTens):
|
||||
url = ("https://leconjugueur.lefigaro.fr/conjugaison/verbe/%s.html" %
|
||||
url = ("http://leconjugueur.lefigaro.fr/conjugaison/verbe/%s.html" %
|
||||
quote(verb.encode("ISO-8859-1")))
|
||||
page = web.getURLContent(url)
|
||||
|
||||
|
|
|
|||
|
|
@ -6,7 +6,7 @@ from bs4 import BeautifulSoup
|
|||
|
||||
from nemubot.hooks import hook
|
||||
from nemubot.tools.web import getURLContent, striphtml
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
|
||||
# GLOBALS #############################################################
|
||||
|
|
@ -25,8 +25,10 @@ def get_info_yt(msg):
|
|||
|
||||
for line in soup.body.find_all('tr'):
|
||||
n = line.find_all('td')
|
||||
if len(n) == 7:
|
||||
res.append_message("\x02%s:\x0F from %s type %s at %s. Weight: %s. %s%s" %
|
||||
tuple([striphtml(x.text).strip() for x in n]))
|
||||
|
||||
if len(n) == 5:
|
||||
try:
|
||||
res.append_message("\x02%s:\x0F from %s type %s at %s. %s" %
|
||||
tuple([striphtml(x.text) for x in n]))
|
||||
except:
|
||||
pass
|
||||
return res
|
||||
|
|
|
|||
|
|
@ -5,67 +5,29 @@
|
|||
from bs4 import BeautifulSoup
|
||||
from urllib.parse import quote
|
||||
|
||||
from nemubot.exception import IMException
|
||||
from nemubot.hooks import hook
|
||||
from nemubot.tools.web import getURLContent, striphtml
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
BASEURL_NIST = 'https://nvd.nist.gov/vuln/detail/'
|
||||
BASEURL_NIST = 'https://web.nvd.nist.gov/view/vuln/detail?vulnId='
|
||||
|
||||
|
||||
# MODULE CORE #########################################################
|
||||
|
||||
VULN_DATAS = {
|
||||
"alert-title": "vuln-warning-status-name",
|
||||
"alert-content": "vuln-warning-banner-content",
|
||||
|
||||
"description": "vuln-description",
|
||||
"published": "vuln-published-on",
|
||||
"last_modified": "vuln-last-modified-on",
|
||||
|
||||
"base_score": "vuln-cvssv3-base-score-link",
|
||||
"severity": "vuln-cvssv3-base-score-severity",
|
||||
"impact_score": "vuln-cvssv3-impact-score",
|
||||
"exploitability_score": "vuln-cvssv3-exploitability-score",
|
||||
|
||||
"av": "vuln-cvssv3-av",
|
||||
"ac": "vuln-cvssv3-ac",
|
||||
"pr": "vuln-cvssv3-pr",
|
||||
"ui": "vuln-cvssv3-ui",
|
||||
"s": "vuln-cvssv3-s",
|
||||
"c": "vuln-cvssv3-c",
|
||||
"i": "vuln-cvssv3-i",
|
||||
"a": "vuln-cvssv3-a",
|
||||
}
|
||||
|
||||
|
||||
def get_cve(cve_id):
|
||||
search_url = BASEURL_NIST + quote(cve_id.upper())
|
||||
|
||||
soup = BeautifulSoup(getURLContent(search_url))
|
||||
vuln = soup.body.find(class_="vuln-detail")
|
||||
cvss = vuln.findAll('div')[4]
|
||||
|
||||
vuln = {}
|
||||
|
||||
for vd in VULN_DATAS:
|
||||
r = soup.body.find(attrs={"data-testid": VULN_DATAS[vd]})
|
||||
if r:
|
||||
vuln[vd] = r.text.strip()
|
||||
|
||||
return vuln
|
||||
|
||||
|
||||
def display_metrics(av, ac, pr, ui, s, c, i, a, **kwargs):
|
||||
ret = []
|
||||
if av != "None": ret.append("Attack Vector: \x02%s\x0F" % av)
|
||||
if ac != "None": ret.append("Attack Complexity: \x02%s\x0F" % ac)
|
||||
if pr != "None": ret.append("Privileges Required: \x02%s\x0F" % pr)
|
||||
if ui != "None": ret.append("User Interaction: \x02%s\x0F" % ui)
|
||||
if s != "Unchanged": ret.append("Scope: \x02%s\x0F" % s)
|
||||
if c != "None": ret.append("Confidentiality: \x02%s\x0F" % c)
|
||||
if i != "None": ret.append("Integrity: \x02%s\x0F" % i)
|
||||
if a != "None": ret.append("Availability: \x02%s\x0F" % a)
|
||||
return ', '.join(ret)
|
||||
return [
|
||||
"Base score: " + cvss.findAll('div')[0].findAll('a')[0].text.strip(),
|
||||
vuln.findAll('p')[0].text, # description
|
||||
striphtml(vuln.findAll('div')[0].text).strip(), # publication date
|
||||
striphtml(vuln.findAll('div')[1].text).strip(), # last revised
|
||||
]
|
||||
|
||||
|
||||
# MODULE INTERFACE ####################################################
|
||||
|
|
@ -80,20 +42,6 @@ def get_cve_desc(msg):
|
|||
if cve_id[:3].lower() != 'cve':
|
||||
cve_id = 'cve-' + cve_id
|
||||
|
||||
cve = get_cve(cve_id)
|
||||
if not cve:
|
||||
raise IMException("CVE %s doesn't exists." % cve_id)
|
||||
|
||||
if "alert-title" in cve or "alert-content" in cve:
|
||||
alert = "\x02%s:\x0F %s " % (cve["alert-title"] if "alert-title" in cve else "",
|
||||
cve["alert-content"] if "alert-content" in cve else "")
|
||||
else:
|
||||
alert = ""
|
||||
|
||||
if "base_score" not in cve and "description" in cve:
|
||||
res.append_message("{alert}Last modified on \x02{last_modified}\x0F. {description}".format(alert=alert, **cve), title=cve_id)
|
||||
else:
|
||||
metrics = display_metrics(**cve)
|
||||
res.append_message("{alert}Base score: \x02{base_score} {severity}\x0F (impact: \x02{impact_score}\x0F, exploitability: \x02{exploitability_score}\x0F; {metrics}), last modified on \x02{last_modified}\x0F. {description}".format(alert=alert, metrics=metrics, **cve), title=cve_id)
|
||||
res.append_message(get_cve(cve_id))
|
||||
|
||||
return res
|
||||
|
|
|
|||
|
|
@ -8,7 +8,7 @@ from nemubot.exception import IMException
|
|||
from nemubot.hooks import hook
|
||||
from nemubot.tools import web
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
# MODULE CORE #########################################################
|
||||
|
||||
|
|
|
|||
|
|
@ -1,94 +0,0 @@
|
|||
"""DNS resolver"""
|
||||
|
||||
# PYTHON STUFFS #######################################################
|
||||
|
||||
import ipaddress
|
||||
import socket
|
||||
|
||||
import dns.exception
|
||||
import dns.name
|
||||
import dns.rdataclass
|
||||
import dns.rdatatype
|
||||
import dns.resolver
|
||||
|
||||
from nemubot.exception import IMException
|
||||
from nemubot.hooks import hook
|
||||
|
||||
from nemubot.module.more import Response
|
||||
|
||||
|
||||
# MODULE INTERFACE ####################################################
|
||||
|
||||
@hook.command("dig",
|
||||
help="Resolve domain name with a basic syntax similar to dig(1)")
|
||||
def dig(msg):
|
||||
lclass = "IN"
|
||||
ltype = "A"
|
||||
ledns = None
|
||||
ltimeout = 6.0
|
||||
ldomain = None
|
||||
lnameservers = []
|
||||
lsearchlist = []
|
||||
loptions = []
|
||||
for a in msg.args:
|
||||
if a in dns.rdatatype._by_text:
|
||||
ltype = a
|
||||
elif a in dns.rdataclass._by_text:
|
||||
lclass = a
|
||||
elif a[0] == "@":
|
||||
try:
|
||||
lnameservers.append(str(ipaddress.ip_address(a[1:])))
|
||||
except ValueError:
|
||||
for r in socket.getaddrinfo(a[1:], 53, proto=socket.IPPROTO_UDP):
|
||||
lnameservers.append(r[4][0])
|
||||
|
||||
elif a[0:8] == "+domain=":
|
||||
lsearchlist.append(dns.name.from_unicode(a[8:]))
|
||||
elif a[0:6] == "+edns=":
|
||||
ledns = int(a[6:])
|
||||
elif a[0:6] == "+time=":
|
||||
ltimeout = float(a[6:])
|
||||
elif a[0] == "+":
|
||||
loptions.append(a[1:])
|
||||
else:
|
||||
ldomain = a
|
||||
|
||||
if not ldomain:
|
||||
raise IMException("indicate a domain to resolve")
|
||||
|
||||
resolv = dns.resolver.Resolver()
|
||||
if ledns:
|
||||
resolv.edns = ledns
|
||||
resolv.lifetime = ltimeout
|
||||
resolv.timeout = ltimeout
|
||||
resolv.flags = (
|
||||
dns.flags.QR | dns.flags.RA |
|
||||
dns.flags.AA if "aaonly" in loptions or "aaflag" in loptions else 0 |
|
||||
dns.flags.AD if "adflag" in loptions else 0 |
|
||||
dns.flags.CD if "cdflag" in loptions else 0 |
|
||||
dns.flags.RD if "norecurse" not in loptions else 0
|
||||
)
|
||||
if lsearchlist:
|
||||
resolv.search = lsearchlist
|
||||
else:
|
||||
resolv.search = [dns.name.from_text(".")]
|
||||
|
||||
if lnameservers:
|
||||
resolv.nameservers = lnameservers
|
||||
|
||||
try:
|
||||
answers = resolv.query(ldomain, ltype, lclass, tcp="tcp" in loptions)
|
||||
except dns.exception.DNSException as e:
|
||||
raise IMException(str(e))
|
||||
|
||||
res = Response(channel=msg.channel, count=" (%s others entries)")
|
||||
for rdata in answers:
|
||||
res.append_message("%s %s %s %s %s" % (
|
||||
answers.qname.to_text(),
|
||||
answers.ttl if not "nottlid" in loptions else "",
|
||||
dns.rdataclass.to_text(answers.rdclass) if not "nocl" in loptions else "",
|
||||
dns.rdatatype.to_text(answers.rdtype),
|
||||
rdata.to_text())
|
||||
)
|
||||
|
||||
return res
|
||||
|
|
@ -1,89 +0,0 @@
|
|||
"""The Ultimate Disassembler Module"""
|
||||
|
||||
# PYTHON STUFFS #######################################################
|
||||
|
||||
import capstone
|
||||
|
||||
from nemubot.exception import IMException
|
||||
from nemubot.hooks import hook
|
||||
|
||||
from nemubot.module.more import Response
|
||||
|
||||
|
||||
# MODULE CORE #########################################################
|
||||
|
||||
ARCHITECTURES = {
|
||||
"arm": capstone.CS_ARCH_ARM,
|
||||
"arm64": capstone.CS_ARCH_ARM64,
|
||||
"mips": capstone.CS_ARCH_MIPS,
|
||||
"ppc": capstone.CS_ARCH_PPC,
|
||||
"sparc": capstone.CS_ARCH_SPARC,
|
||||
"sysz": capstone.CS_ARCH_SYSZ,
|
||||
"x86": capstone.CS_ARCH_X86,
|
||||
"xcore": capstone.CS_ARCH_XCORE,
|
||||
}
|
||||
|
||||
MODES = {
|
||||
"arm": capstone.CS_MODE_ARM,
|
||||
"thumb": capstone.CS_MODE_THUMB,
|
||||
"mips32": capstone.CS_MODE_MIPS32,
|
||||
"mips64": capstone.CS_MODE_MIPS64,
|
||||
"mips32r6": capstone.CS_MODE_MIPS32R6,
|
||||
"16": capstone.CS_MODE_16,
|
||||
"32": capstone.CS_MODE_32,
|
||||
"64": capstone.CS_MODE_64,
|
||||
"le": capstone.CS_MODE_LITTLE_ENDIAN,
|
||||
"be": capstone.CS_MODE_BIG_ENDIAN,
|
||||
"micro": capstone.CS_MODE_MICRO,
|
||||
"mclass": capstone.CS_MODE_MCLASS,
|
||||
"v8": capstone.CS_MODE_V8,
|
||||
"v9": capstone.CS_MODE_V9,
|
||||
}
|
||||
|
||||
# MODULE INTERFACE ####################################################
|
||||
|
||||
@hook.command("disas",
|
||||
help="Display assembly code",
|
||||
help_usage={"CODE": "Display assembly code corresponding to the given CODE"},
|
||||
keywords={
|
||||
"arch=ARCH": "Specify the architecture of the code to disassemble (default: x86, choose between: %s)" % ', '.join(ARCHITECTURES.keys()),
|
||||
"modes=MODE[,MODE]": "Specify hardware mode of the code to disassemble (default: 32, between: %s)" % ', '.join(MODES.keys()),
|
||||
})
|
||||
def cmd_disas(msg):
|
||||
if not len(msg.args):
|
||||
raise IMException("please give me some code")
|
||||
|
||||
# Determine the architecture
|
||||
if "arch" in msg.kwargs:
|
||||
if msg.kwargs["arch"] not in ARCHITECTURES:
|
||||
raise IMException("unknown architectures '%s'" % msg.kwargs["arch"])
|
||||
architecture = ARCHITECTURES[msg.kwargs["arch"]]
|
||||
else:
|
||||
architecture = capstone.CS_ARCH_X86
|
||||
|
||||
# Determine hardware modes
|
||||
modes = 0
|
||||
if "modes" in msg.kwargs:
|
||||
for mode in msg.kwargs["modes"].split(','):
|
||||
if mode not in MODES:
|
||||
raise IMException("unknown mode '%s'" % mode)
|
||||
modes += MODES[mode]
|
||||
elif architecture == capstone.CS_ARCH_X86 or architecture == capstone.CS_ARCH_PPC:
|
||||
modes = capstone.CS_MODE_32
|
||||
elif architecture == capstone.CS_ARCH_ARM or architecture == capstone.CS_ARCH_ARM64:
|
||||
modes = capstone.CS_MODE_ARM
|
||||
elif architecture == capstone.CS_ARCH_MIPS:
|
||||
modes = capstone.CS_MODE_MIPS32
|
||||
|
||||
# Get the code
|
||||
code = bytearray.fromhex(''.join([a.replace("0x", "") for a in msg.args]))
|
||||
|
||||
# Setup capstone
|
||||
md = capstone.Cs(architecture, modes)
|
||||
|
||||
res = Response(channel=msg.channel, nomore="No more instruction")
|
||||
|
||||
for isn in md.disasm(code, 0x1000):
|
||||
res.append_message("%s %s" %(isn.mnemonic, isn.op_str), title="0x%x" % isn.address)
|
||||
|
||||
return res
|
||||
|
|
@ -1,9 +1,7 @@
|
|||
"""Create countdowns and reminders"""
|
||||
|
||||
import calendar
|
||||
from datetime import datetime, timedelta, timezone
|
||||
from functools import partial
|
||||
import re
|
||||
from datetime import datetime, timedelta, timezone
|
||||
|
||||
from nemubot import context
|
||||
from nemubot.exception import IMException
|
||||
|
|
@ -12,84 +10,31 @@ from nemubot.hooks import hook
|
|||
from nemubot.message import Command
|
||||
from nemubot.tools.countdown import countdown_format, countdown
|
||||
from nemubot.tools.date import extractDate
|
||||
from nemubot.tools.xmlparser.basic import DictNode
|
||||
from nemubot.tools.xmlparser.node import ModuleState
|
||||
|
||||
from nemubot.module.more import Response
|
||||
|
||||
|
||||
class Event:
|
||||
|
||||
def __init__(self, server, channel, creator, start_time, end_time=None):
|
||||
self._server = server
|
||||
self._channel = channel
|
||||
self._creator = creator
|
||||
self._start = datetime.utcfromtimestamp(float(start_time)).replace(tzinfo=timezone.utc) if not isinstance(start_time, datetime) else start_time
|
||||
self._end = datetime.utcfromtimestamp(float(end_time)).replace(tzinfo=timezone.utc) if end_time else None
|
||||
self._evt = None
|
||||
|
||||
|
||||
def __del__(self):
|
||||
if self._evt is not None:
|
||||
context.del_event(self._evt)
|
||||
self._evt = None
|
||||
|
||||
|
||||
def saveElement(self, store, tag="event"):
|
||||
attrs = {
|
||||
"server": str(self._server),
|
||||
"channel": str(self._channel),
|
||||
"creator": str(self._creator),
|
||||
"start_time": str(calendar.timegm(self._start.timetuple())),
|
||||
}
|
||||
if self._end:
|
||||
attrs["end_time"] = str(calendar.timegm(self._end.timetuple()))
|
||||
store.startElement(tag, attrs)
|
||||
store.endElement(tag)
|
||||
|
||||
@property
|
||||
def creator(self):
|
||||
return self._creator
|
||||
|
||||
@property
|
||||
def start(self):
|
||||
return self._start
|
||||
|
||||
@property
|
||||
def end(self):
|
||||
return self._end
|
||||
|
||||
@end.setter
|
||||
def end(self, c):
|
||||
self._end = c
|
||||
|
||||
@end.deleter
|
||||
def end(self):
|
||||
self._end = None
|
||||
from more import Response
|
||||
|
||||
|
||||
def help_full ():
|
||||
return "This module store a lot of events: ny, we, " + (", ".join(context.datas.keys()) if hasattr(context, "datas") else "") + "\n!eventslist: gets list of timer\n!start /something/: launch a timer"
|
||||
return "This module store a lot of events: ny, we, " + (", ".join(context.datas.index.keys())) + "\n!eventslist: gets list of timer\n!start /something/: launch a timer"
|
||||
|
||||
|
||||
def load(context):
|
||||
context.set_knodes({
|
||||
"dict": DictNode,
|
||||
"event": Event,
|
||||
})
|
||||
#Define the index
|
||||
context.data.setIndex("name")
|
||||
|
||||
if context.data is None:
|
||||
context.set_default(DictNode())
|
||||
|
||||
# Relaunch all timers
|
||||
for kevt in context.data:
|
||||
if context.data[kevt].end:
|
||||
context.data[kevt]._evt = context.add_event(ModuleEvent(partial(fini, kevt, context.data[kevt]), offset=context.data[kevt].end - datetime.now(timezone.utc), interval=0))
|
||||
for evt in context.data.index.keys():
|
||||
if context.data.index[evt].hasAttribute("end"):
|
||||
event = ModuleEvent(call=fini, call_data=dict(strend=context.data.index[evt]))
|
||||
event._end = context.data.index[evt].getDate("end")
|
||||
idt = context.add_event(event)
|
||||
if idt is not None:
|
||||
context.data.index[evt]["_id"] = idt
|
||||
|
||||
|
||||
def fini(name, evt):
|
||||
context.send_response(evt._server, Response("%s arrivé à échéance." % name, channel=evt._channel, nick=evt.creator))
|
||||
evt._evt = None
|
||||
del context.data[name]
|
||||
def fini(d, strend):
|
||||
context.send_response(strend["server"], Response("%s arrivé à échéance." % strend["name"], channel=strend["channel"], nick=strend["proprio"]))
|
||||
context.data.delChild(context.data.index[strend["name"]])
|
||||
context.save()
|
||||
|
||||
|
||||
|
|
@ -118,10 +63,18 @@ def start_countdown(msg):
|
|||
"""!start /something/: launch a timer"""
|
||||
if len(msg.args) < 1:
|
||||
raise IMException("indique le nom d'un événement à chronométrer")
|
||||
if msg.args[0] in context.data:
|
||||
if msg.args[0] in context.data.index:
|
||||
raise IMException("%s existe déjà ." % msg.args[0])
|
||||
|
||||
evt = Event(server=msg.server, channel=msg.channel, creator=msg.frm, start_time=msg.date)
|
||||
strnd = ModuleState("strend")
|
||||
strnd["server"] = msg.server
|
||||
strnd["channel"] = msg.channel
|
||||
strnd["proprio"] = msg.nick
|
||||
strnd["start"] = msg.date
|
||||
strnd["name"] = msg.args[0]
|
||||
context.data.addChild(strnd)
|
||||
|
||||
evt = ModuleEvent(call=fini, call_data=dict(strend=strnd))
|
||||
|
||||
if len(msg.args) > 1:
|
||||
result1 = re.findall("([0-9]+)([smhdjwyaSMHDJWYA])?", msg.args[1])
|
||||
|
|
@ -139,51 +92,50 @@ def start_countdown(msg):
|
|||
if result2 is None or result2.group(4) is None: yea = now.year
|
||||
else: yea = int(result2.group(4))
|
||||
if result2 is not None and result3 is not None:
|
||||
evt.end = datetime(yea, int(result2.group(3)), int(result2.group(2)), hou, minu, sec, timezone.utc)
|
||||
strnd["end"] = datetime(yea, int(result2.group(3)), int(result2.group(2)), hou, minu, sec, timezone.utc)
|
||||
elif result2 is not None:
|
||||
evt.end = datetime(int(result2.group(4)), int(result2.group(3)), int(result2.group(2)), 0, 0, 0, timezone.utc)
|
||||
strnd["end"] = datetime(int(result2.group(4)), int(result2.group(3)), int(result2.group(2)), 0, 0, 0, timezone.utc)
|
||||
elif result3 is not None:
|
||||
if hou * 3600 + minu * 60 + sec > now.hour * 3600 + now.minute * 60 + now.second:
|
||||
evt.end = datetime(now.year, now.month, now.day, hou, minu, sec, timezone.utc)
|
||||
strnd["end"] = datetime(now.year, now.month, now.day, hou, minu, sec, timezone.utc)
|
||||
else:
|
||||
evt.end = datetime(now.year, now.month, now.day + 1, hou, minu, sec, timezone.utc)
|
||||
strnd["end"] = datetime(now.year, now.month, now.day + 1, hou, minu, sec, timezone.utc)
|
||||
evt._end = strnd.getDate("end")
|
||||
strnd["_id"] = context.add_event(evt)
|
||||
except:
|
||||
context.data.delChild(strnd)
|
||||
raise IMException("Mauvais format de date pour l'événement %s. Il n'a pas été créé." % msg.args[0])
|
||||
|
||||
elif result1 is not None and len(result1) > 0:
|
||||
evt.end = msg.date
|
||||
strnd["end"] = msg.date
|
||||
for (t, g) in result1:
|
||||
if g is None or g == "" or g == "m" or g == "M":
|
||||
evt.end += timedelta(minutes=int(t))
|
||||
strnd["end"] += timedelta(minutes=int(t))
|
||||
elif g == "h" or g == "H":
|
||||
evt.end += timedelta(hours=int(t))
|
||||
strnd["end"] += timedelta(hours=int(t))
|
||||
elif g == "d" or g == "D" or g == "j" or g == "J":
|
||||
evt.end += timedelta(days=int(t))
|
||||
strnd["end"] += timedelta(days=int(t))
|
||||
elif g == "w" or g == "W":
|
||||
evt.end += timedelta(days=int(t)*7)
|
||||
strnd["end"] += timedelta(days=int(t)*7)
|
||||
elif g == "y" or g == "Y" or g == "a" or g == "A":
|
||||
evt.end += timedelta(days=int(t)*365)
|
||||
strnd["end"] += timedelta(days=int(t)*365)
|
||||
else:
|
||||
evt.end += timedelta(seconds=int(t))
|
||||
strnd["end"] += timedelta(seconds=int(t))
|
||||
evt._end = strnd.getDate("end")
|
||||
eid = context.add_event(evt)
|
||||
if eid is not None:
|
||||
strnd["_id"] = eid
|
||||
|
||||
else:
|
||||
raise IMException("Mauvais format de date pour l'événement %s. Il n'a pas été créé." % msg.args[0])
|
||||
|
||||
context.data[msg.args[0]] = evt
|
||||
context.save()
|
||||
|
||||
if evt.end is not None:
|
||||
context.add_event(ModuleEvent(partial(fini, msg.args[0], evt),
|
||||
offset=evt.end - datetime.now(timezone.utc),
|
||||
interval=0))
|
||||
if "end" in strnd:
|
||||
return Response("%s commencé le %s et se terminera le %s." %
|
||||
(msg.args[0], msg.date.strftime("%A %d %B %Y Ã %H:%M:%S"),
|
||||
evt.end.strftime("%A %d %B %Y Ã %H:%M:%S")),
|
||||
channel=msg.channel)
|
||||
strnd.getDate("end").strftime("%A %d %B %Y Ã %H:%M:%S")),
|
||||
nick=msg.frm)
|
||||
else:
|
||||
return Response("%s commencé le %s"% (msg.args[0],
|
||||
msg.date.strftime("%A %d %B %Y Ã %H:%M:%S")),
|
||||
channel=msg.channel)
|
||||
nick=msg.frm)
|
||||
|
||||
|
||||
@hook.command("end")
|
||||
|
|
@ -192,66 +144,67 @@ def end_countdown(msg):
|
|||
if len(msg.args) < 1:
|
||||
raise IMException("quel événement terminer ?")
|
||||
|
||||
if msg.args[0] in context.data:
|
||||
if context.data[msg.args[0]].creator == msg.frm or (msg.cmd == "forceend" and msg.frm_owner):
|
||||
duration = countdown(msg.date - context.data[msg.args[0]].start)
|
||||
del context.data[msg.args[0]]
|
||||
if msg.args[0] in context.data.index:
|
||||
if context.data.index[msg.args[0]]["proprio"] == msg.nick or (msg.cmd == "forceend" and msg.frm_owner):
|
||||
duration = countdown(msg.date - context.data.index[msg.args[0]].getDate("start"))
|
||||
context.del_event(context.data.index[msg.args[0]]["_id"])
|
||||
context.data.delChild(context.data.index[msg.args[0]])
|
||||
context.save()
|
||||
return Response("%s a duré %s." % (msg.args[0], duration),
|
||||
channel=msg.channel, nick=msg.frm)
|
||||
channel=msg.channel, nick=msg.nick)
|
||||
else:
|
||||
raise IMException("Vous ne pouvez pas terminer le compteur %s, créé par %s." % (msg.args[0], context.data[msg.args[0]].creator))
|
||||
raise IMException("Vous ne pouvez pas terminer le compteur %s, créé par %s." % (msg.args[0], context.data.index[msg.args[0]]["proprio"]))
|
||||
else:
|
||||
return Response("%s n'est pas un compteur connu."% (msg.args[0]), channel=msg.channel, nick=msg.frm)
|
||||
return Response("%s n'est pas un compteur connu."% (msg.args[0]), channel=msg.channel, nick=msg.nick)
|
||||
|
||||
|
||||
@hook.command("eventslist")
|
||||
def liste(msg):
|
||||
"""!eventslist: gets list of timer"""
|
||||
if len(msg.args):
|
||||
res = Response(channel=msg.channel)
|
||||
res = list()
|
||||
for user in msg.args:
|
||||
cmptr = [k for k in context.data if context.data[k].creator == user]
|
||||
cmptr = [x["name"] for x in context.data.index.values() if x["proprio"] == user]
|
||||
if len(cmptr) > 0:
|
||||
res.append_message(cmptr, title="Events created by %s" % user)
|
||||
res.append("Compteurs créés par %s : %s" % (user, ", ".join(cmptr)))
|
||||
else:
|
||||
res.append_message("%s doesn't have any counting events" % user)
|
||||
return res
|
||||
res.append("%s n'a pas créé de compteur" % user)
|
||||
return Response(" ; ".join(res), channel=msg.channel)
|
||||
else:
|
||||
return Response(list(context.data.keys()), channel=msg.channel, title="Known events")
|
||||
return Response("Compteurs connus : %s." % ", ".join(context.data.index.keys()), channel=msg.channel)
|
||||
|
||||
|
||||
@hook.command(match=lambda msg: isinstance(msg, Command) and msg.cmd in context.data)
|
||||
@hook.command(match=lambda msg: isinstance(msg, Command) and msg.cmd in context.data.index)
|
||||
def parseanswer(msg):
|
||||
res = Response(channel=msg.channel)
|
||||
|
||||
# Avoid message starting by ! which can be interpreted as command by other bots
|
||||
if msg.cmd[0] == "!":
|
||||
res.nick = msg.frm
|
||||
res.nick = msg.nick
|
||||
|
||||
if msg.cmd in context.data:
|
||||
if context.data[msg.cmd].end:
|
||||
res.append_message("%s commencé il y a %s et se terminera dans %s." % (msg.cmd, countdown(msg.date - context.data[msg.cmd].start), countdown(context.data[msg.cmd].end - msg.date)))
|
||||
if context.data.index[msg.cmd].name == "strend":
|
||||
if context.data.index[msg.cmd].hasAttribute("end"):
|
||||
res.append_message("%s commencé il y a %s et se terminera dans %s." % (msg.cmd, countdown(msg.date - context.data.index[msg.cmd].getDate("start")), countdown(context.data.index[msg.cmd].getDate("end") - msg.date)))
|
||||
else:
|
||||
res.append_message("%s commencé il y a %s." % (msg.cmd, countdown(msg.date - context.data[msg.cmd].start)))
|
||||
res.append_message("%s commencé il y a %s." % (msg.cmd, countdown(msg.date - context.data.index[msg.cmd].getDate("start"))))
|
||||
else:
|
||||
res.append_message(countdown_format(context.data[msg.cmd].start, context.data[msg.cmd]["msg_before"], context.data[msg.cmd]["msg_after"]))
|
||||
res.append_message(countdown_format(context.data.index[msg.cmd].getDate("start"), context.data.index[msg.cmd]["msg_before"], context.data.index[msg.cmd]["msg_after"]))
|
||||
return res
|
||||
|
||||
|
||||
RGXP_ask = re.compile(r"^.*((create|new)\s+(a|an|a\s*new|an\s*other)?\s*(events?|commande?)|(nouvel(le)?|ajoute|cr[ée]{1,3})\s+(un)?\s*([eé]v[ée]nements?|commande?)).*$", re.I)
|
||||
|
||||
@hook.ask(match=lambda msg: RGXP_ask.match(msg.message))
|
||||
@hook.ask(match=lambda msg: RGXP_ask.match(msg.text))
|
||||
def parseask(msg):
|
||||
name = re.match("^.*!([^ \"'@!]+).*$", msg.message)
|
||||
name = re.match("^.*!([^ \"'@!]+).*$", msg.text)
|
||||
if name is None:
|
||||
raise IMException("il faut que tu attribues une commande à l'événement.")
|
||||
if name.group(1) in context.data:
|
||||
if name.group(1) in context.data.index:
|
||||
raise IMException("un événement portant ce nom existe déjà .")
|
||||
|
||||
texts = re.match("^[^\"]*(avant|après|apres|before|after)?[^\"]*\"([^\"]+)\"[^\"]*((avant|après|apres|before|after)?.*\"([^\"]+)\".*)?$", msg.message, re.I)
|
||||
texts = re.match("^[^\"]*(avant|après|apres|before|after)?[^\"]*\"([^\"]+)\"[^\"]*((avant|après|apres|before|after)?.*\"([^\"]+)\".*)?$", msg.text, re.I)
|
||||
if texts is not None and texts.group(3) is not None:
|
||||
extDate = extractDate(msg.message)
|
||||
extDate = extractDate(msg.text)
|
||||
if extDate is None or extDate == "":
|
||||
raise IMException("la date de l'événement est invalide !")
|
||||
|
||||
|
|
@ -270,7 +223,7 @@ def parseask(msg):
|
|||
evt = ModuleState("event")
|
||||
evt["server"] = msg.server
|
||||
evt["channel"] = msg.channel
|
||||
evt["proprio"] = msg.frm
|
||||
evt["proprio"] = msg.nick
|
||||
evt["name"] = name.group(1)
|
||||
evt["start"] = extDate
|
||||
evt["msg_after"] = msg_after
|
||||
|
|
@ -284,7 +237,7 @@ def parseask(msg):
|
|||
evt = ModuleState("event")
|
||||
evt["server"] = msg.server
|
||||
evt["channel"] = msg.channel
|
||||
evt["proprio"] = msg.frm
|
||||
evt["proprio"] = msg.nick
|
||||
evt["name"] = name.group(1)
|
||||
evt["msg_before"] = texts.group (2)
|
||||
context.data.addChild(evt)
|
||||
|
|
|
|||
|
|
@ -1,64 +0,0 @@
|
|||
"""Inform about Free Mobile tarifs"""
|
||||
|
||||
# PYTHON STUFFS #######################################################
|
||||
|
||||
import urllib.parse
|
||||
from bs4 import BeautifulSoup
|
||||
|
||||
from nemubot.exception import IMException
|
||||
from nemubot.hooks import hook
|
||||
from nemubot.tools import web
|
||||
|
||||
from nemubot.module.more import Response
|
||||
|
||||
|
||||
# MODULE CORE #########################################################
|
||||
|
||||
ACT = {
|
||||
"ff_toFixe": "Appel vers les fixes",
|
||||
"ff_toMobile": "Appel vers les mobiles",
|
||||
"ff_smsSendedToCountry": "SMS vers le pays",
|
||||
"ff_mmsSendedToCountry": "MMS vers le pays",
|
||||
"fc_callToFrance": "Appel vers la France",
|
||||
"fc_smsToFrance": "SMS vers la france",
|
||||
"fc_mmsSended": "MMS vers la france",
|
||||
"fc_callToSameCountry": "Réception des appels",
|
||||
"fc_callReceived": "Appel dans le pays",
|
||||
"fc_smsReceived": "SMS (Réception)",
|
||||
"fc_mmsReceived": "MMS (Réception)",
|
||||
"fc_moDataFromCountry": "Data",
|
||||
}
|
||||
|
||||
def get_land_tarif(country, forfait="pkgFREE"):
|
||||
url = "http://mobile.international.free.fr/?" + urllib.parse.urlencode({'pays': country})
|
||||
page = web.getURLContent(url)
|
||||
soup = BeautifulSoup(page)
|
||||
|
||||
fact = soup.find(class_=forfait)
|
||||
|
||||
if fact is None:
|
||||
raise IMException("Country or forfait not found.")
|
||||
|
||||
res = {}
|
||||
for s in ACT.keys():
|
||||
try:
|
||||
res[s] = fact.find(attrs={"data-bind": "text: " + s}).text + " " + fact.find(attrs={"data-bind": "html: " + s + "Unit"}).text
|
||||
except AttributeError:
|
||||
res[s] = "inclus"
|
||||
|
||||
return res
|
||||
|
||||
@hook.command("freetarifs",
|
||||
help="Show Free Mobile tarifs for given contries",
|
||||
help_usage={"COUNTRY": "Show Free Mobile tarifs for given CONTRY"},
|
||||
keywords={
|
||||
"forfait=FORFAIT": "Related forfait between Free (default) and 2euro"
|
||||
})
|
||||
def get_freetarif(msg):
|
||||
res = Response(channel=msg.channel)
|
||||
|
||||
for country in msg.args:
|
||||
t = get_land_tarif(country.lower().capitalize(), "pkg" + (msg.kwargs["forfait"] if "forfait" in msg.kwargs else "FREE").upper())
|
||||
res.append_message(["\x02%s\x0F : %s" % (ACT[k], t[k]) for k in sorted(ACT.keys(), reverse=True)], title=country)
|
||||
|
||||
return res
|
||||
|
|
@ -9,7 +9,7 @@ from nemubot.exception import IMException
|
|||
from nemubot.hooks import hook
|
||||
from nemubot.tools import web
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
|
||||
# MODULE CORE #########################################################
|
||||
|
|
|
|||
|
|
@ -9,7 +9,7 @@ from nemubot.exception import IMException
|
|||
from nemubot.hooks import hook
|
||||
from nemubot.message import Command, Text
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
|
||||
# MODULE CORE #########################################################
|
||||
|
|
@ -73,7 +73,7 @@ def cmd_grep(msg):
|
|||
|
||||
only = "only" in msg.kwargs
|
||||
|
||||
l = [m for m in grep(msg.args[0] if len(msg.args[0]) and msg.args[0][0] == "^" else ".*?(" + msg.args[0] + ").*?",
|
||||
l = [m for m in grep(msg.args[0] if msg.args[0][0] == "^" else ".*?(" + msg.args[0] + ").*?",
|
||||
" ".join(msg.args[1:]),
|
||||
msg,
|
||||
icase="nocase" in msg.kwargs,
|
||||
|
|
|
|||
100
modules/imdb.py
100
modules/imdb.py
|
|
@ -5,56 +5,63 @@
|
|||
import re
|
||||
import urllib.parse
|
||||
|
||||
from bs4 import BeautifulSoup
|
||||
|
||||
from nemubot.exception import IMException
|
||||
from nemubot.hooks import hook
|
||||
from nemubot.tools import web
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
|
||||
# MODULE CORE #########################################################
|
||||
|
||||
def get_movie_by_id(imdbid):
|
||||
def get_movie(title=None, year=None, imdbid=None, fullplot=True, tomatoes=False):
|
||||
"""Returns the information about the matching movie"""
|
||||
|
||||
url = "http://www.imdb.com/title/" + urllib.parse.quote(imdbid)
|
||||
soup = BeautifulSoup(web.getURLContent(url))
|
||||
|
||||
return {
|
||||
"imdbID": imdbid,
|
||||
"Title": soup.body.find('h1').contents[0].strip(),
|
||||
"Year": soup.body.find(id="titleYear").find("a").text.strip() if soup.body.find(id="titleYear") else ", ".join([y.text.strip() for y in soup.body.find(attrs={"class": "seasons-and-year-nav"}).find_all("a")[1:]]),
|
||||
"Duration": soup.body.find(attrs={"class": "title_wrapper"}).find("time").text.strip() if soup.body.find(attrs={"class": "title_wrapper"}).find("time") else None,
|
||||
"imdbRating": soup.body.find(attrs={"class": "ratingValue"}).find("strong").text.strip() if soup.body.find(attrs={"class": "ratingValue"}) else None,
|
||||
"imdbVotes": soup.body.find(attrs={"class": "imdbRating"}).find("a").text.strip() if soup.body.find(attrs={"class": "imdbRating"}) else None,
|
||||
"Plot": re.sub(r"\s+", " ", soup.body.find(attrs={"class": "summary_text"}).text).strip(),
|
||||
|
||||
"Type": "TV Series" if soup.find(id="title-episode-widget") else "Movie",
|
||||
"Genre": ", ".join([x.text.strip() for x in soup.body.find(id="titleStoryLine").find_all("a") if x.get("href") is not None and x.get("href")[:21] == "/search/title?genres="]),
|
||||
"Country": ", ".join([x.text.strip() for x in soup.body.find(id="titleDetails").find_all("a") if x.get("href") is not None and x.get("href")[:32] == "/search/title?country_of_origin="]),
|
||||
"Credits": " ; ".join([x.find("h4").text.strip() + " " + (", ".join([y.text.strip() for y in x.find_all("a") if y.get("href") is not None and y.get("href")[:6] == "/name/"])) for x in soup.body.find_all(attrs={"class": "credit_summary_item"})]),
|
||||
}
|
||||
|
||||
|
||||
def find_movies(title, year=None):
|
||||
"""Find existing movies matching a approximate title"""
|
||||
|
||||
title = title.lower()
|
||||
|
||||
# Built URL
|
||||
url = "https://v2.sg.media-imdb.com/suggests/%s/%s.json" % (urllib.parse.quote(title[0]), urllib.parse.quote(title.replace(" ", "_")))
|
||||
url = "http://www.omdbapi.com/?"
|
||||
if title is not None:
|
||||
url += "t=%s&" % urllib.parse.quote(title)
|
||||
if year is not None:
|
||||
url += "y=%s&" % urllib.parse.quote(year)
|
||||
if imdbid is not None:
|
||||
url += "i=%s&" % urllib.parse.quote(imdbid)
|
||||
if fullplot:
|
||||
url += "plot=full&"
|
||||
if tomatoes:
|
||||
url += "tomatoes=true&"
|
||||
|
||||
# Make the request
|
||||
data = web.getJSON(url, remove_callback=True)
|
||||
data = web.getJSON(url)
|
||||
|
||||
# Return data
|
||||
if "Error" in data:
|
||||
raise IMException(data["Error"])
|
||||
|
||||
elif "Response" in data and data["Response"] == "True":
|
||||
return data
|
||||
|
||||
if "d" not in data:
|
||||
return None
|
||||
elif year is None:
|
||||
return data["d"]
|
||||
else:
|
||||
return [d for d in data["d"] if "y" in d and str(d["y"]) == year]
|
||||
raise IMException("An error occurs during movie search")
|
||||
|
||||
|
||||
def find_movies(title):
|
||||
"""Find existing movies matching a approximate title"""
|
||||
|
||||
# Built URL
|
||||
url = "http://www.omdbapi.com/?s=%s" % urllib.parse.quote(title)
|
||||
|
||||
# Make the request
|
||||
data = web.getJSON(url)
|
||||
|
||||
# Return data
|
||||
if "Error" in data:
|
||||
raise IMException(data["Error"])
|
||||
|
||||
elif "Search" in data:
|
||||
return data
|
||||
|
||||
else:
|
||||
raise IMException("An error occurs during movie search")
|
||||
|
||||
|
||||
# MODULE INTERFACE ####################################################
|
||||
|
|
@ -72,28 +79,23 @@ def cmd_imdb(msg):
|
|||
title = ' '.join(msg.args)
|
||||
|
||||
if re.match("^tt[0-9]{7}$", title) is not None:
|
||||
data = get_movie_by_id(imdbid=title)
|
||||
data = get_movie(imdbid=title)
|
||||
else:
|
||||
rm = re.match(r"^(.+)\s\(([0-9]{4})\)$", title)
|
||||
if rm is not None:
|
||||
data = find_movies(rm.group(1), year=rm.group(2))
|
||||
data = get_movie(title=rm.group(1), year=rm.group(2))
|
||||
else:
|
||||
data = find_movies(title)
|
||||
|
||||
if not data:
|
||||
raise IMException("Movie/series not found")
|
||||
|
||||
data = get_movie_by_id(data[0]["id"])
|
||||
data = get_movie(title=title)
|
||||
|
||||
res = Response(channel=msg.channel,
|
||||
title="%s (%s)" % (data['Title'], data['Year']),
|
||||
nomore="No more information, more at http://www.imdb.com/title/%s" % data['imdbID'])
|
||||
|
||||
res.append_message("%s \x02genre:\x0F %s; \x02rating\x0F: %s (%s votes); \x02plot\x0F: %s" %
|
||||
(data['Type'], data['Genre'], data['imdbRating'], data['imdbVotes'], data['Plot']))
|
||||
res.append_message("%s \x02from\x0F %s; %s"
|
||||
% (data['Type'], data['Country'], data['Credits']))
|
||||
res.append_message("\x02rating\x0F: %s (%s votes); \x02plot\x0F: %s" %
|
||||
(data['imdbRating'], data['imdbVotes'], data['Plot']))
|
||||
|
||||
res.append_message("%s \x02from\x0F %s \x02released on\x0F %s; \x02genre:\x0F %s; \x02directed by:\x0F %s; \x02written by:\x0F %s; \x02main actors:\x0F %s"
|
||||
% (data['Type'], data['Country'], data['Released'], data['Genre'], data['Director'], data['Writer'], data['Actors']))
|
||||
return res
|
||||
|
||||
|
||||
|
|
@ -109,7 +111,7 @@ def cmd_search(msg):
|
|||
data = find_movies(' '.join(msg.args))
|
||||
|
||||
movies = list()
|
||||
for m in data:
|
||||
movies.append("\x02%s\x0F%s with %s" % (m['l'], (" (" + str(m['y']) + ")") if "y" in m else "", m['s']))
|
||||
for m in data['Search']:
|
||||
movies.append("\x02%s\x0F (%s of %s)" % (m['Title'], m['Type'], m['Year']))
|
||||
|
||||
return Response(movies, title="Titles found", channel=msg.channel)
|
||||
|
|
|
|||
|
|
@ -1,7 +1,7 @@
|
|||
from nemubot.hooks import hook
|
||||
from nemubot.exception import IMException
|
||||
from nemubot.tools import web
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
import json
|
||||
|
||||
nemubotversion = 3.4
|
||||
|
|
|
|||
|
|
@ -8,7 +8,7 @@ import os
|
|||
|
||||
from nemubot.hooks import hook
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
|
||||
# GLOBALS #############################################################
|
||||
|
|
|
|||
|
|
@ -9,11 +9,11 @@ from nemubot.exception import IMException
|
|||
from nemubot.hooks import hook
|
||||
from nemubot.tools import web
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
# GLOBALS #############################################################
|
||||
|
||||
URL_API = "https://open.mapquestapi.com/geocoding/v1/address?key=%s&location=%%s"
|
||||
URL_API = "http://open.mapquestapi.com/geocoding/v1/address?key=%s&location=%%s"
|
||||
|
||||
|
||||
# LOADING #############################################################
|
||||
|
|
@ -23,7 +23,7 @@ def load(context):
|
|||
raise ImportError("You need a MapQuest API key in order to use this "
|
||||
"module. Add it to the module configuration file:\n"
|
||||
"<module name=\"mapquest\" key=\"XXXXXXXXXXXXXXXX\" "
|
||||
"/>\nRegister at https://developer.mapquest.com/")
|
||||
"/>\nRegister at http://developer.mapquest.com/")
|
||||
global URL_API
|
||||
URL_API = URL_API % context.config["apikey"].replace("%", "%%")
|
||||
|
||||
|
|
@ -55,7 +55,7 @@ def cmd_geocode(msg):
|
|||
if not len(msg.args):
|
||||
raise IMException("indicate a name")
|
||||
|
||||
res = Response(channel=msg.channel, nick=msg.frm,
|
||||
res = Response(channel=msg.channel, nick=msg.nick,
|
||||
nomore="No more geocode", count=" (%s more geocode)")
|
||||
|
||||
for loc in geocode(' '.join(msg.args)):
|
||||
|
|
|
|||
|
|
@ -11,7 +11,7 @@ from nemubot.tools import web
|
|||
|
||||
nemubotversion = 3.4
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
|
||||
# MEDIAWIKI REQUESTS ##################################################
|
||||
|
|
|
|||
|
|
@ -181,7 +181,6 @@ class Response:
|
|||
return self.nomore
|
||||
|
||||
if self.line_treat is not None and self.elt == 0:
|
||||
try:
|
||||
if isinstance(self.messages[0], list):
|
||||
for x in self.messages[0]:
|
||||
print(x, self.line_treat(x))
|
||||
|
|
@ -189,8 +188,6 @@ class Response:
|
|||
else:
|
||||
self.messages[0] = (self.line_treat(self.messages[0])
|
||||
.replace("\n", " ").strip())
|
||||
except Exception as e:
|
||||
logger.exception(e)
|
||||
|
||||
msg = ""
|
||||
if self.title is not None:
|
||||
|
|
@ -8,7 +8,7 @@ import re
|
|||
from nemubot.exception import IMException
|
||||
from nemubot.hooks import hook
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
from . import isup
|
||||
from . import page
|
||||
|
|
|
|||
|
|
@ -11,7 +11,7 @@ def isup(url):
|
|||
|
||||
o = urllib.parse.urlparse(getNormalizedURL(url), "http")
|
||||
if o.netloc != "":
|
||||
isup = getJSON("https://isitup.org/%s.json" % o.netloc)
|
||||
isup = getJSON("http://isitup.org/%s.json" % o.netloc)
|
||||
if isup is not None and "status_code" in isup and isup["status_code"] == 1:
|
||||
return isup["response_time"]
|
||||
|
||||
|
|
|
|||
|
|
@ -17,7 +17,7 @@ def validator(url):
|
|||
raise IMException("Indicate a valid URL!")
|
||||
|
||||
try:
|
||||
req = urllib.request.Request("https://validator.w3.org/check?uri=%s&output=json" % (urllib.parse.quote(o.geturl())), headers={ 'User-Agent' : "Nemubot v%s" % __version__})
|
||||
req = urllib.request.Request("http://validator.w3.org/check?uri=%s&output=json" % (urllib.parse.quote(o.geturl())), headers={ 'User-Agent' : "Nemubot v%s" % __version__})
|
||||
raw = urllib.request.urlopen(req, timeout=10)
|
||||
except urllib.error.HTTPError as e:
|
||||
raise IMException("HTTP error occurs: %s %s" % (e.code, e.reason))
|
||||
|
|
|
|||
|
|
@ -1,6 +1,5 @@
|
|||
"""Alert on changes on websites"""
|
||||
|
||||
from functools import partial
|
||||
import logging
|
||||
from random import randint
|
||||
import urllib.parse
|
||||
|
|
@ -13,7 +12,7 @@ from nemubot.tools.xmlparser.node import ModuleState
|
|||
|
||||
logger = logging.getLogger("nemubot.module.networking.watchWebsite")
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
from . import page
|
||||
|
||||
|
|
@ -210,14 +209,15 @@ def start_watching(site, offset=0):
|
|||
offset -- offset time to delay the launch of the first check
|
||||
"""
|
||||
|
||||
#o = urlparse(getNormalizedURL(site["url"]), "http")
|
||||
#print("Add %s event for site: %s" % (site["type"], o.netloc))
|
||||
o = urlparse(getNormalizedURL(site["url"]), "http")
|
||||
#print_debug("Add %s event for site: %s" % (site["type"], o.netloc))
|
||||
|
||||
try:
|
||||
evt = ModuleEvent(func=partial(fwatch, url=site["url"]),
|
||||
cmp=site["lastcontent"],
|
||||
offset=offset, interval=site.getInt("time"),
|
||||
call=partial(alert_change, site=site))
|
||||
evt = ModuleEvent(func=fwatch,
|
||||
cmp_data=site["lastcontent"],
|
||||
func_data=site["url"], offset=offset,
|
||||
interval=site.getInt("time"),
|
||||
call=alert_change, call_data=site)
|
||||
site["_evt_id"] = add_event(evt)
|
||||
except IMException:
|
||||
logger.exception("Unable to watch %s", site["url"])
|
||||
|
|
|
|||
|
|
@ -6,10 +6,10 @@ import urllib
|
|||
from nemubot.exception import IMException
|
||||
from nemubot.tools.web import getJSON
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
URL_AVAIL = "https://www.whoisxmlapi.com/whoisserver/WhoisService?cmd=GET_DN_AVAILABILITY&domainName=%%s&outputFormat=json&username=%s&password=%s"
|
||||
URL_WHOIS = "https://www.whoisxmlapi.com/whoisserver/WhoisService?da=2&domainName=%%s&outputFormat=json&userName=%s&password=%s"
|
||||
URL_WHOIS = "http://www.whoisxmlapi.com/whoisserver/WhoisService?da=2&domainName=%%s&outputFormat=json&userName=%s&password=%s"
|
||||
|
||||
|
||||
# LOADING #############################################################
|
||||
|
|
@ -22,7 +22,7 @@ def load(CONF, add_hook):
|
|||
"the !netwhois feature. Add it to the module "
|
||||
"configuration file:\n<whoisxmlapi username=\"XX\" "
|
||||
"password=\"XXX\" />\nRegister at "
|
||||
"https://www.whoisxmlapi.com/newaccount.php")
|
||||
"http://www.whoisxmlapi.com/newaccount.php")
|
||||
|
||||
URL_AVAIL = URL_AVAIL % (urllib.parse.quote(CONF.getNode("whoisxmlapi")["username"]), urllib.parse.quote(CONF.getNode("whoisxmlapi")["password"]))
|
||||
URL_WHOIS = URL_WHOIS % (urllib.parse.quote(CONF.getNode("whoisxmlapi")["username"]), urllib.parse.quote(CONF.getNode("whoisxmlapi")["password"]))
|
||||
|
|
|
|||
|
|
@ -12,8 +12,7 @@ from nemubot.exception import IMException
|
|||
from nemubot.hooks import hook
|
||||
from nemubot.tools import web
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from nemubot.module.urlreducer import reduce_inline
|
||||
from more import Response
|
||||
from nemubot.tools.feed import Feed, AtomEntry
|
||||
|
||||
|
||||
|
|
@ -51,11 +50,10 @@ def cmd_news(msg):
|
|||
links = [x for x in find_rss_links(url)]
|
||||
if len(links) == 0: links = [ url ]
|
||||
|
||||
res = Response(channel=msg.channel, nomore="No more news from %s" % url, line_treat=reduce_inline)
|
||||
res = Response(channel=msg.channel, nomore="No more news from %s" % url)
|
||||
for n in get_last_news(links[0]):
|
||||
res.append_message("%s published %s: %s %s" % (("\x02" + web.striphtml(n.title) + "\x0F") if n.title else "An article without title",
|
||||
(n.updated.strftime("on %A %d. %B %Y at %H:%M") if n.updated else "someday") if isinstance(n, AtomEntry) else n.pubDate,
|
||||
web.striphtml(n.summary) if n.summary else "",
|
||||
n.link if n.link else ""))
|
||||
|
||||
return res
|
||||
|
|
|
|||
4
modules/nextstop.xml
Normal file
4
modules/nextstop.xml
Normal file
|
|
@ -0,0 +1,4 @@
|
|||
<?xml version="1.0" ?>
|
||||
<nemubotmodule name="nextstop">
|
||||
<message type="cmd" name="ratp" call="ask_ratp" />
|
||||
</nemubotmodule>
|
||||
55
modules/nextstop/__init__.py
Normal file
55
modules/nextstop/__init__.py
Normal file
|
|
@ -0,0 +1,55 @@
|
|||
# coding=utf-8
|
||||
|
||||
"""Informe les usagers des prochains passages des transports en communs de la RATP"""
|
||||
|
||||
from nemubot.exception import IMException
|
||||
from nemubot.hooks import hook
|
||||
from more import Response
|
||||
|
||||
nemubotversion = 3.4
|
||||
|
||||
from .external.src import ratp
|
||||
|
||||
def help_full ():
|
||||
return "!ratp transport line [station]: Donne des informations sur les prochains passages du transport en commun séléctionné à l'arrêt désiré. Si aucune station n'est précisée, les liste toutes."
|
||||
|
||||
|
||||
@hook.command("ratp")
|
||||
def ask_ratp(msg):
|
||||
"""Hook entry from !ratp"""
|
||||
if len(msg.args) >= 3:
|
||||
transport = msg.args[0]
|
||||
line = msg.args[1]
|
||||
station = msg.args[2]
|
||||
if len(msg.args) == 4:
|
||||
times = ratp.getNextStopsAtStation(transport, line, station, msg.args[3])
|
||||
else:
|
||||
times = ratp.getNextStopsAtStation(transport, line, station)
|
||||
|
||||
if len(times) == 0:
|
||||
raise IMException("la station %s n'existe pas sur le %s ligne %s." % (station, transport, line))
|
||||
|
||||
(time, direction, stationname) = times[0]
|
||||
return Response(message=["\x03\x02%s\x03\x02 direction %s" % (time, direction) for time, direction, stationname in times],
|
||||
title="Prochains passages du %s ligne %s à l'arrêt %s" % (transport, line, stationname),
|
||||
channel=msg.channel)
|
||||
|
||||
elif len(msg.args) == 2:
|
||||
stations = ratp.getAllStations(msg.args[0], msg.args[1])
|
||||
|
||||
if len(stations) == 0:
|
||||
raise IMException("aucune station trouvée.")
|
||||
return Response([s for s in stations], title="Stations", channel=msg.channel)
|
||||
|
||||
else:
|
||||
raise IMException("Mauvais usage, merci de spécifier un type de transport et une ligne, ou de consulter l'aide du module.")
|
||||
|
||||
@hook.command("ratp_alert")
|
||||
def ratp_alert(msg):
|
||||
if len(msg.args) == 2:
|
||||
transport = msg.args[0]
|
||||
cause = msg.args[1]
|
||||
incidents = ratp.getDisturbance(cause, transport)
|
||||
return Response(incidents, channel=msg.channel, nomore="No more incidents", count=" (%d more incidents)")
|
||||
else:
|
||||
raise IMException("Mauvais usage, merci de spécifier un type de transport et un type d'alerte (alerte, manif, travaux), ou de consulter l'aide du module.")
|
||||
1
modules/nextstop/external
Submodule
1
modules/nextstop/external
Submodule
|
|
@ -0,0 +1 @@
|
|||
Subproject commit 3d5c9b2d52fbd214f5aaad00e5f3952de919b3e5
|
||||
229
modules/nntp.py
229
modules/nntp.py
|
|
@ -1,229 +0,0 @@
|
|||
"""The NNTP module"""
|
||||
|
||||
# PYTHON STUFFS #######################################################
|
||||
|
||||
import email
|
||||
import email.policy
|
||||
from email.utils import mktime_tz, parseaddr, parsedate_tz
|
||||
from functools import partial
|
||||
from nntplib import NNTP, decode_header
|
||||
import re
|
||||
import time
|
||||
from datetime import datetime
|
||||
from zlib import adler32
|
||||
|
||||
from nemubot import context
|
||||
from nemubot.event import ModuleEvent
|
||||
from nemubot.exception import IMException
|
||||
from nemubot.hooks import hook
|
||||
from nemubot.tools.xmlparser.node import ModuleState
|
||||
|
||||
from nemubot.module.more import Response
|
||||
|
||||
|
||||
# LOADING #############################################################
|
||||
|
||||
def load(context):
|
||||
for wn in context.data.getNodes("watched_newsgroup"):
|
||||
watch(**wn.attributes)
|
||||
|
||||
|
||||
# MODULE CORE #########################################################
|
||||
|
||||
def list_groups(group_pattern="*", **server):
|
||||
with NNTP(**server) as srv:
|
||||
response, l = srv.list(group_pattern)
|
||||
for i in l:
|
||||
yield i.group, srv.description(i.group), i.flag
|
||||
|
||||
def read_group(group, **server):
|
||||
with NNTP(**server) as srv:
|
||||
response, count, first, last, name = srv.group(group)
|
||||
resp, overviews = srv.over((first, last))
|
||||
for art_num, over in reversed(overviews):
|
||||
yield over
|
||||
|
||||
def read_article(msg_id, **server):
|
||||
with NNTP(**server) as srv:
|
||||
response, info = srv.article(msg_id)
|
||||
return email.message_from_bytes(b"\r\n".join(info.lines), policy=email.policy.SMTPUTF8)
|
||||
|
||||
|
||||
servers_lastcheck = dict()
|
||||
servers_lastseen = dict()
|
||||
|
||||
def whatsnew(group="*", **server):
|
||||
fill = dict()
|
||||
if "user" in server: fill["user"] = server["user"]
|
||||
if "password" in server: fill["password"] = server["password"]
|
||||
if "host" in server: fill["host"] = server["host"]
|
||||
if "port" in server: fill["port"] = server["port"]
|
||||
|
||||
idx = _indexServer(**server)
|
||||
if idx in servers_lastcheck and servers_lastcheck[idx] is not None:
|
||||
date_last_check = servers_lastcheck[idx]
|
||||
else:
|
||||
date_last_check = datetime.now()
|
||||
|
||||
if idx not in servers_lastseen:
|
||||
servers_lastseen[idx] = []
|
||||
|
||||
with NNTP(**fill) as srv:
|
||||
response, servers_lastcheck[idx] = srv.date()
|
||||
|
||||
response, groups = srv.newgroups(date_last_check)
|
||||
for g in groups:
|
||||
yield g
|
||||
|
||||
response, articles = srv.newnews(group, date_last_check)
|
||||
for msg_id in articles:
|
||||
if msg_id not in servers_lastseen[idx]:
|
||||
servers_lastseen[idx].append(msg_id)
|
||||
response, info = srv.article(msg_id)
|
||||
yield email.message_from_bytes(b"\r\n".join(info.lines))
|
||||
|
||||
# Clean huge lists
|
||||
if len(servers_lastseen[idx]) > 42:
|
||||
servers_lastseen[idx] = servers_lastseen[idx][23:]
|
||||
|
||||
|
||||
def format_article(art, **response_args):
|
||||
art["X-FromName"], art["X-FromEmail"] = parseaddr(art["From"] if "From" in art else "")
|
||||
if art["X-FromName"] == '': art["X-FromName"] = art["X-FromEmail"]
|
||||
|
||||
date = mktime_tz(parsedate_tz(art["Date"]))
|
||||
if date < time.time() - 120:
|
||||
title = "\x0314In \x0F\x03{0:02d}{Newsgroups}\x0F\x0314: on \x0F{Date}\x0314 by \x0F\x03{0:02d}{X-FromName}\x0F \x02{Subject}\x0F"
|
||||
else:
|
||||
title = "\x0314In \x0F\x03{0:02d}{Newsgroups}\x0F\x0314: by \x0F\x03{0:02d}{X-FromName}\x0F \x02{Subject}\x0F"
|
||||
|
||||
return Response(art.get_payload().replace('\n', ' '),
|
||||
title=title.format(adler32(art["Newsgroups"].encode()) & 0xf, adler32(art["X-FromEmail"].encode()) & 0xf, **{h: decode_header(i) for h,i in art.items()}),
|
||||
**response_args)
|
||||
|
||||
|
||||
watches = dict()
|
||||
|
||||
def _indexServer(**kwargs):
|
||||
if "user" not in kwargs: kwargs["user"] = ""
|
||||
if "password" not in kwargs: kwargs["password"] = ""
|
||||
if "host" not in kwargs: kwargs["host"] = ""
|
||||
if "port" not in kwargs: kwargs["port"] = 119
|
||||
return "{user}:{password}@{host}:{port}".format(**kwargs)
|
||||
|
||||
def _newevt(**args):
|
||||
context.add_event(ModuleEvent(call=partial(_ticker, **args), interval=42))
|
||||
|
||||
def _ticker(to_server, to_channel, group, server):
|
||||
_newevt(to_server=to_server, to_channel=to_channel, group=group, server=server)
|
||||
n = 0
|
||||
for art in whatsnew(group, **server):
|
||||
n += 1
|
||||
if n > 10:
|
||||
continue
|
||||
context.send_response(to_server, format_article(art, channel=to_channel))
|
||||
if n > 10:
|
||||
context.send_response(to_server, Response("... and %s others news" % (n - 10), channel=to_channel))
|
||||
|
||||
def watch(to_server, to_channel, group="*", **server):
|
||||
_newevt(to_server=to_server, to_channel=to_channel, group=group, server=server)
|
||||
|
||||
|
||||
# MODULE INTERFACE ####################################################
|
||||
|
||||
keywords_server = {
|
||||
"host=HOST": "hostname or IP of the NNTP server",
|
||||
"port=PORT": "port of the NNTP server",
|
||||
"user=USERNAME": "username to use to connect to the server",
|
||||
"password=PASSWORD": "password to use to connect to the server",
|
||||
}
|
||||
|
||||
@hook.command("nntp_groups",
|
||||
help="Show list of existing groups",
|
||||
help_usage={
|
||||
None: "Display all groups",
|
||||
"PATTERN": "Filter on group matching the PATTERN"
|
||||
},
|
||||
keywords=keywords_server)
|
||||
def cmd_groups(msg):
|
||||
if "host" not in msg.kwargs:
|
||||
raise IMException("please give a hostname in keywords")
|
||||
|
||||
return Response(["\x02\x03{0:02d}{1}\x0F: {2}".format(adler32(g[0].encode()) & 0xf, *g) for g in list_groups(msg.args[0] if len(msg.args) > 0 else "*", **msg.kwargs)],
|
||||
channel=msg.channel,
|
||||
title="Matching groups on %s" % msg.kwargs["host"])
|
||||
|
||||
|
||||
@hook.command("nntp_overview",
|
||||
help="Show an overview of articles in given group(s)",
|
||||
help_usage={
|
||||
"GROUP": "Filter on group matching the PATTERN"
|
||||
},
|
||||
keywords=keywords_server)
|
||||
def cmd_overview(msg):
|
||||
if "host" not in msg.kwargs:
|
||||
raise IMException("please give a hostname in keywords")
|
||||
|
||||
if not len(msg.args):
|
||||
raise IMException("which group would you overview?")
|
||||
|
||||
for g in msg.args:
|
||||
arts = []
|
||||
for grp in read_group(g, **msg.kwargs):
|
||||
grp["X-FromName"], grp["X-FromEmail"] = parseaddr(grp["from"] if "from" in grp else "")
|
||||
if grp["X-FromName"] == '': grp["X-FromName"] = grp["X-FromEmail"]
|
||||
|
||||
arts.append("On {date}, from \x03{0:02d}{X-FromName}\x0F \x02{subject}\x0F: \x0314{message-id}\x0F".format(adler32(grp["X-FromEmail"].encode()) & 0xf, **{h: decode_header(i) for h,i in grp.items()}))
|
||||
|
||||
if len(arts):
|
||||
yield Response(arts,
|
||||
channel=msg.channel,
|
||||
title="In \x03{0:02d}{1}\x0F".format(adler32(g[0].encode()) & 0xf, g))
|
||||
|
||||
|
||||
@hook.command("nntp_read",
|
||||
help="Read an article from a server",
|
||||
help_usage={
|
||||
"MSG_ID": "Read the given message"
|
||||
},
|
||||
keywords=keywords_server)
|
||||
def cmd_read(msg):
|
||||
if "host" not in msg.kwargs:
|
||||
raise IMException("please give a hostname in keywords")
|
||||
|
||||
for msgid in msg.args:
|
||||
if not re.match("<.*>", msgid):
|
||||
msgid = "<" + msgid + ">"
|
||||
art = read_article(msgid, **msg.kwargs)
|
||||
yield format_article(art, channel=msg.channel)
|
||||
|
||||
|
||||
@hook.command("nntp_watch",
|
||||
help="Launch an event looking for new groups and articles on a server",
|
||||
help_usage={
|
||||
None: "Watch all groups",
|
||||
"PATTERN": "Limit the watch on group matching this PATTERN"
|
||||
},
|
||||
keywords=keywords_server)
|
||||
def cmd_watch(msg):
|
||||
if "host" not in msg.kwargs:
|
||||
raise IMException("please give a hostname in keywords")
|
||||
|
||||
if not msg.frm_owner:
|
||||
raise IMException("sorry, this command is currently limited to the owner")
|
||||
|
||||
wnnode = ModuleState("watched_newsgroup")
|
||||
wnnode["id"] = _indexServer(**msg.kwargs)
|
||||
wnnode["to_server"] = msg.server
|
||||
wnnode["to_channel"] = msg.channel
|
||||
wnnode["group"] = msg.args[0] if len(msg.args) > 0 else "*"
|
||||
|
||||
wnnode["user"] = msg.kwargs["user"] if "user" in msg.kwargs else ""
|
||||
wnnode["password"] = msg.kwargs["password"] if "password" in msg.kwargs else ""
|
||||
wnnode["host"] = msg.kwargs["host"] if "host" in msg.kwargs else ""
|
||||
wnnode["port"] = msg.kwargs["port"] if "port" in msg.kwargs else 119
|
||||
|
||||
context.data.addChild(wnnode)
|
||||
watch(**wnnode.attributes)
|
||||
|
||||
return Response("Ok ok, I watch this newsgroup!", channel=msg.channel)
|
||||
|
|
@ -1,87 +0,0 @@
|
|||
"""Perform requests to openai"""
|
||||
|
||||
# PYTHON STUFFS #######################################################
|
||||
|
||||
from openai import OpenAI
|
||||
|
||||
from nemubot import context
|
||||
from nemubot.hooks import hook
|
||||
from nemubot.tools import web
|
||||
|
||||
from nemubot.module.more import Response
|
||||
|
||||
|
||||
# LOADING #############################################################
|
||||
|
||||
CLIENT = None
|
||||
MODEL = "gpt-4"
|
||||
ENDPOINT = None
|
||||
|
||||
def load(context):
|
||||
global CLIENT, ENDPOINT, MODEL
|
||||
if not context.config or ("apikey" not in context.config and "endpoint" not in context.config):
|
||||
raise ImportError ("You need a OpenAI API key in order to use "
|
||||
"this module. Add it to the module configuration: "
|
||||
"\n<module name=\"openai\" "
|
||||
"apikey=\"XXXXXX-XXXXXXXXXX\" endpoint=\"https://...\" model=\"gpt-4\" />")
|
||||
kwargs = {
|
||||
"api_key": context.config["apikey"] or "",
|
||||
}
|
||||
|
||||
if "endpoint" in context.config:
|
||||
ENDPOINT = context.config["endpoint"]
|
||||
kwargs["base_url"] = ENDPOINT
|
||||
|
||||
CLIENT = OpenAI(**kwargs)
|
||||
|
||||
if "model" in context.config:
|
||||
MODEL = context.config["model"]
|
||||
|
||||
|
||||
# MODULE INTERFACE ####################################################
|
||||
|
||||
@hook.command("list_models",
|
||||
help="list available LLM")
|
||||
def cmd_listllm(msg):
|
||||
llms = web.getJSON(ENDPOINT + "/models", timeout=6)
|
||||
return Response(message=[m for m in map(lambda i: i["id"], llms["data"])], title="Here is the available models", channel=msg.channel)
|
||||
|
||||
|
||||
@hook.command("set_model",
|
||||
help="Set the model to use when talking to nemubot")
|
||||
def cmd_setllm(msg):
|
||||
if len(msg.args) != 1:
|
||||
raise IMException("Indicate 1 model to use")
|
||||
|
||||
wanted_model = msg.args[0]
|
||||
|
||||
llms = web.getJSON(ENDPOINT + "/models", timeout=6)
|
||||
for model in llms["data"]:
|
||||
if wanted_model == model["id"]:
|
||||
break
|
||||
else:
|
||||
raise IMException("Unable to set such model: unknown")
|
||||
|
||||
MODEL = wanted_model
|
||||
return Response("New model in use: " + wanted_model, channel=msg.channel)
|
||||
|
||||
|
||||
@hook.ask()
|
||||
def parseask(msg):
|
||||
chat_completion = CLIENT.chat.completions.create(
|
||||
messages=[
|
||||
{
|
||||
"role": "system",
|
||||
"content": "You are a kind multilingual assistant. Respond to the user request in 255 characters maximum. Be conscise, go directly to the point. Never add useless terms.",
|
||||
},
|
||||
{
|
||||
"role": "user",
|
||||
"content": msg.message,
|
||||
}
|
||||
],
|
||||
model=MODEL,
|
||||
)
|
||||
|
||||
return Response(chat_completion.choices[0].message.content,
|
||||
msg.channel,
|
||||
msg.frm)
|
||||
|
|
@ -1,158 +0,0 @@
|
|||
"""Lost? use our commands to find your way!"""
|
||||
|
||||
# PYTHON STUFFS #######################################################
|
||||
|
||||
import re
|
||||
import urllib.parse
|
||||
|
||||
from nemubot.exception import IMException
|
||||
from nemubot.hooks import hook
|
||||
from nemubot.tools import web
|
||||
|
||||
from nemubot.module.more import Response
|
||||
|
||||
# GLOBALS #############################################################
|
||||
|
||||
URL_DIRECTIONS_API = "https://api.openrouteservice.org/directions?api_key=%s&"
|
||||
URL_GEOCODE_API = "https://api.openrouteservice.org/geocoding?api_key=%s&"
|
||||
|
||||
waytype = [
|
||||
"unknown",
|
||||
"state road",
|
||||
"road",
|
||||
"street",
|
||||
"path",
|
||||
"track",
|
||||
"cycleway",
|
||||
"footway",
|
||||
"steps",
|
||||
"ferry",
|
||||
"construction",
|
||||
]
|
||||
|
||||
|
||||
# LOADING #############################################################
|
||||
|
||||
def load(context):
|
||||
if not context.config or "apikey" not in context.config:
|
||||
raise ImportError("You need an OpenRouteService API key in order to use this "
|
||||
"module. Add it to the module configuration file:\n"
|
||||
"<module name=\"ors\" apikey=\"XXXXXXXXXXXXXXXX\" "
|
||||
"/>\nRegister at https://developers.openrouteservice.org")
|
||||
global URL_DIRECTIONS_API
|
||||
URL_DIRECTIONS_API = URL_DIRECTIONS_API % context.config["apikey"]
|
||||
global URL_GEOCODE_API
|
||||
URL_GEOCODE_API = URL_GEOCODE_API % context.config["apikey"]
|
||||
|
||||
|
||||
# MODULE CORE #########################################################
|
||||
|
||||
def approx_distance(lng):
|
||||
if lng > 1111:
|
||||
return "%f km" % (lng / 1000)
|
||||
else:
|
||||
return "%f m" % lng
|
||||
|
||||
|
||||
def approx_duration(sec):
|
||||
days = int(sec / 86400)
|
||||
if days > 0:
|
||||
return "%d days %f hours" % (days, (sec % 86400) / 3600)
|
||||
hours = int((sec % 86400) / 3600)
|
||||
if hours > 0:
|
||||
return "%d hours %f minutes" % (hours, (sec % 3600) / 60)
|
||||
minutes = (sec % 3600) / 60
|
||||
if minutes > 0:
|
||||
return "%d minutes" % minutes
|
||||
else:
|
||||
return "%d seconds" % sec
|
||||
|
||||
|
||||
def geocode(query, limit=7):
|
||||
obj = web.getJSON(URL_GEOCODE_API + urllib.parse.urlencode({
|
||||
'query': query,
|
||||
'limit': limit,
|
||||
}))
|
||||
|
||||
for f in obj["features"]:
|
||||
yield f["geometry"]["coordinates"], f["properties"]
|
||||
|
||||
|
||||
def firstgeocode(query):
|
||||
for g in geocode(query, limit=1):
|
||||
return g
|
||||
|
||||
|
||||
def where(loc):
|
||||
return "{name} {city} {state} {county} {country}".format(**loc)
|
||||
|
||||
|
||||
def directions(coordinates, **kwargs):
|
||||
kwargs['coordinates'] = '|'.join(coordinates)
|
||||
|
||||
print(URL_DIRECTIONS_API + urllib.parse.urlencode(kwargs))
|
||||
return web.getJSON(URL_DIRECTIONS_API + urllib.parse.urlencode(kwargs), decode_error=True)
|
||||
|
||||
|
||||
# MODULE INTERFACE ####################################################
|
||||
|
||||
@hook.command("geocode",
|
||||
help="Get GPS coordinates of a place",
|
||||
help_usage={
|
||||
"PLACE": "Get GPS coordinates of PLACE"
|
||||
})
|
||||
def cmd_geocode(msg):
|
||||
res = Response(channel=msg.channel, nick=msg.frm,
|
||||
nomore="No more geocode", count=" (%s more geocode)")
|
||||
|
||||
for loc in geocode(' '.join(msg.args)):
|
||||
res.append_message("%s is at %s,%s" % (
|
||||
where(loc[1]),
|
||||
loc[0][1], loc[0][0],
|
||||
))
|
||||
|
||||
return res
|
||||
|
||||
|
||||
@hook.command("directions",
|
||||
help="Get routing instructions",
|
||||
help_usage={
|
||||
"POINT1 POINT2 ...": "Get routing instructions to go from POINT1 to the last POINTX via intermediates POINTX"
|
||||
},
|
||||
keywords={
|
||||
"profile=PROF": "One of driving-car, driving-hgv, cycling-regular, cycling-road, cycling-safe, cycling-mountain, cycling-tour, cycling-electric, foot-walking, foot-hiking, wheelchair. Default: foot-walking",
|
||||
"preference=PREF": "One of fastest, shortest, recommended. Default: recommended",
|
||||
"lang=LANG": "default: en",
|
||||
})
|
||||
def cmd_directions(msg):
|
||||
drcts = directions(["{0},{1}".format(*firstgeocode(g)[0]) for g in msg.args],
|
||||
profile=msg.kwargs["profile"] if "profile" in msg.kwargs else "foot-walking",
|
||||
preference=msg.kwargs["preference"] if "preference" in msg.kwargs else "recommended",
|
||||
units="m",
|
||||
language=msg.kwargs["lang"] if "lang" in msg.kwargs else "en",
|
||||
geometry=False,
|
||||
instructions=True,
|
||||
instruction_format="text")
|
||||
if "error" in drcts and "message" in drcts["error"] and drcts["error"]["message"]:
|
||||
raise IMException(drcts["error"]["message"])
|
||||
|
||||
if "routes" not in drcts or not drcts["routes"]:
|
||||
raise IMException("No route available for this trip")
|
||||
|
||||
myway = drcts["routes"][0]
|
||||
myway["summary"]["strduration"] = approx_duration(myway["summary"]["duration"])
|
||||
myway["summary"]["strdistance"] = approx_distance(myway["summary"]["distance"])
|
||||
res = Response("Trip summary: {strdistance} in approximate {strduration}; elevation +{ascent} m -{descent} m".format(**myway["summary"]), channel=msg.channel, count=" (%d more steps)", nomore="You have arrived!")
|
||||
|
||||
def formatSegments(segments):
|
||||
for segment in segments:
|
||||
for step in segment["steps"]:
|
||||
step["strtype"] = waytype[step["type"]]
|
||||
step["strduration"] = approx_duration(step["duration"])
|
||||
step["strdistance"] = approx_distance(step["distance"])
|
||||
yield "{instruction} for {strdistance} on {strtype} (approximate time: {strduration})".format(**step)
|
||||
|
||||
if "segments" in myway:
|
||||
res.append_message([m for m in formatSegments(myway["segments"])])
|
||||
|
||||
return res
|
||||
|
|
@ -1,68 +0,0 @@
|
|||
"""Get information about common software"""
|
||||
|
||||
# PYTHON STUFFS #######################################################
|
||||
|
||||
import portage
|
||||
|
||||
from nemubot import context
|
||||
from nemubot.exception import IMException
|
||||
from nemubot.hooks import hook
|
||||
|
||||
from nemubot.module.more import Response
|
||||
|
||||
DB = None
|
||||
|
||||
# MODULE CORE #########################################################
|
||||
|
||||
def get_db():
|
||||
global DB
|
||||
if DB is None:
|
||||
DB = portage.db[portage.root]["porttree"].dbapi
|
||||
return DB
|
||||
|
||||
|
||||
def package_info(pkgname):
|
||||
pv = get_db().xmatch("match-all", pkgname)
|
||||
if not pv:
|
||||
raise IMException("No package named '%s' found" % pkgname)
|
||||
|
||||
bv = get_db().xmatch("bestmatch-visible", pkgname)
|
||||
pvsplit = portage.catpkgsplit(bv if bv else pv[-1])
|
||||
info = get_db().aux_get(bv if bv else pv[-1], ["DESCRIPTION", "HOMEPAGE", "LICENSE", "IUSE", "KEYWORDS"])
|
||||
|
||||
return {
|
||||
"pkgname": '/'.join(pvsplit[:2]),
|
||||
"category": pvsplit[0],
|
||||
"shortname": pvsplit[1],
|
||||
"lastvers": '-'.join(pvsplit[2:]) if pvsplit[3] != "r0" else pvsplit[2],
|
||||
"othersvers": ['-'.join(portage.catpkgsplit(p)[2:]) for p in pv if p != bv],
|
||||
"description": info[0],
|
||||
"homepage": info[1],
|
||||
"license": info[2],
|
||||
"uses": info[3],
|
||||
"keywords": info[4],
|
||||
}
|
||||
|
||||
|
||||
# MODULE INTERFACE ####################################################
|
||||
|
||||
@hook.command("eix",
|
||||
help="Get information about a package",
|
||||
help_usage={
|
||||
"NAME": "Get information about a software NAME"
|
||||
})
|
||||
def cmd_eix(msg):
|
||||
if not len(msg.args):
|
||||
raise IMException("please give me a package to search")
|
||||
|
||||
def srch(term):
|
||||
try:
|
||||
yield package_info(term)
|
||||
except portage.exception.AmbiguousPackageName as e:
|
||||
for i in e.args[0]:
|
||||
yield package_info(i)
|
||||
|
||||
res = Response(channel=msg.channel, count=" (%d more packages)", nomore="No more package '%s'" % msg.args[0])
|
||||
for pi in srch(msg.args[0]):
|
||||
res.append_message("\x03\x02{pkgname}:\x03\x02 {description} - {homepage} - {license} - last revisions: \x03\x02{lastvers}\x03\x02{ov}".format(ov=(", " + ', '.join(pi["othersvers"])) if pi["othersvers"] else "", **pi))
|
||||
return res
|
||||
|
|
@ -1,74 +0,0 @@
|
|||
"""Informe les usagers des prochains passages des transports en communs de la RATP"""
|
||||
|
||||
# PYTHON STUFFS #######################################################
|
||||
|
||||
from nemubot.exception import IMException
|
||||
from nemubot.hooks import hook
|
||||
from nemubot.module.more import Response
|
||||
|
||||
from nextstop import ratp
|
||||
|
||||
@hook.command("ratp",
|
||||
help="Affiche les prochains horaires de passage",
|
||||
help_usage={
|
||||
"TRANSPORT": "Affiche les lignes du moyen de transport donné",
|
||||
"TRANSPORT LINE": "Affiche les stations sur la ligne de transport donnée",
|
||||
"TRANSPORT LINE STATION": "Affiche les prochains horaires de passage à l'arrêt donné",
|
||||
"TRANSPORT LINE STATION DESTINATION": "Affiche les prochains horaires de passage dans la direction donnée",
|
||||
})
|
||||
def ask_ratp(msg):
|
||||
l = len(msg.args)
|
||||
|
||||
transport = msg.args[0] if l > 0 else None
|
||||
line = msg.args[1] if l > 1 else None
|
||||
station = msg.args[2] if l > 2 else None
|
||||
direction = msg.args[3] if l > 3 else None
|
||||
|
||||
if station is not None:
|
||||
times = sorted(ratp.getNextStopsAtStation(transport, line, station, direction), key=lambda i: i[0])
|
||||
|
||||
if len(times) == 0:
|
||||
raise IMException("la station %s n'existe pas sur le %s ligne %s." % (station, transport, line))
|
||||
|
||||
(time, direction, stationname) = times[0]
|
||||
return Response(message=["\x03\x02%s\x03\x02 direction %s" % (time, direction) for time, direction, stationname in times],
|
||||
title="Prochains passages du %s ligne %s à l'arrêt %s" % (transport, line, stationname),
|
||||
channel=msg.channel)
|
||||
|
||||
elif line is not None:
|
||||
stations = ratp.getAllStations(transport, line)
|
||||
|
||||
if len(stations) == 0:
|
||||
raise IMException("aucune station trouvée.")
|
||||
return Response(stations, title="Stations", channel=msg.channel)
|
||||
|
||||
elif transport is not None:
|
||||
lines = ratp.getTransportLines(transport)
|
||||
if len(lines) == 0:
|
||||
raise IMException("aucune ligne trouvée.")
|
||||
return Response(lines, title="Lignes", channel=msg.channel)
|
||||
|
||||
else:
|
||||
raise IMException("précise au moins un moyen de transport.")
|
||||
|
||||
|
||||
@hook.command("ratp_alert",
|
||||
help="Affiche les perturbations en cours sur le réseau")
|
||||
def ratp_alert(msg):
|
||||
if len(msg.args) == 0:
|
||||
raise IMException("précise au moins un moyen de transport.")
|
||||
|
||||
l = len(msg.args)
|
||||
transport = msg.args[0] if l > 0 else None
|
||||
line = msg.args[1] if l > 1 else None
|
||||
|
||||
if line is not None:
|
||||
d = ratp.getDisturbanceFromLine(transport, line)
|
||||
if "date" in d and d["date"] is not None:
|
||||
incidents = "Au {date[date]}, {title}: {message}".format(**d)
|
||||
else:
|
||||
incidents = "{title}: {message}".format(**d)
|
||||
else:
|
||||
incidents = ratp.getDisturbance(None, transport)
|
||||
|
||||
return Response(incidents, channel=msg.channel, nomore="No more incidents", count=" (%d more incidents)")
|
||||
|
|
@ -10,7 +10,7 @@ from nemubot.tools import web
|
|||
|
||||
nemubotversion = 3.4
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
|
||||
def help_full():
|
||||
|
|
@ -40,7 +40,7 @@ def cmd_subreddit(msg):
|
|||
else:
|
||||
where = "r"
|
||||
|
||||
sbr = web.getJSON("https://www.reddit.com/%s/%s/about.json" %
|
||||
sbr = web.getJSON("http://www.reddit.com/%s/%s/about.json" %
|
||||
(where, sub.group(2)))
|
||||
|
||||
if sbr is None:
|
||||
|
|
@ -64,29 +64,22 @@ def cmd_subreddit(msg):
|
|||
channel=msg.channel))
|
||||
else:
|
||||
all_res.append(Response("%s is not a valid subreddit" % osub,
|
||||
channel=msg.channel, nick=msg.frm))
|
||||
channel=msg.channel, nick=msg.nick))
|
||||
|
||||
return all_res
|
||||
|
||||
|
||||
@hook.message()
|
||||
def parselisten(msg):
|
||||
global LAST_SUBS
|
||||
|
||||
if hasattr(msg, "message") and msg.message and type(msg.message) == str:
|
||||
urls = re.findall("www.reddit.com(/\w/\w+/?)", msg.message)
|
||||
for url in urls:
|
||||
for recv in msg.to:
|
||||
if recv not in LAST_SUBS:
|
||||
LAST_SUBS[recv] = list()
|
||||
LAST_SUBS[recv].append(url)
|
||||
parseresponse(msg)
|
||||
return None
|
||||
|
||||
|
||||
@hook.post()
|
||||
def parseresponse(msg):
|
||||
global LAST_SUBS
|
||||
|
||||
if hasattr(msg, "text") and msg.text and type(msg.text) == str:
|
||||
if hasattr(msg, "text") and msg.text:
|
||||
urls = re.findall("www.reddit.com(/\w/\w+/?)", msg.text)
|
||||
for url in urls:
|
||||
for recv in msg.to:
|
||||
|
|
|
|||
|
|
@ -1,94 +0,0 @@
|
|||
# coding=utf-8
|
||||
|
||||
"""Repology.org module: the packaging hub"""
|
||||
|
||||
import datetime
|
||||
import re
|
||||
|
||||
from nemubot import context
|
||||
from nemubot.exception import IMException
|
||||
from nemubot.hooks import hook
|
||||
from nemubot.tools import web
|
||||
from nemubot.tools.xmlparser.node import ModuleState
|
||||
|
||||
nemubotversion = 4.0
|
||||
|
||||
from nemubot.module.more import Response
|
||||
|
||||
URL_REPOAPI = "https://repology.org/api/v1/project/%s"
|
||||
|
||||
def get_json_project(project):
|
||||
prj = web.getJSON(URL_REPOAPI % (project))
|
||||
|
||||
return prj
|
||||
|
||||
|
||||
@hook.command("repology",
|
||||
help="Display version information about a package",
|
||||
help_usage={
|
||||
"PACKAGE_NAME": "Retrieve informations about PACKAGE_NAME",
|
||||
},
|
||||
keywords={
|
||||
"distro=DISTRO": "filter by disto",
|
||||
"status=STATUS[,STATUS...]": "filter by status",
|
||||
})
|
||||
def cmd_repology(msg):
|
||||
if len(msg.args) == 0:
|
||||
raise IMException("Please provide at least a package name")
|
||||
|
||||
res = Response(channel=msg.channel, nomore="No more information on package")
|
||||
|
||||
for project in msg.args:
|
||||
prj = get_json_project(project)
|
||||
if len(prj) == 0:
|
||||
raise IMException("Unable to find package " + project)
|
||||
|
||||
pkg_versions = {}
|
||||
pkg_maintainers = {}
|
||||
pkg_licenses = {}
|
||||
summary = None
|
||||
|
||||
for repo in prj:
|
||||
# Apply filters
|
||||
if "distro" in msg.kwargs and repo["repo"].find(msg.kwargs["distro"]) < 0:
|
||||
continue
|
||||
if "status" in msg.kwargs and repo["status"] not in msg.kwargs["status"].split(","):
|
||||
continue
|
||||
|
||||
name = repo["visiblename"] if "visiblename" in repo else repo["name"]
|
||||
status = repo["status"] if "status" in repo else "unknown"
|
||||
if name not in pkg_versions:
|
||||
pkg_versions[name] = {}
|
||||
if status not in pkg_versions[name]:
|
||||
pkg_versions[name][status] = []
|
||||
if repo["version"] not in pkg_versions[name][status]:
|
||||
pkg_versions[name][status].append(repo["version"])
|
||||
|
||||
if "maintainers" in repo:
|
||||
if name not in pkg_maintainers:
|
||||
pkg_maintainers[name] = []
|
||||
for maintainer in repo["maintainers"]:
|
||||
if maintainer not in pkg_maintainers[name]:
|
||||
pkg_maintainers[name].append(maintainer)
|
||||
|
||||
if "licenses" in repo:
|
||||
if name not in pkg_licenses:
|
||||
pkg_licenses[name] = []
|
||||
for lic in repo["licenses"]:
|
||||
if lic not in pkg_licenses[name]:
|
||||
pkg_licenses[name].append(lic)
|
||||
|
||||
if "summary" in repo and summary is None:
|
||||
summary = repo["summary"]
|
||||
|
||||
for pkgname in sorted(pkg_versions.keys()):
|
||||
m = "Package " + pkgname + " (" + summary + ")"
|
||||
if pkgname in pkg_licenses:
|
||||
m += " under " + ", ".join(pkg_licenses[pkgname])
|
||||
m += ": " + " - ".join([status + ": " + ", ".join(pkg_versions[pkgname][status]) for status in ["newest", "devel", "unique", "outdated", "legacy", "rolling", "noscheme", "untrusted", "ignored"] if status in pkg_versions[pkgname]])
|
||||
if "distro" in msg.kwargs and pkgname in pkg_maintainers:
|
||||
m += " - Maintained by " + ", ".join(pkg_maintainers[pkgname])
|
||||
|
||||
res.append_message(m)
|
||||
|
||||
return res
|
||||
|
|
@ -9,7 +9,7 @@ from nemubot import context
|
|||
from nemubot.exception import IMException
|
||||
from nemubot.hooks import hook
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
|
||||
# MODULE INTERFACE ####################################################
|
||||
|
|
@ -21,7 +21,7 @@ def cmd_choice(msg):
|
|||
|
||||
return Response(random.choice(msg.args),
|
||||
channel=msg.channel,
|
||||
nick=msg.frm)
|
||||
nick=msg.nick)
|
||||
|
||||
|
||||
@hook.command("choicecmd")
|
||||
|
|
|
|||
|
|
@ -12,7 +12,7 @@ from nemubot.tools import web
|
|||
|
||||
nemubotversion = 4.0
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
|
||||
def help_full():
|
||||
|
|
@ -25,7 +25,7 @@ def cmd_tcode(msg):
|
|||
raise IMException("indicate a transaction code or "
|
||||
"a keyword to search!")
|
||||
|
||||
url = ("https://www.tcodesearch.com/tcodes/search?q=%s" %
|
||||
url = ("http://www.tcodesearch.com/tcodes/search?q=%s" %
|
||||
urllib.parse.quote(msg.args[0]))
|
||||
|
||||
page = web.getURLContent(url)
|
||||
|
|
|
|||
|
|
@ -1,104 +0,0 @@
|
|||
"""Search engine for IoT"""
|
||||
|
||||
# PYTHON STUFFS #######################################################
|
||||
|
||||
from datetime import datetime
|
||||
import ipaddress
|
||||
import urllib.parse
|
||||
|
||||
from nemubot import context
|
||||
from nemubot.exception import IMException
|
||||
from nemubot.hooks import hook
|
||||
from nemubot.tools import web
|
||||
|
||||
from nemubot.module.more import Response
|
||||
|
||||
|
||||
# GLOBALS #############################################################
|
||||
|
||||
BASEURL = "https://api.shodan.io/shodan/"
|
||||
|
||||
|
||||
# LOADING #############################################################
|
||||
|
||||
def load(context):
|
||||
if not context.config or "apikey" not in context.config:
|
||||
raise ImportError("You need a Shodan API key in order to use this "
|
||||
"module. Add it to the module configuration file:\n"
|
||||
"<module name=\"shodan\" apikey=\"XXXXXXXXXXXXXXXX\" "
|
||||
"/>\nRegister at https://account.shodan.io/register")
|
||||
|
||||
|
||||
# MODULE CORE #########################################################
|
||||
|
||||
def host_lookup(ip):
|
||||
url = BASEURL + "host/" + urllib.parse.quote(ip) + "?" + urllib.parse.urlencode({'key': context.config["apikey"]})
|
||||
return web.getJSON(url)
|
||||
|
||||
|
||||
def search_hosts(query):
|
||||
url = BASEURL + "host/search?" + urllib.parse.urlencode({'query': query, 'key': context.config["apikey"]})
|
||||
return web.getJSON(url, max_size=4194304)
|
||||
|
||||
|
||||
def print_ssl(ssl):
|
||||
return (
|
||||
"SSL: " +
|
||||
" ".join([v for v in ssl["versions"] if v[0] != "-"]) +
|
||||
"; cipher used: " + ssl["cipher"]["name"] +
|
||||
("; certificate: " + ssl["cert"]["sig_alg"] +
|
||||
" issued by: " + ssl["cert"]["issuer"]["CN"] +
|
||||
" expires on: " + str(datetime.strptime(ssl["cert"]["expires"], "%Y%m%d%H%M%SZ")) if "cert" in ssl else "")
|
||||
)
|
||||
|
||||
def print_service(svc):
|
||||
ip = ipaddress.ip_address(svc["ip_str"])
|
||||
return ((svc["ip_str"] if ip.version == 4 else "[%s]" % svc["ip_str"]) +
|
||||
":{port}/{transport} ({module}):" +
|
||||
(" {os}" if svc["os"] else "") +
|
||||
(" {product}" if "product" in svc else "") +
|
||||
(" {version}" if "version" in svc else "") +
|
||||
(" {info}" if "info" in svc else "") +
|
||||
(" Vulns: " + ", ".join(svc["opts"]["vulns"]) if "opts" in svc and "vulns" in svc["opts"] else "") +
|
||||
(" " + print_ssl(svc["ssl"]) if "ssl" in svc else "") +
|
||||
(" \x03\x1D" + svc["data"].replace("\r\n", "\n").split("\n")[0] + "\x03\x1D" if "data" in svc else "") +
|
||||
(" " + svc["title"] if "title" in svc else "")
|
||||
).format(module=svc["_shodan"]["module"], **svc)
|
||||
|
||||
|
||||
# MODULE INTERFACE ####################################################
|
||||
|
||||
@hook.command("shodan",
|
||||
help="Use shodan.io to get information on machines connected to Internet",
|
||||
help_usage={
|
||||
"IP": "retrieve information about the given IP (can be v4 or v6)",
|
||||
"TERM": "retrieve all hosts matching TERM somewhere in their exposed stuff"
|
||||
})
|
||||
def shodan(msg):
|
||||
if not msg.args:
|
||||
raise IMException("indicate an IP or a term to search!")
|
||||
|
||||
terms = " ".join(msg.args)
|
||||
|
||||
try:
|
||||
ip = ipaddress.ip_address(terms)
|
||||
except ValueError:
|
||||
ip = None
|
||||
|
||||
if ip:
|
||||
h = host_lookup(terms)
|
||||
res = Response(channel=msg.channel,
|
||||
title="%s" % ((h["ip_str"] if ip.version == 4 else "[%s]" % h["ip_str"]) + (" (" + ", ".join(h["hostnames"]) + ")") if h["hostnames"] else ""))
|
||||
res.append_message("{isp} ({asn}) -> {city} ({country_code}), running {os}. Vulns: {vulns_str}. Open ports: {open_ports}. Last update: {last_update}".format(
|
||||
open_ports=", ".join(map(lambda a: str(a), h["ports"])), vulns_str=", ".join(h["vulns"]) if "vulns" in h else None, **h).strip())
|
||||
for d in h["data"]:
|
||||
res.append_message(print_service(d))
|
||||
|
||||
else:
|
||||
q = search_hosts(terms)
|
||||
res = Response(channel=msg.channel,
|
||||
count=" (%%s/%s results)" % q["total"])
|
||||
for r in q["matches"]:
|
||||
res.append_message(print_service(r))
|
||||
|
||||
return res
|
||||
|
|
@ -10,7 +10,7 @@ from nemubot.hooks import hook
|
|||
|
||||
nemubotversion = 3.4
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
|
||||
def help_full():
|
||||
|
|
|
|||
116
modules/smmry.py
116
modules/smmry.py
|
|
@ -1,116 +0,0 @@
|
|||
"""Summarize texts"""
|
||||
|
||||
# PYTHON STUFFS #######################################################
|
||||
|
||||
from urllib.parse import quote
|
||||
|
||||
from nemubot import context
|
||||
from nemubot.exception import IMException
|
||||
from nemubot.hooks import hook
|
||||
from nemubot.tools import web
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from nemubot.module.urlreducer import LAST_URLS
|
||||
|
||||
|
||||
# GLOBALS #############################################################
|
||||
|
||||
URL_API = "https://api.smmry.com/?SM_API_KEY=%s"
|
||||
|
||||
|
||||
# LOADING #############################################################
|
||||
|
||||
def load(context):
|
||||
if not context.config or "apikey" not in context.config:
|
||||
raise ImportError("You need a Smmry API key in order to use this "
|
||||
"module. Add it to the module configuration file:\n"
|
||||
"<module name=\"smmry\" apikey=\"XXXXXXXXXXXXXXXX\" "
|
||||
"/>\nRegister at https://smmry.com/partner")
|
||||
global URL_API
|
||||
URL_API = URL_API % context.config["apikey"]
|
||||
|
||||
|
||||
# MODULE INTERFACE ####################################################
|
||||
|
||||
@hook.command("smmry",
|
||||
help="Summarize the following words/command return",
|
||||
help_usage={
|
||||
"WORDS/CMD": ""
|
||||
},
|
||||
keywords={
|
||||
"keywords?=X": "Returns keywords instead of summary (count optional)",
|
||||
"length=7": "The number of sentences returned, default 7",
|
||||
"break": "inserts the string [BREAK] between sentences",
|
||||
"ignore_length": "returns summary regardless of quality or length",
|
||||
"quote_avoid": "sentences with quotations will be excluded",
|
||||
"question_avoid": "sentences with question will be excluded",
|
||||
"exclamation_avoid": "sentences with exclamation marks will be excluded",
|
||||
})
|
||||
def cmd_smmry(msg):
|
||||
if not len(msg.args):
|
||||
global LAST_URLS
|
||||
if msg.channel in LAST_URLS and len(LAST_URLS[msg.channel]) > 0:
|
||||
msg.args.append(LAST_URLS[msg.channel].pop())
|
||||
else:
|
||||
raise IMException("I have no more URL to sum up.")
|
||||
|
||||
URL = URL_API
|
||||
if "length" in msg.kwargs:
|
||||
if int(msg.kwargs["length"]) > 0 :
|
||||
URL += "&SM_LENGTH=" + msg.kwargs["length"]
|
||||
else:
|
||||
msg.kwargs["ignore_length"] = True
|
||||
if "break" in msg.kwargs: URL += "&SM_WITH_BREAK"
|
||||
if "ignore_length" in msg.kwargs: URL += "&SM_IGNORE_LENGTH"
|
||||
if "quote_avoid" in msg.kwargs: URL += "&SM_QUOTE_AVOID"
|
||||
if "question_avoid" in msg.kwargs: URL += "&SM_QUESTION_AVOID"
|
||||
if "exclamation_avoid" in msg.kwargs: URL += "&SM_EXCLAMATION_AVOID"
|
||||
if "keywords" in msg.kwargs and msg.kwargs["keywords"] is not None and int(msg.kwargs["keywords"]) > 0: URL += "&SM_KEYWORD_COUNT=" + msg.kwargs["keywords"]
|
||||
|
||||
res = Response(channel=msg.channel)
|
||||
|
||||
if web.isURL(" ".join(msg.args)):
|
||||
smmry = web.getJSON(URL + "&SM_URL=" + quote(" ".join(msg.args)), timeout=23)
|
||||
else:
|
||||
cnt = ""
|
||||
for r in context.subtreat(context.subparse(msg, " ".join(msg.args))):
|
||||
if isinstance(r, Response):
|
||||
for i in range(len(r.messages) - 1, -1, -1):
|
||||
if isinstance(r.messages[i], list):
|
||||
for j in range(len(r.messages[i]) - 1, -1, -1):
|
||||
cnt += r.messages[i][j] + "\n"
|
||||
elif isinstance(r.messages[i], str):
|
||||
cnt += r.messages[i] + "\n"
|
||||
else:
|
||||
cnt += str(r.messages) + "\n"
|
||||
|
||||
elif isinstance(r, Text):
|
||||
cnt += r.message + "\n"
|
||||
|
||||
else:
|
||||
cnt += str(r) + "\n"
|
||||
|
||||
smmry = web.getJSON(URL, body="sm_api_input=" + quote(cnt), timeout=23)
|
||||
|
||||
if "sm_api_error" in smmry:
|
||||
if smmry["sm_api_error"] == 0:
|
||||
title = "Internal server problem (not your fault)"
|
||||
elif smmry["sm_api_error"] == 1:
|
||||
title = "Incorrect submission variables"
|
||||
elif smmry["sm_api_error"] == 2:
|
||||
title = "Intentional restriction (low credits?)"
|
||||
elif smmry["sm_api_error"] == 3:
|
||||
title = "Summarization error"
|
||||
else:
|
||||
title = "Unknown error"
|
||||
raise IMException(title + ": " + smmry['sm_api_message'].lower())
|
||||
|
||||
if "keywords" in msg.kwargs:
|
||||
smmry["sm_api_content"] = ", ".join(smmry["sm_api_keyword_array"])
|
||||
|
||||
if "sm_api_title" in smmry and smmry["sm_api_title"] != "":
|
||||
res.append_message(smmry["sm_api_content"], title=smmry["sm_api_title"])
|
||||
else:
|
||||
res.append_message(smmry["sm_api_content"])
|
||||
|
||||
return res
|
||||
110
modules/sms.py
110
modules/sms.py
|
|
@ -16,7 +16,7 @@ from nemubot.tools.xmlparser.node import ModuleState
|
|||
|
||||
nemubotversion = 3.4
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
def load(context):
|
||||
context.data.setIndex("name", "phone")
|
||||
|
|
@ -46,89 +46,47 @@ def send_sms(frm, api_usr, api_key, content):
|
|||
|
||||
return None
|
||||
|
||||
def check_sms_dests(dests, cur_epoch):
|
||||
"""Raise exception if one of the dest is not known or has already receive a SMS recently
|
||||
"""
|
||||
for u in dests:
|
||||
if u not in context.data.index:
|
||||
raise IMException("Désolé, je sais pas comment envoyer de SMS à %s." % u)
|
||||
elif cur_epoch - float(context.data.index[u]["lastuse"]) < 42:
|
||||
raise IMException("Un peu de calme, %s a déjà reçu un SMS il n'y a pas si longtemps." % u)
|
||||
return True
|
||||
|
||||
|
||||
def send_sms_to_list(msg, frm, dests, content, cur_epoch):
|
||||
fails = list()
|
||||
for u in dests:
|
||||
context.data.index[u]["lastuse"] = cur_epoch
|
||||
test = send_sms(frm, context.data.index[u]["user"], context.data.index[u]["key"], content)
|
||||
if test is not None:
|
||||
fails.append( "%s: %s" % (u, test) )
|
||||
|
||||
if len(fails) > 0:
|
||||
return Response("quelque chose ne s'est pas bien passé durant l'envoi du SMS : " + ", ".join(fails), msg.channel, msg.frm)
|
||||
else:
|
||||
return Response("le SMS a bien été envoyé", msg.channel, msg.frm)
|
||||
|
||||
|
||||
@hook.command("sms")
|
||||
def cmd_sms(msg):
|
||||
if not len(msg.args):
|
||||
raise IMException("À qui veux-tu envoyer ce SMS ?")
|
||||
|
||||
cur_epoch = time.mktime(time.localtime())
|
||||
dests = msg.args[0].split(",")
|
||||
frm = msg.frm if msg.to_response[0] == msg.frm else msg.frm + "@" + msg.to[0]
|
||||
content = " ".join(msg.args[1:])
|
||||
# Check dests
|
||||
cur_epoch = time.mktime(time.localtime());
|
||||
for u in msg.args[0].split(","):
|
||||
if u not in context.data.index:
|
||||
raise IMException("Désolé, je sais pas comment envoyer de SMS à %s." % u)
|
||||
elif cur_epoch - float(context.data.index[u]["lastuse"]) < 42:
|
||||
raise IMException("Un peu de calme, %s a déjà reçu un SMS il n'y a pas si longtemps." % u)
|
||||
|
||||
check_sms_dests(dests, cur_epoch)
|
||||
return send_sms_to_list(msg, frm, dests, content, cur_epoch)
|
||||
|
||||
|
||||
@hook.command("smscmd")
|
||||
def cmd_smscmd(msg):
|
||||
if not len(msg.args):
|
||||
raise IMException("À qui veux-tu envoyer ce SMS ?")
|
||||
|
||||
cur_epoch = time.mktime(time.localtime())
|
||||
dests = msg.args[0].split(",")
|
||||
frm = msg.frm if msg.to_response[0] == msg.frm else msg.frm + "@" + msg.to[0]
|
||||
cmd = " ".join(msg.args[1:])
|
||||
|
||||
content = None
|
||||
for r in context.subtreat(context.subparse(msg, cmd)):
|
||||
if isinstance(r, Response):
|
||||
for m in r.messages:
|
||||
if isinstance(m, list):
|
||||
for n in m:
|
||||
content = n
|
||||
break
|
||||
if content is not None:
|
||||
break
|
||||
elif isinstance(m, str):
|
||||
content = m
|
||||
break
|
||||
|
||||
elif isinstance(r, Text):
|
||||
content = r.message
|
||||
|
||||
if content is None:
|
||||
raise IMException("Aucun SMS envoyé : le résultat de la commande n'a pas retourné de contenu.")
|
||||
|
||||
check_sms_dests(dests, cur_epoch)
|
||||
return send_sms_to_list(msg, frm, dests, content, cur_epoch)
|
||||
# Go!
|
||||
fails = list()
|
||||
for u in msg.args[0].split(","):
|
||||
context.data.index[u]["lastuse"] = cur_epoch
|
||||
if msg.to_response[0] == msg.frm:
|
||||
frm = msg.frm
|
||||
else:
|
||||
frm = msg.frm + "@" + msg.to[0]
|
||||
test = send_sms(frm, context.data.index[u]["user"], context.data.index[u]["key"], " ".join(msg.args[1:]))
|
||||
if test is not None:
|
||||
fails.append( "%s: %s" % (u, test) )
|
||||
|
||||
if len(fails) > 0:
|
||||
return Response("quelque chose ne s'est pas bien passé durant l'envoi du SMS : " + ", ".join(fails), msg.channel, msg.nick)
|
||||
else:
|
||||
return Response("le SMS a bien été envoyé", msg.channel, msg.nick)
|
||||
|
||||
apiuser_ask = re.compile(r"(utilisateur|user|numéro|numero|compte|abonne|abone|abonné|account)\s+(est|is)\s+(?P<user>[0-9]{7,})", re.IGNORECASE)
|
||||
apikey_ask = re.compile(r"(clef|key|password|mot de passe?)\s+(?:est|is)?\s+(?P<key>[a-zA-Z0-9]{10,})", re.IGNORECASE)
|
||||
|
||||
@hook.ask()
|
||||
def parseask(msg):
|
||||
if msg.message.find("Free") >= 0 and (
|
||||
msg.message.find("API") >= 0 or msg.message.find("api") >= 0) and (
|
||||
msg.message.find("SMS") >= 0 or msg.message.find("sms") >= 0):
|
||||
resuser = apiuser_ask.search(msg.message)
|
||||
reskey = apikey_ask.search(msg.message)
|
||||
if msg.text.find("Free") >= 0 and (
|
||||
msg.text.find("API") >= 0 or msg.text.find("api") >= 0) and (
|
||||
msg.text.find("SMS") >= 0 or msg.text.find("sms") >= 0):
|
||||
resuser = apiuser_ask.search(msg.text)
|
||||
reskey = apikey_ask.search(msg.text)
|
||||
if resuser is not None and reskey is not None:
|
||||
apiuser = resuser.group("user")
|
||||
apikey = reskey.group("key")
|
||||
|
|
@ -136,18 +94,18 @@ def parseask(msg):
|
|||
test = send_sms("nemubot", apiuser, apikey,
|
||||
"Vous avez enregistré vos codes d'authentification dans nemubot, félicitation !")
|
||||
if test is not None:
|
||||
return Response("je n'ai pas pu enregistrer tes identifiants : %s" % test, msg.channel, msg.frm)
|
||||
return Response("je n'ai pas pu enregistrer tes identifiants : %s" % test, msg.channel, msg.nick)
|
||||
|
||||
if msg.frm in context.data.index:
|
||||
context.data.index[msg.frm]["user"] = apiuser
|
||||
context.data.index[msg.frm]["key"] = apikey
|
||||
if msg.nick in context.data.index:
|
||||
context.data.index[msg.nick]["user"] = apiuser
|
||||
context.data.index[msg.nick]["key"] = apikey
|
||||
else:
|
||||
ms = ModuleState("phone")
|
||||
ms.setAttribute("name", msg.frm)
|
||||
ms.setAttribute("name", msg.nick)
|
||||
ms.setAttribute("user", apiuser)
|
||||
ms.setAttribute("key", apikey)
|
||||
ms.setAttribute("lastuse", 0)
|
||||
context.data.addChild(ms)
|
||||
context.save()
|
||||
return Response("ok, c'est noté. Je t'ai envoyé un SMS pour tester ;)",
|
||||
msg.channel, msg.frm)
|
||||
msg.channel, msg.nick)
|
||||
|
|
|
|||
|
|
@ -10,7 +10,7 @@ from nemubot.tools.xmlparser.node import ModuleState
|
|||
from .pyaspell import Aspell
|
||||
from .pyaspell import AspellError
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
|
||||
# LOADING #############################################################
|
||||
|
|
@ -64,15 +64,15 @@ def cmd_spell(msg):
|
|||
raise IMException("Je n'ai pas le dictionnaire `%s' :(" % lang)
|
||||
|
||||
if r == True:
|
||||
add_score(msg.frm, "correct")
|
||||
add_score(msg.nick, "correct")
|
||||
res.append_message("l'orthographe de `%s' est correcte" % word)
|
||||
|
||||
elif len(r) > 0:
|
||||
add_score(msg.frm, "bad")
|
||||
add_score(msg.nick, "bad")
|
||||
res.append_message(r, title="suggestions pour `%s'" % word)
|
||||
|
||||
else:
|
||||
add_score(msg.frm, "bad")
|
||||
add_score(msg.nick, "bad")
|
||||
res.append_message("aucune suggestion pour `%s'" % word)
|
||||
|
||||
return res
|
||||
|
|
|
|||
210
modules/suivi.py
210
modules/suivi.py
|
|
@ -2,22 +2,23 @@
|
|||
|
||||
# PYTHON STUFF ############################################
|
||||
|
||||
import json
|
||||
import urllib.request
|
||||
import urllib.parse
|
||||
from bs4 import BeautifulSoup
|
||||
import re
|
||||
|
||||
from nemubot.hooks import hook
|
||||
from nemubot.exception import IMException
|
||||
from nemubot.tools.web import getURLContent, getURLHeaders, getJSON
|
||||
from nemubot.module.more import Response
|
||||
from nemubot.tools.web import getURLContent
|
||||
from more import Response
|
||||
|
||||
|
||||
# POSTAGE SERVICE PARSERS ############################################
|
||||
|
||||
def get_tnt_info(track_id):
|
||||
values = []
|
||||
data = getURLContent('https://www.tnt.fr/public/suivi_colis/recherche/visubontransport.do?bonTransport=%s' % track_id)
|
||||
data = getURLContent('www.tnt.fr/public/suivi_colis/recherche/'
|
||||
'visubontransport.do?bonTransport=%s' % track_id)
|
||||
soup = BeautifulSoup(data)
|
||||
status_list = soup.find('div', class_='result__content')
|
||||
if not status_list:
|
||||
|
|
@ -31,23 +32,24 @@ def get_tnt_info(track_id):
|
|||
|
||||
|
||||
def get_colissimo_info(colissimo_id):
|
||||
colissimo_data = getURLContent("https://www.laposte.fr/particulier/outils/suivre-vos-envois?code=%s" % colissimo_id)
|
||||
colissimo_data = getURLContent("http://www.colissimo.fr/portail_colissimo/"
|
||||
"suivre.do?colispart=%s" % colissimo_id)
|
||||
soup = BeautifulSoup(colissimo_data)
|
||||
|
||||
dataArray = soup.find(class_='results-suivi')
|
||||
if dataArray and dataArray.table and dataArray.table.tbody and dataArray.table.tbody.tr:
|
||||
td = dataArray.table.tbody.tr.find_all('td')
|
||||
if len(td) > 2:
|
||||
date = td[0].get_text()
|
||||
libelle = re.sub(r'[\n\t\r]', '', td[1].get_text())
|
||||
site = td[2].get_text().strip()
|
||||
dataArray = soup.find(class_='dataArray')
|
||||
if dataArray and dataArray.tbody and dataArray.tbody.tr:
|
||||
date = dataArray.tbody.tr.find(headers="Date").get_text()
|
||||
libelle = re.sub(r'[\n\t\r]', '',
|
||||
dataArray.tbody.tr.find(headers="Libelle").get_text())
|
||||
site = dataArray.tbody.tr.find(headers="site").get_text().strip()
|
||||
return (date, libelle, site.strip())
|
||||
|
||||
|
||||
def get_chronopost_info(track_id):
|
||||
data = urllib.parse.urlencode({'listeNumeros': track_id})
|
||||
track_baseurl = "https://www.chronopost.fr/expedier/inputLTNumbersNoJahia.do?lang=fr_FR"
|
||||
track_data = getURLContent(track_baseurl, data.encode('utf-8'))
|
||||
track_baseurl = "http://www.chronopost.fr/expedier/" \
|
||||
"inputLTNumbersNoJahia.do?lang=fr_FR"
|
||||
track_data = urllib.request.urlopen(track_baseurl, data.encode('utf-8'))
|
||||
soup = BeautifulSoup(track_data)
|
||||
|
||||
infoClass = soup.find(class_='numeroColi2')
|
||||
|
|
@ -63,8 +65,9 @@ def get_chronopost_info(track_id):
|
|||
|
||||
def get_colisprive_info(track_id):
|
||||
data = urllib.parse.urlencode({'numColis': track_id})
|
||||
track_baseurl = "https://www.colisprive.com/moncolis/pages/detailColis.aspx"
|
||||
track_data = getURLContent(track_baseurl, data.encode('utf-8'))
|
||||
track_baseurl = "https://www.colisprive.com/moncolis/pages/" \
|
||||
"detailColis.aspx"
|
||||
track_data = urllib.request.urlopen(track_baseurl, data.encode('utf-8'))
|
||||
soup = BeautifulSoup(track_data)
|
||||
|
||||
dataArray = soup.find(class_='BandeauInfoColis')
|
||||
|
|
@ -75,117 +78,34 @@ def get_colisprive_info(track_id):
|
|||
return status
|
||||
|
||||
|
||||
def get_ups_info(track_id):
|
||||
data = json.dumps({'Locale': "en_US", 'TrackingNumber': [track_id]})
|
||||
track_baseurl = "https://www.ups.com/track/api/Track/GetStatus?loc=en_US"
|
||||
track_data = getJSON(track_baseurl, data.encode('utf-8'), header={"Content-Type": "application/json"})
|
||||
return (track_data["trackDetails"][0]["trackingNumber"],
|
||||
track_data["trackDetails"][0]["packageStatus"],
|
||||
track_data["trackDetails"][0]["shipmentProgressActivities"][0]["date"] + " " + track_data["trackDetails"][0]["shipmentProgressActivities"][0]["time"],
|
||||
track_data["trackDetails"][0]["shipmentProgressActivities"][0]["location"],
|
||||
track_data["trackDetails"][0]["shipmentProgressActivities"][0]["activityScan"])
|
||||
|
||||
|
||||
def get_laposte_info(laposte_id):
|
||||
status, laposte_headers = getURLHeaders("https://www.laposte.fr/outils/suivre-vos-envois?" + urllib.parse.urlencode({'code': laposte_id}))
|
||||
data = urllib.parse.urlencode({'id': laposte_id})
|
||||
laposte_baseurl = "http://www.part.csuivi.courrier.laposte.fr/suivi/index"
|
||||
|
||||
laposte_cookie = None
|
||||
for k,v in laposte_headers:
|
||||
if k.lower() == "set-cookie" and v.find("access_token") >= 0:
|
||||
laposte_cookie = v.split(";")[0]
|
||||
|
||||
laposte_data = getJSON("https://api.laposte.fr/ssu/v1/suivi-unifie/idship/%s?lang=fr_FR" % urllib.parse.quote(laposte_id), header={"Accept": "application/json", "Cookie": laposte_cookie})
|
||||
|
||||
shipment = laposte_data["shipment"]
|
||||
return (shipment["product"], shipment["idShip"], shipment["event"][0]["label"], shipment["event"][0]["date"])
|
||||
|
||||
|
||||
def get_postnl_info(postnl_id):
|
||||
data = urllib.parse.urlencode({'barcodes': postnl_id})
|
||||
postnl_baseurl = "http://www.postnl.post/details/"
|
||||
|
||||
postnl_data = getURLContent(postnl_baseurl, data.encode('utf-8'))
|
||||
soup = BeautifulSoup(postnl_data)
|
||||
if (soup.find(id='datatables')
|
||||
and soup.find(id='datatables').tbody
|
||||
and soup.find(id='datatables').tbody.tr):
|
||||
search_res = soup.find(id='datatables').tbody.tr
|
||||
if len(search_res.find_all('td')) >= 3:
|
||||
field = field.find_next('td')
|
||||
post_date = field.get_text()
|
||||
laposte_data = urllib.request.urlopen(laposte_baseurl,
|
||||
data.encode('utf-8'))
|
||||
soup = BeautifulSoup(laposte_data)
|
||||
search_res = soup.find(class_='resultat_rech_simple_table').tbody.tr
|
||||
if (soup.find(class_='resultat_rech_simple_table').thead
|
||||
and soup.find(class_='resultat_rech_simple_table').thead.tr
|
||||
and len(search_res.find_all('td')) > 3):
|
||||
field = search_res.find('td')
|
||||
poste_id = field.get_text()
|
||||
|
||||
field = field.find_next('td')
|
||||
post_status = field.get_text()
|
||||
poste_type = field.get_text()
|
||||
|
||||
field = field.find_next('td')
|
||||
post_destination = field.get_text()
|
||||
poste_date = field.get_text()
|
||||
|
||||
return (post_status.lower(), post_destination, post_date)
|
||||
field = field.find_next('td')
|
||||
poste_location = field.get_text()
|
||||
|
||||
field = field.find_next('td')
|
||||
poste_status = field.get_text()
|
||||
|
||||
def get_usps_info(usps_id):
|
||||
usps_parcelurl = "https://tools.usps.com/go/TrackConfirmAction_input?" + urllib.parse.urlencode({'qtc_tLabels1': usps_id})
|
||||
|
||||
usps_data = getURLContent(usps_parcelurl)
|
||||
soup = BeautifulSoup(usps_data)
|
||||
if (soup.find(id="trackingHistory_1")
|
||||
and soup.find(class_="tracking_history").find(class_="row_notification")
|
||||
and soup.find(class_="tracking_history").find(class_="row_top").find_all("td")):
|
||||
notification = soup.find(class_="tracking_history").find(class_="row_notification").text.strip()
|
||||
date = re.sub(r"\s+", " ", soup.find(class_="tracking_history").find(class_="row_top").find_all("td")[0].text.strip())
|
||||
status = soup.find(class_="tracking_history").find(class_="row_top").find_all("td")[1].text.strip()
|
||||
last_location = soup.find(class_="tracking_history").find(class_="row_top").find_all("td")[2].text.strip()
|
||||
|
||||
print(notification)
|
||||
|
||||
return (notification, date, status, last_location)
|
||||
|
||||
|
||||
def get_fedex_info(fedex_id, lang="en_US"):
|
||||
data = urllib.parse.urlencode({
|
||||
'data': json.dumps({
|
||||
"TrackPackagesRequest": {
|
||||
"appType": "WTRK",
|
||||
"appDeviceType": "DESKTOP",
|
||||
"uniqueKey": "",
|
||||
"processingParameters": {},
|
||||
"trackingInfoList": [
|
||||
{
|
||||
"trackNumberInfo": {
|
||||
"trackingNumber": str(fedex_id),
|
||||
"trackingQualifier": "",
|
||||
"trackingCarrier": ""
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
}),
|
||||
'action': "trackpackages",
|
||||
'locale': lang,
|
||||
'version': 1,
|
||||
'format': "json"
|
||||
})
|
||||
fedex_baseurl = "https://www.fedex.com/trackingCal/track"
|
||||
|
||||
fedex_data = getJSON(fedex_baseurl, data.encode('utf-8'))
|
||||
|
||||
if ("TrackPackagesResponse" in fedex_data and
|
||||
"packageList" in fedex_data["TrackPackagesResponse"] and
|
||||
len(fedex_data["TrackPackagesResponse"]["packageList"]) and
|
||||
(not fedex_data["TrackPackagesResponse"]["errorList"][0]["code"] or
|
||||
fedex_data["TrackPackagesResponse"]["errorList"][0]["code"] == '0') and
|
||||
not fedex_data["TrackPackagesResponse"]["packageList"][0]["errorList"][0]["code"]
|
||||
):
|
||||
return fedex_data["TrackPackagesResponse"]["packageList"][0]
|
||||
|
||||
|
||||
def get_dhl_info(dhl_id, lang="en"):
|
||||
dhl_parcelurl = "http://www.dhl.com/shipmentTracking?" + urllib.parse.urlencode({'AWB': dhl_id})
|
||||
|
||||
dhl_data = getJSON(dhl_parcelurl)
|
||||
|
||||
if "results" in dhl_data and dhl_data["results"]:
|
||||
return dhl_data["results"][0]
|
||||
return (poste_type.lower(), poste_id.strip(), poste_status.lower(),
|
||||
poste_location, poste_date)
|
||||
|
||||
|
||||
# TRACKING HANDLERS ###################################################
|
||||
|
|
@ -206,33 +126,11 @@ def handle_tnt(tracknum):
|
|||
def handle_laposte(tracknum):
|
||||
info = get_laposte_info(tracknum)
|
||||
if info:
|
||||
poste_type, poste_id, poste_status, poste_date = info
|
||||
return ("\x02%s\x0F : \x02%s\x0F est actuellement "
|
||||
"\x02%s\x0F (Mis à jour le \x02%s\x0F"
|
||||
")." % (poste_type, poste_id, poste_status, poste_date))
|
||||
|
||||
|
||||
def handle_postnl(tracknum):
|
||||
info = get_postnl_info(tracknum)
|
||||
if info:
|
||||
post_status, post_destination, post_date = info
|
||||
return ("PostNL \x02%s\x0F est actuellement "
|
||||
"\x02%s\x0F vers le pays \x02%s\x0F (Mis à jour le \x02%s\x0F"
|
||||
")." % (tracknum, post_status, post_destination, post_date))
|
||||
|
||||
|
||||
def handle_usps(tracknum):
|
||||
info = get_usps_info(tracknum)
|
||||
if info:
|
||||
notif, last_date, last_status, last_location = info
|
||||
return ("USPS \x02{tracknum}\x0F: {last_status} in \x02{last_location}\x0F as of {last_date}: {notif}".format(tracknum=tracknum, notif=notif, last_date=last_date, last_status=last_status.lower(), last_location=last_location))
|
||||
|
||||
|
||||
def handle_ups(tracknum):
|
||||
info = get_ups_info(tracknum)
|
||||
if info:
|
||||
tracknum, status, last_date, last_location, last_status = info
|
||||
return ("UPS \x02{tracknum}\x0F: {status}: in \x02{last_location}\x0F as of {last_date}: {last_status}".format(tracknum=tracknum, status=status, last_date=last_date, last_status=last_status.lower(), last_location=last_location))
|
||||
poste_type, poste_id, poste_status, poste_location, poste_date = info
|
||||
return ("Le courrier de type \x02%s\x0F : \x02%s\x0F est actuellement "
|
||||
"\x02%s\x0F dans la zone \x02%s\x0F (Mis à jour le \x02%s\x0F"
|
||||
")." % (poste_type, poste_id, poste_status,
|
||||
poste_location, poste_date))
|
||||
|
||||
|
||||
def handle_colissimo(tracknum):
|
||||
|
|
@ -258,34 +156,12 @@ def handle_coliprive(tracknum):
|
|||
return ("Colis Privé: \x02%s\x0F : \x02%s\x0F." % (tracknum, info))
|
||||
|
||||
|
||||
def handle_fedex(tracknum):
|
||||
info = get_fedex_info(tracknum)
|
||||
if info:
|
||||
if info["displayActDeliveryDateTime"] != "":
|
||||
return ("{trackingCarrierDesc}: \x02{statusWithDetails}\x0F: in \x02{statusLocationCity}, {statusLocationCntryCD}\x0F, delivered on: {displayActDeliveryDateTime}.".format(**info))
|
||||
elif info["statusLocationCity"] != "":
|
||||
return ("{trackingCarrierDesc}: \x02{statusWithDetails}\x0F: estimated delivery: {displayEstDeliveryDateTime}.".format(**info))
|
||||
else:
|
||||
return ("{trackingCarrierDesc}: \x02{statusWithDetails}\x0F: in \x02{statusLocationCity}, {statusLocationCntryCD}\x0F, estimated delivery: {displayEstDeliveryDateTime}.".format(**info))
|
||||
|
||||
|
||||
def handle_dhl(tracknum):
|
||||
info = get_dhl_info(tracknum)
|
||||
if info:
|
||||
return "DHL {label} {id}: \x02{description}\x0F".format(**info)
|
||||
|
||||
|
||||
TRACKING_HANDLERS = {
|
||||
'laposte': handle_laposte,
|
||||
'postnl': handle_postnl,
|
||||
'colissimo': handle_colissimo,
|
||||
'chronopost': handle_chronopost,
|
||||
'coliprive': handle_coliprive,
|
||||
'tnt': handle_tnt,
|
||||
'fedex': handle_fedex,
|
||||
'dhl': handle_dhl,
|
||||
'usps': handle_usps,
|
||||
'ups': handle_ups,
|
||||
}
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -9,7 +9,7 @@ from nemubot.exception import IMException
|
|||
from nemubot.hooks import hook
|
||||
from nemubot.tools import web
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
|
||||
# LOADING #############################################################
|
||||
|
|
@ -29,7 +29,7 @@ def load(context):
|
|||
# MODULE CORE #########################################################
|
||||
|
||||
def get_french_synos(word):
|
||||
url = "https://crisco.unicaen.fr/des/synonymes/" + quote(word)
|
||||
url = "http://www.crisco.unicaen.fr/des/synonymes/" + quote(word.encode("ISO-8859-1"))
|
||||
page = web.getURLContent(url)
|
||||
|
||||
best = list(); synos = list(); anton = list()
|
||||
|
|
@ -53,7 +53,7 @@ def get_french_synos(word):
|
|||
|
||||
|
||||
def get_english_synos(key, word):
|
||||
cnt = web.getJSON("https://words.bighugelabs.com/api/2/%s/%s/json" %
|
||||
cnt = web.getJSON("http://words.bighugelabs.com/api/2/%s/%s/json" %
|
||||
(quote(key), quote(word.encode("ISO-8859-1"))))
|
||||
|
||||
best = list(); synos = list(); anton = list()
|
||||
|
|
|
|||
|
|
@ -8,7 +8,7 @@ from nemubot.tools.web import getJSON
|
|||
|
||||
nemubotversion = 4.0
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
URL_TPBAPI = None
|
||||
|
||||
|
|
|
|||
|
|
@ -8,7 +8,7 @@ from nemubot.exception import IMException
|
|||
from nemubot.hooks import hook
|
||||
from nemubot.tools import web
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
|
||||
# GLOBALS #############################################################
|
||||
|
|
|
|||
|
|
@ -8,13 +8,13 @@ from nemubot.exception import IMException
|
|||
from nemubot.hooks import hook
|
||||
from nemubot.tools import web
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
# MODULE CORE #########################################################
|
||||
|
||||
def search(terms):
|
||||
return web.getJSON(
|
||||
"https://api.urbandictionary.com/v0/define?term=%s"
|
||||
"http://api.urbandictionary.com/v0/define?term=%s"
|
||||
% quote(' '.join(terms)))
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -21,7 +21,7 @@ def default_reducer(url, data):
|
|||
|
||||
|
||||
def ycc_reducer(url, data):
|
||||
return "https://ycc.fr/%s" % default_reducer(url, data)
|
||||
return "http://ycc.fr/%s" % default_reducer(url, data)
|
||||
|
||||
def lstu_reducer(url, data):
|
||||
json_data = json.loads(web.getURLContent(url, "lsturl=" + quote(data),
|
||||
|
|
@ -36,8 +36,8 @@ def lstu_reducer(url, data):
|
|||
# MODULE VARIABLES ####################################################
|
||||
|
||||
PROVIDERS = {
|
||||
"tinyurl": (default_reducer, "https://tinyurl.com/api-create.php?url="),
|
||||
"ycc": (ycc_reducer, "https://ycc.fr/redirection/create/"),
|
||||
"tinyurl": (default_reducer, "http://tinyurl.com/api-create.php?url="),
|
||||
"ycc": (ycc_reducer, "http://ycc.fr/redirection/create/"),
|
||||
"framalink": (lstu_reducer, "https://frama.link/a?format=json"),
|
||||
"huitre": (lstu_reducer, "https://huit.re/a?format=json"),
|
||||
"lstu": (lstu_reducer, "https://lstu.fr/a?format=json"),
|
||||
|
|
@ -60,20 +60,12 @@ def load(context):
|
|||
|
||||
# MODULE CORE #########################################################
|
||||
|
||||
def reduce_inline(txt, provider=None):
|
||||
for url in re.findall("([a-zA-Z0-9+.-]+:(?://)?(?:[^ :/]+:[0-9]+)?[^ :]+)", txt):
|
||||
txt = txt.replace(url, reduce(url, provider))
|
||||
return txt
|
||||
|
||||
|
||||
def reduce(url, provider=None):
|
||||
def reduce(url, provider=DEFAULT_PROVIDER):
|
||||
"""Ask the url shortner website to reduce given URL
|
||||
|
||||
Argument:
|
||||
url -- the URL to reduce
|
||||
"""
|
||||
if provider is None:
|
||||
provider = DEFAULT_PROVIDER
|
||||
return PROVIDERS[provider][0](PROVIDERS[provider][1], url)
|
||||
|
||||
|
||||
|
|
@ -92,22 +84,8 @@ LAST_URLS = dict()
|
|||
|
||||
@hook.message()
|
||||
def parselisten(msg):
|
||||
global LAST_URLS
|
||||
if hasattr(msg, "message") and isinstance(msg.message, str):
|
||||
urls = re.findall("([a-zA-Z0-9+.-]+:(?://)?(?:[^ :/]+:[0-9]+)?[^ :]+)",
|
||||
msg.message)
|
||||
for url in urls:
|
||||
o = urlparse(web._getNormalizedURL(url), "http")
|
||||
|
||||
# Skip short URLs
|
||||
if (o.netloc == "" or o.netloc in PROVIDERS or
|
||||
len(o.netloc) + len(o.path) < 17):
|
||||
continue
|
||||
|
||||
for recv in msg.to:
|
||||
if recv not in LAST_URLS:
|
||||
LAST_URLS[recv] = list()
|
||||
LAST_URLS[recv].append(url)
|
||||
parseresponse(msg)
|
||||
return None
|
||||
|
||||
|
||||
@hook.post()
|
||||
|
|
|
|||
|
|
@ -9,7 +9,7 @@ from nemubot.exception import IMException
|
|||
from nemubot.hooks import hook
|
||||
from nemubot.tools import web
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
|
||||
# LOADING #############################################################
|
||||
|
|
|
|||
|
|
@ -10,12 +10,12 @@ from nemubot.exception import IMException
|
|||
from nemubot.hooks import hook
|
||||
from nemubot.tools import web
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from nemubot.module import mapquest
|
||||
from more import Response
|
||||
import mapquest
|
||||
|
||||
# GLOBALS #############################################################
|
||||
|
||||
URL_API = "https://public-api.adsbexchange.com/VirtualRadar/AircraftList.json?fcallC=%s"
|
||||
URL_API = "http://public-api.adsbexchange.com/VirtualRadar/AircraftList.json?fcallC=%s"
|
||||
|
||||
SPEED_TYPES = {
|
||||
0: 'Ground speed',
|
||||
|
|
@ -80,7 +80,7 @@ def cmd_flight(msg):
|
|||
if not len(msg.args):
|
||||
raise IMException("please indicate a flight")
|
||||
|
||||
res = Response(channel=msg.channel, nick=msg.frm,
|
||||
res = Response(channel=msg.channel, nick=msg.nick,
|
||||
nomore="No more flights", count=" (%s more flights)")
|
||||
|
||||
for param in msg.args:
|
||||
|
|
|
|||
|
|
@ -1,6 +1,6 @@
|
|||
# coding=utf-8
|
||||
|
||||
"""The weather module. Powered by Dark Sky <https://darksky.net/poweredby/>"""
|
||||
"""The weather module"""
|
||||
|
||||
import datetime
|
||||
import re
|
||||
|
|
@ -11,71 +11,71 @@ from nemubot.hooks import hook
|
|||
from nemubot.tools import web
|
||||
from nemubot.tools.xmlparser.node import ModuleState
|
||||
|
||||
from nemubot.module import mapquest
|
||||
import mapquest
|
||||
|
||||
nemubotversion = 4.0
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
URL_DSAPI = "https://api.darksky.net/forecast/%s/%%s,%%s?lang=%%s&units=%%s"
|
||||
|
||||
UNITS = {
|
||||
"ca": {
|
||||
"temperature": "°C",
|
||||
"distance": "km",
|
||||
"precipIntensity": "mm/h",
|
||||
"precip": "cm",
|
||||
"speed": "km/h",
|
||||
"pressure": "hPa",
|
||||
},
|
||||
"uk2": {
|
||||
"temperature": "°C",
|
||||
"distance": "mi",
|
||||
"precipIntensity": "mm/h",
|
||||
"precip": "cm",
|
||||
"speed": "mi/h",
|
||||
"pressure": "hPa",
|
||||
},
|
||||
"us": {
|
||||
"temperature": "°F",
|
||||
"distance": "mi",
|
||||
"precipIntensity": "in/h",
|
||||
"precip": "in",
|
||||
"speed": "mi/h",
|
||||
"pressure": "mbar",
|
||||
},
|
||||
"si": {
|
||||
"temperature": "°C",
|
||||
"distance": "km",
|
||||
"precipIntensity": "mm/h",
|
||||
"precip": "cm",
|
||||
"speed": "m/s",
|
||||
"pressure": "hPa",
|
||||
},
|
||||
}
|
||||
URL_DSAPI = "https://api.forecast.io/forecast/%s/%%s,%%s"
|
||||
|
||||
def load(context):
|
||||
if not context.config or "darkskyapikey" not in context.config:
|
||||
raise ImportError("You need a Dark-Sky API key in order to use this "
|
||||
"module. Add it to the module configuration file:\n"
|
||||
"<module name=\"weather\" darkskyapikey=\"XXX\" />\n"
|
||||
"Register at https://developer.forecast.io/")
|
||||
"Register at http://developer.forecast.io/")
|
||||
context.data.setIndex("name", "city")
|
||||
global URL_DSAPI
|
||||
URL_DSAPI = URL_DSAPI % context.config["darkskyapikey"]
|
||||
|
||||
|
||||
def format_wth(wth, flags):
|
||||
units = UNITS[flags["units"] if "units" in flags and flags["units"] in UNITS else "si"]
|
||||
return ("{temperature} {units[temperature]} {summary}; precipitation ({precipProbability:.0%} chance) intensity: {precipIntensity} {units[precipIntensity]}; relative humidity: {humidity:.0%}; wind speed: {windSpeed} {units[speed]} {windBearing}°; cloud coverage: {cloudCover:.0%}; pressure: {pressure} {units[pressure]}; ozone: {ozone} DU"
|
||||
.format(units=units, **wth)
|
||||
)
|
||||
def help_full ():
|
||||
return "!weather /city/: Display the current weather in /city/."
|
||||
|
||||
|
||||
def format_forecast_daily(wth, flags):
|
||||
units = UNITS[flags["units"] if "units" in flags and flags["units"] in UNITS else "si"]
|
||||
print(units)
|
||||
return ("{summary}; between {temperatureMin}-{temperatureMax} {units[temperature]}; precipitation ({precipProbability:.0%} chance) intensity: maximum {precipIntensity} {units[precipIntensity]}; relative humidity: {humidity:.0%}; wind speed: {windSpeed} {units[speed]} {windBearing}°; cloud coverage: {cloudCover:.0%}; pressure: {pressure} {units[pressure]}; ozone: {ozone} DU".format(units=units, **wth))
|
||||
def fahrenheit2celsius(temp):
|
||||
return int((temp - 32) * 50/9)/10
|
||||
|
||||
|
||||
def mph2kmph(speed):
|
||||
return int(speed * 160.9344)/100
|
||||
|
||||
|
||||
def inh2mmh(size):
|
||||
return int(size * 254)/10
|
||||
|
||||
|
||||
def format_wth(wth):
|
||||
return ("%s °C %s; precipitation (%s %% chance) intensity: %s mm/h; relative humidity: %s %%; wind speed: %s km/h %s°; cloud coverage: %s %%; pressure: %s hPa; ozone: %s DU" %
|
||||
(
|
||||
fahrenheit2celsius(wth["temperature"]),
|
||||
wth["summary"],
|
||||
int(wth["precipProbability"] * 100),
|
||||
inh2mmh(wth["precipIntensity"]),
|
||||
int(wth["humidity"] * 100),
|
||||
mph2kmph(wth["windSpeed"]),
|
||||
wth["windBearing"],
|
||||
int(wth["cloudCover"] * 100),
|
||||
int(wth["pressure"]),
|
||||
int(wth["ozone"])
|
||||
))
|
||||
|
||||
|
||||
def format_forecast_daily(wth):
|
||||
return ("%s; between %s-%s °C; precipitation (%s %% chance) intensity: maximum %s mm/h; relative humidity: %s %%; wind speed: %s km/h %s°; cloud coverage: %s %%; pressure: %s hPa; ozone: %s DU" %
|
||||
(
|
||||
wth["summary"],
|
||||
fahrenheit2celsius(wth["temperatureMin"]), fahrenheit2celsius(wth["temperatureMax"]),
|
||||
int(wth["precipProbability"] * 100),
|
||||
inh2mmh(wth["precipIntensityMax"]),
|
||||
int(wth["humidity"] * 100),
|
||||
mph2kmph(wth["windSpeed"]),
|
||||
wth["windBearing"],
|
||||
int(wth["cloudCover"] * 100),
|
||||
int(wth["pressure"]),
|
||||
int(wth["ozone"])
|
||||
))
|
||||
|
||||
|
||||
def format_timestamp(timestamp, tzname, tzoffset, format="%c"):
|
||||
|
|
@ -126,8 +126,8 @@ def treat_coord(msg):
|
|||
raise IMException("indique-moi un nom de ville ou des coordonnées.")
|
||||
|
||||
|
||||
def get_json_weather(coords, lang="en", units="ca"):
|
||||
wth = web.getJSON(URL_DSAPI % (float(coords[0]), float(coords[1]), lang, units))
|
||||
def get_json_weather(coords):
|
||||
wth = web.getJSON(URL_DSAPI % (float(coords[0]), float(coords[1])))
|
||||
|
||||
# First read flags
|
||||
if wth is None or "darksky-unavailable" in wth["flags"]:
|
||||
|
|
@ -149,53 +149,31 @@ def cmd_coordinates(msg):
|
|||
return Response("Les coordonnées de %s sont %s,%s" % (msg.args[0], coords["lat"], coords["long"]), channel=msg.channel)
|
||||
|
||||
|
||||
@hook.command("alert",
|
||||
keywords={
|
||||
"lang=LANG": "change the output language of weather sumarry; default: en",
|
||||
"units=UNITS": "return weather conditions in the requested units; default: ca",
|
||||
})
|
||||
@hook.command("alert")
|
||||
def cmd_alert(msg):
|
||||
loc, coords, specific = treat_coord(msg)
|
||||
wth = get_json_weather(coords,
|
||||
lang=msg.kwargs["lang"] if "lang" in msg.kwargs else "en",
|
||||
units=msg.kwargs["units"] if "units" in msg.kwargs else "ca")
|
||||
wth = get_json_weather(coords)
|
||||
|
||||
res = Response(channel=msg.channel, nomore="No more weather alert", count=" (%d more alerts)")
|
||||
|
||||
if "alerts" in wth:
|
||||
for alert in wth["alerts"]:
|
||||
if "expires" in alert:
|
||||
res.append_message("\x03\x02%s\x03\x02 (see %s expire on %s): %s" % (alert["title"], alert["uri"], format_timestamp(int(alert["expires"]), wth["timezone"], wth["offset"]), alert["description"].replace("\n", " ")))
|
||||
else:
|
||||
res.append_message("\x03\x02%s\x03\x02 (see %s): %s" % (alert["title"], alert["uri"], alert["description"].replace("\n", " ")))
|
||||
|
||||
return res
|
||||
|
||||
|
||||
@hook.command("météo",
|
||||
help="Display current weather and previsions",
|
||||
help_usage={
|
||||
"CITY": "Display the current weather and previsions in CITY",
|
||||
},
|
||||
keywords={
|
||||
"lang=LANG": "change the output language of weather sumarry; default: en",
|
||||
"units=UNITS": "return weather conditions in the requested units; default: ca",
|
||||
})
|
||||
@hook.command("météo")
|
||||
def cmd_weather(msg):
|
||||
loc, coords, specific = treat_coord(msg)
|
||||
wth = get_json_weather(coords,
|
||||
lang=msg.kwargs["lang"] if "lang" in msg.kwargs else "en",
|
||||
units=msg.kwargs["units"] if "units" in msg.kwargs else "ca")
|
||||
wth = get_json_weather(coords)
|
||||
|
||||
res = Response(channel=msg.channel, nomore="No more weather information")
|
||||
|
||||
if "alerts" in wth:
|
||||
alert_msgs = list()
|
||||
for alert in wth["alerts"]:
|
||||
if "expires" in alert:
|
||||
alert_msgs.append("\x03\x02%s\x03\x02 expire on %s" % (alert["title"], format_timestamp(int(alert["expires"]), wth["timezone"], wth["offset"])))
|
||||
else:
|
||||
alert_msgs.append("\x03\x02%s\x03\x02" % (alert["title"]))
|
||||
res.append_message("\x03\x16\x03\x02/!\\\x03\x02 Alert%s:\x03\x16 " % ("s" if len(alert_msgs) > 1 else "") + ", ".join(alert_msgs))
|
||||
|
||||
if specific is not None:
|
||||
|
|
@ -207,17 +185,17 @@ def cmd_weather(msg):
|
|||
|
||||
if gr.group(2).lower() == "h" and gr1 < len(wth["hourly"]["data"]):
|
||||
hour = wth["hourly"]["data"][gr1]
|
||||
res.append_message("\x03\x02At %sh:\x03\x02 %s" % (format_timestamp(int(hour["time"]), wth["timezone"], wth["offset"], '%H'), format_wth(hour, wth["flags"])))
|
||||
res.append_message("\x03\x02At %sh:\x03\x02 %s" % (format_timestamp(int(hour["time"]), wth["timezone"], wth["offset"], '%H'), format_wth(hour)))
|
||||
|
||||
elif gr.group(2).lower() == "d" and gr1 < len(wth["daily"]["data"]):
|
||||
day = wth["daily"]["data"][gr1]
|
||||
res.append_message("\x03\x02On %s:\x03\x02 %s" % (format_timestamp(int(day["time"]), wth["timezone"], wth["offset"], '%A'), format_forecast_daily(day, wth["flags"])))
|
||||
res.append_message("\x03\x02On %s:\x03\x02 %s" % (format_timestamp(int(day["time"]), wth["timezone"], wth["offset"], '%A'), format_forecast_daily(day)))
|
||||
|
||||
else:
|
||||
res.append_message("I don't understand %s or information is not available" % specific)
|
||||
|
||||
else:
|
||||
res.append_message("\x03\x02Currently:\x03\x02 " + format_wth(wth["currently"], wth["flags"]))
|
||||
res.append_message("\x03\x02Currently:\x03\x02 " + format_wth(wth["currently"]))
|
||||
|
||||
nextres = "\x03\x02Today:\x03\x02 %s " % wth["daily"]["data"][0]["summary"]
|
||||
if "minutely" in wth:
|
||||
|
|
@ -227,11 +205,11 @@ def cmd_weather(msg):
|
|||
|
||||
for hour in wth["hourly"]["data"][1:4]:
|
||||
res.append_message("\x03\x02At %sh:\x03\x02 %s" % (format_timestamp(int(hour["time"]), wth["timezone"], wth["offset"], '%H'),
|
||||
format_wth(hour, wth["flags"])))
|
||||
format_wth(hour)))
|
||||
|
||||
for day in wth["daily"]["data"][1:]:
|
||||
res.append_message("\x03\x02On %s:\x03\x02 %s" % (format_timestamp(int(day["time"]), wth["timezone"], wth["offset"], '%A'),
|
||||
format_forecast_daily(day, wth["flags"])))
|
||||
format_forecast_daily(day)))
|
||||
|
||||
return res
|
||||
|
||||
|
|
@ -241,7 +219,7 @@ gps_ask = re.compile(r"^\s*(?P<city>.*\w)\s*(?:(?:se|est)\s+(?:trouve|situ[ée]*
|
|||
|
||||
@hook.ask()
|
||||
def parseask(msg):
|
||||
res = gps_ask.match(msg.message)
|
||||
res = gps_ask.match(msg.text)
|
||||
if res is not None:
|
||||
city_name = res.group("city").lower()
|
||||
gps_lat = res.group("lat").replace(",", ".")
|
||||
|
|
@ -258,4 +236,4 @@ def parseask(msg):
|
|||
context.data.addChild(ms)
|
||||
context.save()
|
||||
return Response("ok, j'ai bien noté les coordonnées de %s" % res.group("city"),
|
||||
msg.channel, msg.frm)
|
||||
msg.channel, msg.nick)
|
||||
|
|
|
|||
|
|
@ -1,6 +1,5 @@
|
|||
# coding=utf-8
|
||||
|
||||
import json
|
||||
import re
|
||||
|
||||
from nemubot import context
|
||||
|
|
@ -10,70 +9,47 @@ from nemubot.tools.xmlparser.node import ModuleState
|
|||
|
||||
nemubotversion = 3.4
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from nemubot.module.networking.page import headers
|
||||
from more import Response
|
||||
from networking.page import headers
|
||||
|
||||
PASSWD_FILE = None
|
||||
# You can get one with: curl -b "sessionid=YOURSESSIONID" 'https://accounts.cri.epita.net/api/users/?limit=10000' > users.json
|
||||
APIEXTRACT_FILE = None
|
||||
|
||||
def load(context):
|
||||
global PASSWD_FILE
|
||||
if not context.config or "passwd" not in context.config:
|
||||
print("No passwd file given")
|
||||
else:
|
||||
PASSWD_FILE = context.config["passwd"]
|
||||
print("passwd file loaded:", PASSWD_FILE)
|
||||
|
||||
global APIEXTRACT_FILE
|
||||
if not context.config or "apiextract" not in context.config:
|
||||
print("No passwd file given")
|
||||
else:
|
||||
APIEXTRACT_FILE = context.config["apiextract"]
|
||||
print("JSON users file loaded:", APIEXTRACT_FILE)
|
||||
|
||||
if PASSWD_FILE is None and APIEXTRACT_FILE is None:
|
||||
return None
|
||||
PASSWD_FILE = context.config["passwd"]
|
||||
|
||||
if not context.data.hasNode("aliases"):
|
||||
context.data.addChild(ModuleState("aliases"))
|
||||
context.data.getNode("aliases").setIndex("from", "alias")
|
||||
|
||||
if not context.data.hasNode("pics"):
|
||||
context.data.addChild(ModuleState("pics"))
|
||||
context.data.getNode("pics").setIndex("login", "pict")
|
||||
|
||||
import nemubot.hooks
|
||||
context.add_hook(nemubot.hooks.Command(cmd_whois, "whois", keywords={"lookup": "Perform a lookup of the begining of the login instead of an exact search."}),
|
||||
context.add_hook(nemubot.hooks.Command(cmd_whois, "whois"),
|
||||
"in","Command")
|
||||
|
||||
class Login:
|
||||
|
||||
def __init__(self, line=None, login=None, uidNumber=None, firstname=None, lastname=None, promo=None, **kwargs):
|
||||
if line is not None:
|
||||
def __init__(self, line):
|
||||
s = line.split(":")
|
||||
self.login = s[0]
|
||||
self.uid = s[2]
|
||||
self.gid = s[3]
|
||||
self.cn = s[4]
|
||||
self.home = s[5]
|
||||
else:
|
||||
self.login = login
|
||||
self.uid = uidNumber
|
||||
self.promo = promo
|
||||
self.cn = firstname + " " + lastname
|
||||
try:
|
||||
self.gid = "epita" + str(int(promo))
|
||||
except:
|
||||
self.gid = promo
|
||||
|
||||
def get_promo(self):
|
||||
if hasattr(self, "promo"):
|
||||
return self.promo
|
||||
if hasattr(self, "home"):
|
||||
try:
|
||||
return self.home.split("/")[2].replace("_", " ")
|
||||
except:
|
||||
return self.gid
|
||||
|
||||
def get_photo(self):
|
||||
for url in [ "https://photos.cri.epita.fr/%s", "https://intra-bocal.epitech.eu/trombi/%s.jpg" ]:
|
||||
if self.login in context.data.getNode("pics").index:
|
||||
return context.data.getNode("pics").index[self.login]["url"]
|
||||
for url in [ "https://static.acu.epita.fr/photos/%s", "https://static.acu.epita.fr/photos/%s/%%s" % self.gid, "https://intra-bocal.epitech.eu/trombi/%s.jpg", "http://whois.23.tf/p/%s/%%s.jpg" % self.gid ]:
|
||||
url = url % self.login
|
||||
try:
|
||||
_, status, _, _ = headers(url)
|
||||
|
|
@ -84,53 +60,38 @@ class Login:
|
|||
return None
|
||||
|
||||
|
||||
def login_lookup(login, search=False):
|
||||
def found_login(login):
|
||||
if login in context.data.getNode("aliases").index:
|
||||
login = context.data.getNode("aliases").index[login]["to"]
|
||||
|
||||
if APIEXTRACT_FILE:
|
||||
with open(APIEXTRACT_FILE, encoding="utf-8") as f:
|
||||
api = json.load(f)
|
||||
for l in api["results"]:
|
||||
if (not search and l["login"] == login) or (search and (("login" in l and l["login"].find(login) != -1) or ("cn" in l and l["cn"].find(login) != -1) or ("uid" in l and str(l["uid"]) == login))):
|
||||
yield Login(**l)
|
||||
|
||||
login_ = login + (":" if not search else "")
|
||||
login_ = login + ":"
|
||||
lsize = len(login_)
|
||||
|
||||
if PASSWD_FILE:
|
||||
with open(PASSWD_FILE, encoding="iso-8859-15") as f:
|
||||
for l in f.readlines():
|
||||
if l[:lsize] == login_:
|
||||
yield Login(l.strip())
|
||||
return Login(l.strip())
|
||||
return None
|
||||
|
||||
def cmd_whois(msg):
|
||||
if len(msg.args) < 1:
|
||||
raise IMException("Provide a name")
|
||||
|
||||
def format_response(t):
|
||||
srch, l = t
|
||||
if type(l) is Login:
|
||||
pic = l.get_photo()
|
||||
return "%s is %s (%s %s): %s%s" % (srch, l.cn.capitalize(), l.login, l.uid, l.get_promo(), " and looks like %s" % pic if pic is not None else "")
|
||||
else:
|
||||
return l % srch
|
||||
|
||||
res = Response(channel=msg.channel, count=" (%d more logins)", line_treat=format_response)
|
||||
res = Response(channel=msg.channel, count=" (%d more logins)")
|
||||
for srch in msg.args:
|
||||
found = False
|
||||
for l in login_lookup(srch, "lookup" in msg.kwargs):
|
||||
found = True
|
||||
res.append_message((srch, l))
|
||||
if not found:
|
||||
res.append_message((srch, "Unknown %s :("))
|
||||
l = found_login(srch)
|
||||
if l is not None:
|
||||
pic = l.get_photo()
|
||||
res.append_message("%s is %s (%s %s): %s%s" % (srch, l.cn.capitalize(), l.login, l.uid, l.get_promo(), " and looks like %s" % pic if pic is not None else ""))
|
||||
else:
|
||||
res.append_message("Unknown %s :(" % srch)
|
||||
return res
|
||||
|
||||
@hook.command("nicks")
|
||||
def cmd_nicks(msg):
|
||||
if len(msg.args) < 1:
|
||||
raise IMException("Provide a login")
|
||||
nick = login_lookup(msg.args[0])
|
||||
nick = found_login(msg.args[0])
|
||||
if nick is None:
|
||||
nick = msg.args[0]
|
||||
else:
|
||||
|
|
@ -147,12 +108,12 @@ def cmd_nicks(msg):
|
|||
|
||||
@hook.ask()
|
||||
def parseask(msg):
|
||||
res = re.match(r"^(\S+)\s*('s|suis|est|is|was|were)\s+([a-zA-Z0-9_-]{3,8})$", msg.message, re.I)
|
||||
res = re.match(r"^(\S+)\s*('s|suis|est|is|was|were)\s+([a-zA-Z0-9_-]{3,8})$", msg.text, re.I)
|
||||
if res is not None:
|
||||
nick = res.group(1)
|
||||
login = res.group(3)
|
||||
if nick == "my" or nick == "I" or nick == "i" or nick == "je" or nick == "mon" or nick == "ma":
|
||||
nick = msg.frm
|
||||
nick = msg.nick
|
||||
if nick in context.data.getNode("aliases").index:
|
||||
context.data.getNode("aliases").index[nick]["to"] = login
|
||||
else:
|
||||
|
|
@ -164,4 +125,4 @@ def parseask(msg):
|
|||
return Response("ok, c'est noté, %s est %s"
|
||||
% (nick, login),
|
||||
channel=msg.channel,
|
||||
nick=msg.frm)
|
||||
nick=msg.nick)
|
||||
|
|
|
|||
|
|
@ -10,12 +10,12 @@ from nemubot.exception import IMException
|
|||
from nemubot.hooks import hook
|
||||
from nemubot.tools import web
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
|
||||
# LOADING #############################################################
|
||||
|
||||
URL_API = "https://api.wolframalpha.com/v2/query?input=%%s&format=plaintext&appid=%s"
|
||||
URL_API = "http://api.wolframalpha.com/v2/query?input=%%s&appid=%s"
|
||||
|
||||
def load(context):
|
||||
global URL_API
|
||||
|
|
@ -24,7 +24,7 @@ def load(context):
|
|||
"this module. Add it to the module configuration: "
|
||||
"\n<module name=\"wolframalpha\" "
|
||||
"apikey=\"XXXXXX-XXXXXXXXXX\" />\n"
|
||||
"Register at https://products.wolframalpha.com/api/")
|
||||
"Register at http://products.wolframalpha.com/api/")
|
||||
URL_API = URL_API % quote(context.config["apikey"]).replace("%", "%%")
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -1,28 +1,27 @@
|
|||
# coding=utf-8
|
||||
|
||||
"""The 2014,2018 football worldcup module"""
|
||||
"""The 2014 football worldcup module"""
|
||||
|
||||
from datetime import datetime, timezone
|
||||
from functools import partial
|
||||
import json
|
||||
import re
|
||||
from urllib.parse import quote
|
||||
from urllib.request import urlopen
|
||||
|
||||
from nemubot import context
|
||||
from nemubot.event import ModuleEvent
|
||||
from nemubot.exception import IMException
|
||||
from nemubot.hooks import hook
|
||||
from nemubot.tools.xmlparser.node import ModuleState
|
||||
|
||||
nemubotversion = 3.4
|
||||
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
API_URL="http://worldcup.sfg.io/%s"
|
||||
|
||||
def load(context):
|
||||
context.add_event(ModuleEvent(func=partial(lambda url: urlopen(url, timeout=10).read().decode(), API_URL % "matches/current?by_date=DESC"), call=current_match_new_action, interval=30))
|
||||
from nemubot.event import ModuleEvent
|
||||
context.add_event(ModuleEvent(func=lambda url: urlopen(url, timeout=10).read().decode(), func_data=API_URL % "matches/current?by_date=DESC", call=current_match_new_action, interval=30))
|
||||
|
||||
|
||||
def help_full ():
|
||||
|
|
@ -33,7 +32,7 @@ def start_watch(msg):
|
|||
w = ModuleState("watch")
|
||||
w["server"] = msg.server
|
||||
w["channel"] = msg.channel
|
||||
w["proprio"] = msg.frm
|
||||
w["proprio"] = msg.nick
|
||||
w["start"] = datetime.now(timezone.utc)
|
||||
context.data.addChild(w)
|
||||
context.save()
|
||||
|
|
@ -66,10 +65,10 @@ def cmd_watch(msg):
|
|||
context.save()
|
||||
raise IMException("This channel will not anymore receives world cup events.")
|
||||
|
||||
def current_match_new_action(matches):
|
||||
def cmp(om, nm):
|
||||
return len(nm) and (len(om) == 0 or len(nm[0]["home_team_events"]) != len(om[0]["home_team_events"]) or len(nm[0]["away_team_events"]) != len(om[0]["away_team_events"]))
|
||||
context.add_event(ModuleEvent(func=partial(lambda url: json.loads(urlopen(url).read().decode()), API_URL % "matches/current?by_date=DESC"), cmp=partial(cmp, matches), call=current_match_new_action, interval=30))
|
||||
def current_match_new_action(match_str, osef):
|
||||
context.add_event(ModuleEvent(func=lambda url: urlopen(url).read().decode(), func_data=API_URL % "matches/current?by_date=DESC", call=current_match_new_action, interval=30))
|
||||
|
||||
matches = json.loads(match_str)
|
||||
|
||||
for match in matches:
|
||||
if is_valid(match):
|
||||
|
|
@ -121,19 +120,20 @@ def detail_event(evt):
|
|||
return evt + " par"
|
||||
|
||||
def txt_event(e):
|
||||
return "%s minute : %s %s (%s)" % (e["time"], detail_event(e["type_of_event"]), e["player"], e["team"]["code"])
|
||||
return "%se minutes : %s %s (%s)" % (e["time"], detail_event(e["type_of_event"]), e["player"], e["team"]["code"])
|
||||
|
||||
def prettify(match):
|
||||
matchdate = datetime.strptime(match["datetime"].replace(':', ''), "%Y-%m-%dT%H%M%SZ").replace(tzinfo=timezone.utc)
|
||||
matchdate_local = datetime.strptime(match["datetime"].replace(':', ''), "%Y-%m-%dT%H%M%S.%f%z")
|
||||
matchdate = matchdate_local - (matchdate_local.utcoffset() - datetime.timedelta(hours=2))
|
||||
if match["status"] == "future":
|
||||
return ["Match à venir (%s) le %s : %s vs. %s" % (match["fifa_id"], matchdate.strftime("%A %d à %H:%M"), match["home_team"]["country"], match["away_team"]["country"])]
|
||||
return ["Match à venir (%s) le %s : %s vs. %s" % (match["match_number"], matchdate.strftime("%A %d à %H:%M"), match["home_team"]["country"], match["away_team"]["country"])]
|
||||
else:
|
||||
msgs = list()
|
||||
msg = ""
|
||||
if match["status"] == "completed":
|
||||
msg += "Match (%s) du %s terminé : " % (match["fifa_id"], matchdate.strftime("%A %d à %H:%M"))
|
||||
msg += "Match (%s) du %s terminé : " % (match["match_number"], matchdate.strftime("%A %d à %H:%M"))
|
||||
else:
|
||||
msg += "Match en cours (%s) depuis %d minutes : " % (match["fifa_id"], (datetime.now(tz=timezone.utc) - matchdate).total_seconds() / 60)
|
||||
msg += "Match en cours (%s) depuis %d minutes : " % (match["match_number"], (datetime.now(matchdate.tzinfo) - matchdate_local).total_seconds() / 60)
|
||||
|
||||
msg += "%s %d - %d %s" % (match["home_team"]["country"], match["home_team"]["goals"], match["away_team"]["goals"], match["away_team"]["country"])
|
||||
|
||||
|
|
@ -163,7 +163,7 @@ def is_valid(match):
|
|||
def get_match(url, matchid):
|
||||
allm = get_matches(url)
|
||||
for m in allm:
|
||||
if int(m["fifa_id"]) == matchid:
|
||||
if int(m["match_number"]) == matchid:
|
||||
return [ m ]
|
||||
|
||||
def get_matches(url):
|
||||
|
|
@ -192,7 +192,7 @@ def cmd_worldcup(msg):
|
|||
elif len(msg.args[0]) == 3:
|
||||
url = "matches/country?fifa_code=%s&by_date=DESC" % msg.args[0]
|
||||
elif is_int(msg.args[0]):
|
||||
url = int(msg.args[0])
|
||||
url = int(msg.arg[0])
|
||||
else:
|
||||
raise IMException("unrecognized request; choose between 'today', 'tomorrow', a FIFA country code or a match identifier")
|
||||
|
||||
|
|
|
|||
|
|
@ -4,7 +4,7 @@ import re, json, subprocess
|
|||
from nemubot.exception import IMException
|
||||
from nemubot.hooks import hook
|
||||
from nemubot.tools.web import _getNormalizedURL, getURLContent
|
||||
from nemubot.module.more import Response
|
||||
from more import Response
|
||||
|
||||
"""Get information of youtube videos"""
|
||||
|
||||
|
|
@ -82,7 +82,7 @@ def parselisten(msg):
|
|||
@hook.post()
|
||||
def parseresponse(msg):
|
||||
global LAST_URLS
|
||||
if hasattr(msg, "text") and msg.text and type(msg.text) == str:
|
||||
if hasattr(msg, "text") and msg.text:
|
||||
urls = re.findall("([a-zA-Z0-9+.-]+:(?://)?[^ :]+)", msg.text)
|
||||
for url in urls:
|
||||
o = urlparse(_getNormalizedURL(url))
|
||||
|
|
|
|||
|
|
@ -17,9 +17,9 @@
|
|||
__version__ = '4.0.dev3'
|
||||
__author__ = 'nemunaire'
|
||||
|
||||
from nemubot.modulecontext import _ModuleContext
|
||||
from nemubot.modulecontext import ModuleContext
|
||||
|
||||
context = _ModuleContext()
|
||||
context = ModuleContext(None, None, None)
|
||||
|
||||
|
||||
def requires_version(min=None, max=None):
|
||||
|
|
@ -39,15 +39,11 @@ def requires_version(min=None, max=None):
|
|||
"but this is nemubot v%s." % (str(max), __version__))
|
||||
|
||||
|
||||
def attach(pidfile, socketfile):
|
||||
def attach(pid, socketfile):
|
||||
import socket
|
||||
import sys
|
||||
|
||||
# Read PID from pidfile
|
||||
with open(pidfile, "r") as f:
|
||||
pid = int(f.readline())
|
||||
|
||||
print("nemubot is launched with PID %d. Attaching to Unix socket at: %s" % (pid, socketfile))
|
||||
print("nemubot is already launched with PID %d. Attaching to Unix socket at: %s" % (pid, socketfile))
|
||||
|
||||
sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
||||
try:
|
||||
|
|
@ -57,20 +53,13 @@ def attach(pidfile, socketfile):
|
|||
sys.stderr.write("\n")
|
||||
return 1
|
||||
|
||||
import select
|
||||
mypoll = select.poll()
|
||||
|
||||
mypoll.register(sys.stdin.fileno(), select.POLLIN | select.POLLPRI)
|
||||
mypoll.register(sock.fileno(), select.POLLIN | select.POLLPRI)
|
||||
from select import select
|
||||
try:
|
||||
print("Connection established.")
|
||||
while True:
|
||||
for fd, flag in mypoll.poll():
|
||||
if flag & (select.POLLERR | select.POLLHUP | select.POLLNVAL):
|
||||
sock.close()
|
||||
print("Connection closed.")
|
||||
return 1
|
||||
rl, wl, xl = select([sys.stdin, sock], [], [])
|
||||
|
||||
if fd == sys.stdin.fileno():
|
||||
if sys.stdin in rl:
|
||||
line = sys.stdin.readline().strip()
|
||||
if line == "exit" or line == "quit":
|
||||
return 0
|
||||
|
|
@ -98,9 +87,8 @@ def attach(pidfile, socketfile):
|
|||
else:
|
||||
sock.send(line.encode() + b'\r\n')
|
||||
|
||||
if fd == sock.fileno():
|
||||
if sock in rl:
|
||||
sys.stdout.write(sock.recv(2048).decode())
|
||||
|
||||
except KeyboardInterrupt:
|
||||
pass
|
||||
except:
|
||||
|
|
@ -110,7 +98,7 @@ def attach(pidfile, socketfile):
|
|||
return 0
|
||||
|
||||
|
||||
def daemonize(socketfile=None):
|
||||
def daemonize():
|
||||
"""Detach the running process to run as a daemon
|
||||
"""
|
||||
|
||||
|
|
|
|||
|
|
@ -1,5 +1,5 @@
|
|||
# Nemubot is a smart and modulable IM bot.
|
||||
# Copyright (C) 2012-2017 Mercier Pierre-Olivier
|
||||
# Copyright (C) 2012-2016 Mercier Pierre-Olivier
|
||||
#
|
||||
# This program is free software: you can redistribute it and/or modify
|
||||
# it under the terms of the GNU Affero General Public License as published by
|
||||
|
|
@ -15,6 +15,7 @@
|
|||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
def main():
|
||||
import functools
|
||||
import os
|
||||
import signal
|
||||
import sys
|
||||
|
|
@ -37,9 +38,6 @@ def main():
|
|||
default=["./modules/"],
|
||||
help="directory to use as modules store")
|
||||
|
||||
parser.add_argument("-A", "--no-attach", action="store_true",
|
||||
help="don't attach after fork")
|
||||
|
||||
parser.add_argument("-d", "--debug", action="store_true",
|
||||
help="don't deamonize, keep in foreground")
|
||||
|
||||
|
|
@ -71,25 +69,33 @@ def main():
|
|||
|
||||
# Resolve relatives paths
|
||||
args.data_path = os.path.abspath(os.path.expanduser(args.data_path))
|
||||
args.pidfile = os.path.abspath(os.path.expanduser(args.pidfile)) if args.pidfile is not None and args.pidfile != "" else None
|
||||
args.socketfile = os.path.abspath(os.path.expanduser(args.socketfile)) if args.socketfile is not None and args.socketfile != "" else None
|
||||
args.pidfile = os.path.abspath(os.path.expanduser(args.pidfile))
|
||||
args.socketfile = os.path.abspath(os.path.expanduser(args.socketfile))
|
||||
args.logfile = os.path.abspath(os.path.expanduser(args.logfile))
|
||||
args.files = [x for x in map(os.path.abspath, args.files)]
|
||||
args.modules_path = [x for x in map(os.path.abspath, args.modules_path)]
|
||||
|
||||
# Prepare the attached client, before setting other stuff
|
||||
if not args.debug and not args.no_attach and args.socketfile is not None and args.pidfile is not None:
|
||||
# Check if an instance is already launched
|
||||
if args.pidfile is not None and os.path.isfile(args.pidfile):
|
||||
with open(args.pidfile, "r") as f:
|
||||
pid = int(f.readline())
|
||||
try:
|
||||
pid = os.fork()
|
||||
if pid > 0:
|
||||
import time
|
||||
os.waitpid(pid, 0)
|
||||
time.sleep(1)
|
||||
os.kill(pid, 0)
|
||||
except OSError:
|
||||
pass
|
||||
else:
|
||||
from nemubot import attach
|
||||
sys.exit(attach(args.pidfile, args.socketfile))
|
||||
except OSError as err:
|
||||
sys.stderr.write("Unable to fork: %s\n" % err)
|
||||
sys.exit(1)
|
||||
sys.exit(attach(pid, args.socketfile))
|
||||
|
||||
# Daemonize
|
||||
if not args.debug:
|
||||
from nemubot import daemonize
|
||||
daemonize()
|
||||
|
||||
# Store PID to pidfile
|
||||
if args.pidfile is not None:
|
||||
with open(args.pidfile, "w+") as f:
|
||||
f.write(str(os.getpid()))
|
||||
|
||||
# Setup logging interface
|
||||
import logging
|
||||
|
|
@ -110,18 +116,6 @@ def main():
|
|||
fh.setFormatter(formatter)
|
||||
logger.addHandler(fh)
|
||||
|
||||
# Check if an instance is already launched
|
||||
if args.pidfile is not None and os.path.isfile(args.pidfile):
|
||||
with open(args.pidfile, "r") as f:
|
||||
pid = int(f.readline())
|
||||
try:
|
||||
os.kill(pid, 0)
|
||||
except OSError:
|
||||
pass
|
||||
else:
|
||||
from nemubot import attach
|
||||
sys.exit(attach(args.pidfile, args.socketfile))
|
||||
|
||||
# Add modules dir paths
|
||||
modules_paths = list()
|
||||
for path in args.modules_path:
|
||||
|
|
@ -132,92 +126,59 @@ def main():
|
|||
|
||||
# Create bot context
|
||||
from nemubot import datastore
|
||||
from nemubot.bot import Bot
|
||||
context = Bot(modules_paths=modules_paths,
|
||||
data_store=datastore.XML(args.data_path),
|
||||
debug=args.verbose > 0)
|
||||
from nemubot.bot import Bot#, sync_act
|
||||
context = Bot()
|
||||
|
||||
if args.no_connect:
|
||||
context.noautoconnect = True
|
||||
|
||||
# Register the hook for futur import
|
||||
from nemubot.importer import ModuleFinder
|
||||
module_finder = ModuleFinder(context.modules_paths, context.add_module)
|
||||
sys.meta_path.append(module_finder)
|
||||
#from nemubot.importer import ModuleFinder
|
||||
#module_finder = ModuleFinder(modules_paths, context.add_module)
|
||||
#sys.meta_path.append(module_finder)
|
||||
|
||||
# Load requested configuration files
|
||||
for path in args.files:
|
||||
if not os.path.isfile(path):
|
||||
logger.error("%s is not a readable file", path)
|
||||
continue
|
||||
|
||||
config = load_config(path)
|
||||
|
||||
# Preset each server in this file
|
||||
for server in config.servers:
|
||||
# Add the server in the context
|
||||
for i in [0,1,2,3]:
|
||||
srv = server.server(config, trynb=i)
|
||||
try:
|
||||
if context.add_server(srv):
|
||||
logger.info("Server '%s' successfully added.", srv.name)
|
||||
else:
|
||||
logger.error("Can't add server '%s'.", srv.name)
|
||||
except Exception as e:
|
||||
logger.error("Unable to connect to '%s': %s", srv.name, e)
|
||||
continue
|
||||
break
|
||||
|
||||
# Load module and their configuration
|
||||
for mod in config.modules:
|
||||
context.modules_configuration[mod.name] = mod
|
||||
if mod.autoload:
|
||||
try:
|
||||
__import__("nemubot.module." + mod.name)
|
||||
except:
|
||||
logger.exception("Exception occurs when loading module"
|
||||
" '%s'", mod.name)
|
||||
|
||||
# Load files asked by the configuration file
|
||||
args.files += config.includes
|
||||
|
||||
#for path in args.files:
|
||||
# if os.path.isfile(path):
|
||||
# sync_act("loadconf", path)
|
||||
# else:
|
||||
# logger.error("%s is not a readable file", path)
|
||||
|
||||
if args.module:
|
||||
for module in args.module:
|
||||
__import__("nemubot.module." + module)
|
||||
__import__(module)
|
||||
|
||||
if args.socketfile:
|
||||
from nemubot.server.socket import UnixSocketListener
|
||||
context.add_server(UnixSocketListener(new_server_cb=context.add_server,
|
||||
location=args.socketfile,
|
||||
name="master_socket"))
|
||||
# Signals handling ################################################
|
||||
|
||||
# Daemonize
|
||||
if not args.debug:
|
||||
from nemubot import daemonize
|
||||
daemonize(args.socketfile)
|
||||
# SIGINT and SIGTERM
|
||||
|
||||
# Signals handling
|
||||
def sigtermhandler(signum, frame):
|
||||
def ask_exit(signame):
|
||||
"""On SIGTERM and SIGINT, quit nicely"""
|
||||
context.quit()
|
||||
signal.signal(signal.SIGINT, sigtermhandler)
|
||||
signal.signal(signal.SIGTERM, sigtermhandler)
|
||||
context.stop()
|
||||
|
||||
def sighuphandler(signum, frame):
|
||||
"""On SIGHUP, perform a deep reload"""
|
||||
for sig in (signal.SIGINT, signal.SIGTERM):
|
||||
context._loop.add_signal_handler(sig, functools.partial(ask_exit, sig))
|
||||
|
||||
|
||||
# SIGHUP
|
||||
|
||||
def ask_reload():
|
||||
"""Perform a deep reload"""
|
||||
nonlocal context
|
||||
|
||||
logger.debug("SIGHUP receive, iniate reload procedure...")
|
||||
|
||||
# Reload configuration file
|
||||
for path in args.files:
|
||||
if os.path.isfile(path):
|
||||
sync_act("loadconf", path)
|
||||
signal.signal(signal.SIGHUP, sighuphandler)
|
||||
#for path in args.files:
|
||||
# if os.path.isfile(path):
|
||||
# sync_act("loadconf", path)
|
||||
context._loop.add_signal_handler(signal.SIGHUP, ask_reload)
|
||||
|
||||
def sigusr1handler(signum, frame):
|
||||
"""On SIGHUSR1, display stacktraces"""
|
||||
|
||||
# SIGUSR1
|
||||
|
||||
def ask_debug_info():
|
||||
"""Display debug informations and stacktraces"""
|
||||
import threading, traceback
|
||||
for threadId, stack in sys._current_frames().items():
|
||||
thName = "#%d" % threadId
|
||||
|
|
@ -228,52 +189,26 @@ def main():
|
|||
logger.debug("########### Thread %s:\n%s",
|
||||
thName,
|
||||
"".join(traceback.format_stack(stack)))
|
||||
signal.signal(signal.SIGUSR1, sigusr1handler)
|
||||
context._loop.add_signal_handler(signal.SIGUSR1, ask_debug_info)
|
||||
|
||||
# Store PID to pidfile
|
||||
if args.pidfile is not None:
|
||||
with open(args.pidfile, "w+") as f:
|
||||
f.write(str(os.getpid()))
|
||||
#if args.socketfile:
|
||||
# from nemubot.server.socket import UnixSocketListener
|
||||
# context.add_server(UnixSocketListener(new_server_cb=context.add_server,
|
||||
# location=args.socketfile,
|
||||
# name="master_socket"))
|
||||
|
||||
# context can change when performing an hotswap, always join the latest context
|
||||
oldcontext = None
|
||||
while oldcontext != context:
|
||||
oldcontext = context
|
||||
context.start()
|
||||
context.join()
|
||||
context.run()
|
||||
|
||||
# Wait for consumers
|
||||
logger.info("Waiting for other threads shuts down...")
|
||||
if args.debug:
|
||||
sigusr1handler(0, None)
|
||||
ask_debug_info()
|
||||
sys.exit(0)
|
||||
|
||||
|
||||
def load_config(filename):
|
||||
"""Load a configuration file
|
||||
|
||||
Arguments:
|
||||
filename -- the path to the file to load
|
||||
"""
|
||||
|
||||
from nemubot.channel import Channel
|
||||
from nemubot import config
|
||||
from nemubot.tools.xmlparser import XMLParser
|
||||
|
||||
try:
|
||||
p = XMLParser({
|
||||
"nemubotconfig": config.Nemubot,
|
||||
"server": config.Server,
|
||||
"channel": Channel,
|
||||
"module": config.Module,
|
||||
"include": config.Include,
|
||||
})
|
||||
return p.parse_file(filename)
|
||||
except:
|
||||
logger.exception("Can't load `%s'; this is not a valid nemubot "
|
||||
"configuration file.", filename)
|
||||
return None
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
|
|
|||
527
nemubot/bot.py
527
nemubot/bot.py
|
|
@ -14,535 +14,34 @@
|
|||
# You should have received a copy of the GNU Affero General Public License
|
||||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
from datetime import datetime, timezone
|
||||
import logging
|
||||
from multiprocessing import JoinableQueue
|
||||
import threading
|
||||
import select
|
||||
import sys
|
||||
import weakref
|
||||
|
||||
from nemubot import __version__
|
||||
from nemubot.consumer import Consumer, EventConsumer, MessageConsumer
|
||||
from nemubot import datastore
|
||||
import nemubot.hooks
|
||||
|
||||
logger = logging.getLogger("nemubot")
|
||||
|
||||
sync_queue = JoinableQueue()
|
||||
|
||||
def sync_act(*args):
|
||||
sync_queue.put(list(args))
|
||||
|
||||
|
||||
class Bot(threading.Thread):
|
||||
class Bot:
|
||||
|
||||
"""Class containing the bot context and ensuring key goals"""
|
||||
|
||||
def __init__(self, ip="127.0.0.1", modules_paths=list(),
|
||||
data_store=datastore.Abstract(), debug=False):
|
||||
"""Initialize the bot context
|
||||
logger = logging.getLogger("nemubot")
|
||||
|
||||
Keyword arguments:
|
||||
ip -- The external IP of the bot (default: 127.0.0.1)
|
||||
modules_paths -- Paths to all directories where looking for modules
|
||||
data_store -- An instance of the nemubot datastore for bot's modules
|
||||
debug -- enable debug
|
||||
def __init__(self):
|
||||
"""Initialize the bot context
|
||||
"""
|
||||
|
||||
super().__init__(name="Nemubot main")
|
||||
from nemubot import __version__
|
||||
Bot.logger.info("Initiate nemubot v%s, running on Python %s",
|
||||
__version__, sys.version.split("\n")[0])
|
||||
|
||||
logger.info("Initiate nemubot v%s (running on Python %s.%s.%s)",
|
||||
__version__,
|
||||
sys.version_info.major, sys.version_info.minor, sys.version_info.micro)
|
||||
|
||||
self.debug = debug
|
||||
self.stop = True
|
||||
|
||||
# External IP for accessing this bot
|
||||
import ipaddress
|
||||
self.ip = ipaddress.ip_address(ip)
|
||||
|
||||
# Context paths
|
||||
self.modules_paths = modules_paths
|
||||
self.datastore = data_store
|
||||
self.datastore.open()
|
||||
|
||||
# Keep global context: servers and modules
|
||||
self._poll = select.poll()
|
||||
self.servers = dict()
|
||||
self.modules = dict()
|
||||
self.modules_configuration = dict()
|
||||
|
||||
# Events
|
||||
self.events = list()
|
||||
self.event_timer = None
|
||||
|
||||
# Own hooks
|
||||
from nemubot.treatment import MessageTreater
|
||||
self.treater = MessageTreater()
|
||||
|
||||
import re
|
||||
def in_ping(msg):
|
||||
return msg.respond("pong")
|
||||
self.treater.hm.add_hook(nemubot.hooks.Message(in_ping,
|
||||
match=lambda msg: re.match("^ *(m[' ]?entends?[ -]+tu|h?ear me|do you copy|ping)",
|
||||
msg.message, re.I)),
|
||||
"in", "DirectAsk")
|
||||
|
||||
def in_echo(msg):
|
||||
from nemubot.message import Text
|
||||
return Text(msg.frm + ": " + " ".join(msg.args), to=msg.to_response)
|
||||
self.treater.hm.add_hook(nemubot.hooks.Command(in_echo, "echo"), "in", "Command")
|
||||
|
||||
def _help_msg(msg):
|
||||
"""Parse and response to help messages"""
|
||||
from nemubot.module.more import Response
|
||||
res = Response(channel=msg.to_response)
|
||||
if len(msg.args) >= 1:
|
||||
if "nemubot.module." + msg.args[0] in self.modules and self.modules["nemubot.module." + msg.args[0]]() is not None:
|
||||
mname = "nemubot.module." + msg.args[0]
|
||||
if hasattr(self.modules[mname](), "help_full"):
|
||||
hlp = self.modules[mname]().help_full()
|
||||
if isinstance(hlp, Response):
|
||||
return hlp
|
||||
else:
|
||||
res.append_message(hlp)
|
||||
else:
|
||||
res.append_message([str(h) for s,h in self.modules[mname]().__nemubot_context__.hooks], title="Available commands for module " + msg.args[0])
|
||||
elif msg.args[0][0] == "!":
|
||||
from nemubot.message.command import Command
|
||||
for h in self.treater._in_hooks(Command(msg.args[0][1:])):
|
||||
if h.help_usage:
|
||||
lp = ["\x03\x02%s%s\x03\x02: %s" % (msg.args[0], (" " + k if k is not None else ""), h.help_usage[k]) for k in h.help_usage]
|
||||
jp = h.keywords.help()
|
||||
return res.append_message(lp + ([". Moreover, you can provides some optional parameters: "] + jp if len(jp) else []), title="Usage for command %s" % msg.args[0])
|
||||
elif h.help:
|
||||
return res.append_message("Command %s: %s" % (msg.args[0], h.help))
|
||||
else:
|
||||
return res.append_message("Sorry, there is currently no help for the command %s. Feel free to make a pull request at https://github.com/nemunaire/nemubot/compare" % msg.args[0])
|
||||
res.append_message("Sorry, there is no command %s" % msg.args[0])
|
||||
else:
|
||||
res.append_message("Sorry, there is no module named %s" % msg.args[0])
|
||||
else:
|
||||
res.append_message("Pour me demander quelque chose, commencez "
|
||||
"votre message par mon nom ; je réagis "
|
||||
"également à certaine commandes commençant par"
|
||||
" !. Pour plus d'informations, envoyez le "
|
||||
"message \"!more\".")
|
||||
res.append_message("Mon code source est libre, publié sous "
|
||||
"licence AGPL (http://www.gnu.org/licenses/). "
|
||||
"Vous pouvez le consulter, le dupliquer, "
|
||||
"envoyer des rapports de bogues ou bien "
|
||||
"contribuer au projet sur GitHub : "
|
||||
"https://github.com/nemunaire/nemubot/")
|
||||
res.append_message(title="Pour plus de détails sur un module, "
|
||||
"envoyez \"!help nomdumodule\". Voici la liste"
|
||||
" de tous les modules disponibles localement",
|
||||
message=["\x03\x02%s\x03\x02 (%s)" % (im, self.modules[im]().__doc__) for im in self.modules if self.modules[im]() is not None and self.modules[im]().__doc__])
|
||||
return res
|
||||
self.treater.hm.add_hook(nemubot.hooks.Command(_help_msg, "help"), "in", "Command")
|
||||
|
||||
import os
|
||||
from queue import Queue
|
||||
# Messages to be treated — shared across all server connections.
|
||||
# cnsr_active tracks consumers currently inside stm.run() (not idle),
|
||||
# which lets us spawn a new thread the moment all existing ones are busy.
|
||||
self.cnsr_queue = Queue()
|
||||
self.cnsr_thrd = list()
|
||||
self.cnsr_lock = threading.Lock()
|
||||
self.cnsr_active = 0 # consumers currently executing a task
|
||||
self.cnsr_max = os.cpu_count() or 4 # upper bound on concurrent consumer threads
|
||||
|
||||
|
||||
def __del__(self):
|
||||
self.datastore.close()
|
||||
import asyncio
|
||||
self._loop = asyncio.get_event_loop()
|
||||
|
||||
|
||||
def run(self):
|
||||
global sync_queue
|
||||
self._loop.run_forever()
|
||||
self._loop.close()
|
||||
|
||||
# Rewrite the sync_queue, as the daemonization process tend to disturb it
|
||||
old_sync_queue, sync_queue = sync_queue, JoinableQueue()
|
||||
while not old_sync_queue.empty():
|
||||
sync_queue.put_nowait(old_sync_queue.get())
|
||||
|
||||
self._poll.register(sync_queue._reader, select.POLLIN | select.POLLPRI)
|
||||
|
||||
|
||||
self.stop = False
|
||||
|
||||
# Relaunch events
|
||||
self._update_event_timer()
|
||||
|
||||
logger.info("Starting main loop")
|
||||
while not self.stop:
|
||||
for fd, flag in self._poll.poll():
|
||||
# Handle internal socket passing orders
|
||||
if fd != sync_queue._reader.fileno() and fd in self.servers:
|
||||
srv = self.servers[fd]
|
||||
|
||||
if flag & (select.POLLERR | select.POLLHUP | select.POLLNVAL):
|
||||
try:
|
||||
srv.exception(flag)
|
||||
except:
|
||||
logger.exception("Uncatched exception on server exception")
|
||||
|
||||
if srv.fileno() > 0:
|
||||
if flag & (select.POLLOUT):
|
||||
try:
|
||||
srv.async_write()
|
||||
except:
|
||||
logger.exception("Uncatched exception on server write")
|
||||
|
||||
if flag & (select.POLLIN | select.POLLPRI):
|
||||
try:
|
||||
for i in srv.async_read():
|
||||
self.receive_message(srv, i)
|
||||
except:
|
||||
logger.exception("Uncatched exception on server read")
|
||||
|
||||
else:
|
||||
del self.servers[fd]
|
||||
|
||||
|
||||
# Always check the sync queue
|
||||
while not sync_queue.empty():
|
||||
args = sync_queue.get()
|
||||
action = args.pop(0)
|
||||
|
||||
logger.debug("Executing sync_queue action %s%s", action, args)
|
||||
|
||||
if action == "sckt" and len(args) >= 2:
|
||||
try:
|
||||
if args[0] == "write":
|
||||
self._poll.modify(int(args[1]), select.POLLOUT | select.POLLIN | select.POLLPRI)
|
||||
elif args[0] == "unwrite":
|
||||
self._poll.modify(int(args[1]), select.POLLIN | select.POLLPRI)
|
||||
|
||||
elif args[0] == "register":
|
||||
self._poll.register(int(args[1]), select.POLLIN | select.POLLPRI)
|
||||
elif args[0] == "unregister":
|
||||
try:
|
||||
self._poll.unregister(int(args[1]))
|
||||
except KeyError:
|
||||
pass
|
||||
except:
|
||||
logger.exception("Unhandled excpetion during action:")
|
||||
|
||||
elif action == "exit":
|
||||
self.quit()
|
||||
|
||||
elif action == "launch_consumer":
|
||||
pass # This is treated after the loop
|
||||
|
||||
sync_queue.task_done()
|
||||
|
||||
|
||||
# Spawn a new consumer whenever the queue has work and every
|
||||
# existing consumer is already busy executing a task.
|
||||
with self.cnsr_lock:
|
||||
while (not self.cnsr_queue.empty()
|
||||
and self.cnsr_active >= len(self.cnsr_thrd)
|
||||
and len(self.cnsr_thrd) < self.cnsr_max):
|
||||
c = Consumer(self)
|
||||
self.cnsr_thrd.append(c)
|
||||
c.start()
|
||||
sync_queue = None
|
||||
logger.info("Ending main loop")
|
||||
|
||||
|
||||
|
||||
# Events methods
|
||||
|
||||
def add_event(self, evt, eid=None, module_src=None):
|
||||
"""Register an event and return its identifiant for futur update
|
||||
|
||||
Return:
|
||||
None if the event is not in the queue (eg. if it has been executed during the call) or
|
||||
returns the event ID.
|
||||
|
||||
Argument:
|
||||
evt -- The event object to add
|
||||
|
||||
Keyword arguments:
|
||||
eid -- The desired event ID (object or string UUID)
|
||||
module_src -- The module to which the event is attached to
|
||||
"""
|
||||
|
||||
import uuid
|
||||
|
||||
# Generate the event id if no given
|
||||
if eid is None:
|
||||
eid = uuid.uuid1()
|
||||
|
||||
# Fill the id field of the event
|
||||
if type(eid) is uuid.UUID:
|
||||
evt.id = str(eid)
|
||||
else:
|
||||
# Ok, this is quiet useless...
|
||||
try:
|
||||
evt.id = str(uuid.UUID(eid))
|
||||
except ValueError:
|
||||
evt.id = eid
|
||||
|
||||
# TODO: mutex here plz
|
||||
|
||||
# Add the event in its place
|
||||
t = evt.current
|
||||
i = 0 # sentinel
|
||||
for i in range(0, len(self.events)):
|
||||
if self.events[i].current > t:
|
||||
break
|
||||
self.events.insert(i, evt)
|
||||
|
||||
if i == 0 and not self.stop:
|
||||
# First event changed, reset timer
|
||||
self._update_event_timer()
|
||||
if len(self.events) <= 0 or self.events[i] != evt:
|
||||
# Our event has been executed and removed from queue
|
||||
return None
|
||||
|
||||
# Register the event in the source module
|
||||
if module_src is not None:
|
||||
module_src.__nemubot_context__.events.append((evt, evt.id))
|
||||
evt.module_src = module_src
|
||||
|
||||
logger.info("New event registered in %d position: %s", i, t)
|
||||
return evt.id
|
||||
|
||||
|
||||
def del_event(self, evt, module_src=None):
|
||||
"""Find and remove an event from list
|
||||
|
||||
Return:
|
||||
True if the event has been found and removed, False else
|
||||
|
||||
Argument:
|
||||
evt -- The ModuleEvent object to remove or just the event identifier
|
||||
|
||||
Keyword arguments:
|
||||
module_src -- The module to which the event is attached to (ignored if evt is a ModuleEvent)
|
||||
"""
|
||||
|
||||
logger.info("Removing event: %s from %s", evt, module_src)
|
||||
|
||||
from nemubot.event import ModuleEvent
|
||||
if type(evt) is ModuleEvent:
|
||||
id = evt.id
|
||||
module_src = evt.module_src
|
||||
else:
|
||||
id = evt
|
||||
|
||||
if len(self.events) > 0 and id == self.events[0].id:
|
||||
if module_src is not None:
|
||||
module_src.__nemubot_context__.events.remove((self.events[0], id))
|
||||
self.events.remove(self.events[0])
|
||||
self._update_event_timer()
|
||||
return True
|
||||
|
||||
for evt in self.events:
|
||||
if evt.id == id:
|
||||
self.events.remove(evt)
|
||||
|
||||
if module_src is not None:
|
||||
module_src.__nemubot_context__.events.remove((evt, evt.id))
|
||||
return True
|
||||
return False
|
||||
|
||||
|
||||
def _update_event_timer(self):
|
||||
"""(Re)launch the timer to end with the closest event"""
|
||||
|
||||
# Reset the timer if this is the first item
|
||||
if self.event_timer is not None:
|
||||
self.event_timer.cancel()
|
||||
|
||||
if len(self.events):
|
||||
try:
|
||||
remaining = self.events[0].time_left.total_seconds()
|
||||
except:
|
||||
logger.exception("An error occurs during event time calculation:")
|
||||
self.events.pop(0)
|
||||
return self._update_event_timer()
|
||||
|
||||
logger.debug("Update timer: next event in %d seconds", remaining)
|
||||
self.event_timer = threading.Timer(remaining if remaining > 0 else 0, self._end_event_timer)
|
||||
self.event_timer.start()
|
||||
|
||||
else:
|
||||
logger.debug("Update timer: no timer left")
|
||||
|
||||
|
||||
def _end_event_timer(self):
|
||||
"""Function called at the end of the event timer"""
|
||||
|
||||
while len(self.events) > 0 and datetime.now(timezone.utc) >= self.events[0].current:
|
||||
evt = self.events.pop(0)
|
||||
self.cnsr_queue.put_nowait(EventConsumer(evt))
|
||||
sync_act("launch_consumer")
|
||||
|
||||
self._update_event_timer()
|
||||
|
||||
|
||||
# Consumers methods
|
||||
|
||||
def add_server(self, srv, autoconnect=True):
|
||||
"""Add a new server to the context
|
||||
|
||||
Arguments:
|
||||
srv -- a concrete AbstractServer instance
|
||||
autoconnect -- connect after add?
|
||||
"""
|
||||
|
||||
fileno = srv.fileno()
|
||||
if fileno not in self.servers:
|
||||
self.servers[fileno] = srv
|
||||
self.servers[srv.name] = srv
|
||||
if autoconnect and not hasattr(self, "noautoconnect"):
|
||||
srv.connect()
|
||||
return True
|
||||
|
||||
else:
|
||||
return False
|
||||
|
||||
|
||||
# Modules methods
|
||||
|
||||
def import_module(self, name):
|
||||
"""Load a module
|
||||
|
||||
Argument:
|
||||
name -- name of the module to load
|
||||
"""
|
||||
|
||||
if name in self.modules:
|
||||
self.unload_module(name)
|
||||
|
||||
__import__(name)
|
||||
|
||||
|
||||
def add_module(self, module):
|
||||
"""Add a module to the context, if already exists, unload the
|
||||
old one before"""
|
||||
module_name = module.__spec__.name if hasattr(module, "__spec__") else module.__name__
|
||||
|
||||
# Check if the module already exists
|
||||
if module_name in self.modules:
|
||||
self.unload_module(module_name)
|
||||
|
||||
# Overwrite print built-in
|
||||
def prnt(*args):
|
||||
if hasattr(module, "logger"):
|
||||
module.logger.info(" ".join([str(s) for s in args]))
|
||||
else:
|
||||
logger.info("[%s] %s", module_name, " ".join([str(s) for s in args]))
|
||||
module.print = prnt
|
||||
|
||||
# Create module context
|
||||
from nemubot.modulecontext import _ModuleContext, ModuleContext
|
||||
module.__nemubot_context__ = ModuleContext(self, module)
|
||||
|
||||
if not hasattr(module, "logger"):
|
||||
module.logger = logging.getLogger("nemubot.module." + module_name)
|
||||
|
||||
# Replace imported context by real one
|
||||
for attr in module.__dict__:
|
||||
if attr != "__nemubot_context__" and type(module.__dict__[attr]) == _ModuleContext:
|
||||
module.__dict__[attr] = module.__nemubot_context__
|
||||
|
||||
# Register decorated functions
|
||||
import nemubot.hooks
|
||||
for s, h in nemubot.hooks.hook.last_registered:
|
||||
module.__nemubot_context__.add_hook(h, *s if isinstance(s, list) else s)
|
||||
nemubot.hooks.hook.last_registered = []
|
||||
|
||||
# Launch the module
|
||||
if hasattr(module, "load"):
|
||||
try:
|
||||
module.load(module.__nemubot_context__)
|
||||
except:
|
||||
module.__nemubot_context__.unload()
|
||||
raise
|
||||
|
||||
# Save a reference to the module
|
||||
self.modules[module_name] = weakref.ref(module)
|
||||
logger.info("Module '%s' successfully loaded.", module_name)
|
||||
|
||||
|
||||
def unload_module(self, name):
|
||||
"""Unload a module"""
|
||||
if name in self.modules and self.modules[name]() is not None:
|
||||
module = self.modules[name]()
|
||||
module.print("Unloading module %s" % name)
|
||||
|
||||
# Call the user defined unload method
|
||||
if hasattr(module, "unload"):
|
||||
module.unload(self)
|
||||
module.__nemubot_context__.unload()
|
||||
|
||||
# Remove from the nemubot dict
|
||||
del self.modules[name]
|
||||
|
||||
# Remove from the Python dict
|
||||
del sys.modules[name]
|
||||
for mod in [i for i in sys.modules]:
|
||||
if mod[:len(name) + 1] == name + ".":
|
||||
logger.debug("Module '%s' also removed from system modules list.", mod)
|
||||
del sys | ||||