nemunaire
/
nemubot
1
0
Fork 0
Code Releases Activity

suivi: kuse new laposte API to get infos

This commit is contained in:
nemunaire 2019-12-02 19:31:57 +01:00
parent 904ad72316
commit 37a230e70e
1 changed files with 13 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
modules/suivi.py
View File

@ -9,7 +9,7 @@ import re
from nemubot.hooks import hook from nemubot.hooks import hook
from nemubot.exception import IMException from nemubot.exception import IMException
from nemubot.tools.web import getURLContent, getJSON from nemubot.tools.web import getURLContent, getURLHeaders, getJSON
from nemubot.module.more import Response from nemubot.module.more import Response
@ -76,32 +76,17 @@ def get_colisprive_info(track_id):
def get_laposte_info(laposte_id): def get_laposte_info(laposte_id):
data = urllib.parse.urlencode({'id': laposte_id}) status, laposte_headers = getURLHeaders("https://www.laposte.fr/outils/suivre-vos-envois?" + urllib.parse.urlencode({'code': laposte_id}))
laposte_baseurl = "http://www.part.csuivi.courrier.laposte.fr/suivi/index"
laposte_data = getURLContent(laposte_baseurl, data.encode('utf-8')) laposte_cookie = None
soup = BeautifulSoup(laposte_data) for k,v in laposte_headers:
search_res = soup.find(class_='resultat_rech_simple_table').tbody.tr if k.lower() == "set-cookie" and v.find("access_token") >= 0:
if (soup.find(class_='resultat_rech_simple_table').thead laposte_cookie = v.split(";")[0]
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') 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})
poste_type = field.get_text()
field = field.find_next('td') shipment = laposte_data["shipment"]
poste_date = field.get_text() return (shipment["product"], shipment["idShip"], shipment["event"][0]["label"], shipment["event"][0]["date"])
field = field.find_next('td')
poste_location = field.get_text()
field = field.find_next('td')
poste_status = field.get_text()
return (poste_type.lower(), poste_id.strip(), poste_status.lower(),
poste_location, poste_date)
def get_postnl_info(postnl_id): def get_postnl_info(postnl_id):
@ -210,11 +195,10 @@ def handle_tnt(tracknum):
def handle_laposte(tracknum): def handle_laposte(tracknum):
info = get_laposte_info(tracknum) info = get_laposte_info(tracknum)
if info: if info:
poste_type, poste_id, poste_status, poste_location, poste_date = info poste_type, poste_id, poste_status, poste_date = info
return ("Le courrier de type \x02%s\x0F : \x02%s\x0F est actuellement " return ("\x02%s\x0F : \x02%s\x0F est actuellement "
"\x02%s\x0F dans la zone \x02%s\x0F (Mis à jour le \x02%s\x0F" "\x02%s\x0F (Mis à jour le \x02%s\x0F"
")." % (poste_type, poste_id, poste_status, ")." % (poste_type, poste_id, poste_status, poste_date))
poste_location, poste_date))
def handle_postnl(tracknum): def handle_postnl(tracknum):