From 04d5be04fabfdc68990e758761ff2c9e9e57f8c9 Mon Sep 17 00:00:00 2001
From: Max <max@23.tf>
Date: Thu, 29 Oct 2015 02:10:46 +0100
Subject: [PATCH] [suivi] Add TNT support

---
 modules/suivi.py | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/modules/suivi.py b/modules/suivi.py
index 80e0345..bdd9322 100644
--- a/modules/suivi.py
+++ b/modules/suivi.py
@@ -13,6 +13,15 @@ nemubotversion = 4.0
 # POSTAGE SERVICE PARSERS ############################################
 
 
+def get_tnt_info(track_id):
+    data = getURLContent('www.tnt.fr/public/suivi_colis/recherche/'
+                         'visubontransport.do?bonTransport=%s' % track_id)
+    soup = BeautifulSoup(data)
+    status = soup.find('p', class_='suivi-title-selected')
+    if status:
+        return status.get_text()
+
+
 def get_colissimo_info(colissimo_id):
     colissimo_data = getURLContent("http://www.colissimo.fr/portail_colissimo/"
                                    "suivre.do?colispart=%s" % colissimo_id)
@@ -93,6 +102,13 @@ def get_laposte_info(laposte_id):
 # TRACKING HANDLERS ###################################################
 
 
+def handle_tnt(tracknum):
+    info = get_tnt_info(tracknum)
+    if info:
+        return ('Le colis \x02{trackid}\x0f a actuellement le status: '
+                '\x02{status}\x0F'.format(trackid=tracknum, status=info))
+
+
 def handle_laposte(tracknum):
     info = get_laposte_info(tracknum)
     if info:
@@ -129,7 +145,8 @@ TRACKING_HANDLERS = {
     'laposte': handle_laposte,
     'colissimo': handle_colissimo,
     'chronopost': handle_chronopost,
-    'coliprive': handle_coliprive
+    'coliprive': handle_coliprive,
+    'tnt': handle_tnt
 }