nemunaire/nemubot
1
0
Fork 0
Code Releases Activity

[suivi] Now using nemubot.tools.web for queries

Browse Source
This commit is contained in:
Max 2015-07-13 22:25:40 +01:00 committed by nemunaire
parent 5b039edb62
commit 710896f711
1 changed files with 5 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
modules/suivi.py
View File

@ -4,17 +4,16 @@ from bs4 import BeautifulSoup
from nemubot.hooks import hook from nemubot.hooks import hook
from nemubot.exception import IRCException from nemubot.exception import IRCException
from nemubot.tools.web import getURLContent
from more import Response from more import Response
nemubotversion = 3.4 nemubotversion = 4.0
def help_full(): def help_full():
return "Traquez vos courriers La Poste en utilisant la commande: !laposte <tracking number>\nCe service se base sur http://www.csuivi.courrier.laposte.fr/suivi/index" return "Traquez vos courriers La Poste ou Colissimo en utilisant la commande: !laposte <tracking number> ou !colissimo <tracking number>\nCe service se base sur http://www.csuivi.courrier.laposte.fr/suivi/index et http://www.colissimo.fr/portail_colissimo/suivre.do"
def get_colissimo_info(colissimo_id): def get_colissimo_info(colissimo_id):
data = urllib.parse.urlencode({'colispart': colissimo_id}) colissimo_data = getURLContent("http://www.colissimo.fr/portail_colissimo/suivre.do?colispart=%s" % colissimo_id)
colissimo_baseurl = "http://www.colissimo.fr/portail_colissimo/suivre.do"
colissimo_data = urllib.request.urlopen(colissimo_baseurl, data.encode('utf-8'))
soup = BeautifulSoup(colissimo_data) soup = BeautifulSoup(colissimo_data)
dataArray = soup.find(class_='dataArray') dataArray = soup.find(class_='dataArray')
@ -25,10 +24,7 @@ def get_colissimo_info(colissimo_id):
return (date, libelle, site.strip()) return (date, libelle, site.strip())
def get_laposte_info(laposte_id): def get_laposte_info(laposte_id):
data = urllib.parse.urlencode({'id': laposte_id}) laposte_data = getURLContent("http://www.part.csuivi.courrier.laposte.fr/suivi/index?id=%s" % laposte_id)
laposte_baseurl = "http://www.part.csuivi.courrier.laposte.fr/suivi/index"
laposte_data = urllib.request.urlopen(laposte_baseurl, data.encode('utf-8'))
soup = BeautifulSoup(laposte_data) soup = BeautifulSoup(laposte_data)
search_res = soup.find(class_='resultat_rech_simple_table').tbody.tr search_res = soup.find(class_='resultat_rech_simple_table').tbody.tr
if (soup.find(class_='resultat_rech_simple_table').thead if (soup.find(class_='resultat_rech_simple_table').thead
@ -49,7 +45,6 @@ def get_laposte_info(laposte_id):
poste_status = field.get_text() poste_status = field.get_text()
return (poste_type.lower(), poste_id.strip(), poste_status.lower(), poste_location, poste_date) return (poste_type.lower(), poste_id.strip(), poste_status.lower(), poste_location, poste_date)
@hook("cmd_hook", "colissimo") @hook("cmd_hook", "colissimo")
def get_colissimo_tracking_info(msg): def get_colissimo_tracking_info(msg):
if not len(msg.args): if not len(msg.args):