python-telegram-bot/telegram/utils/botan.py
2016-01-23 16:58:01 +03:00

53 lines
1.7 KiB
Python

#!/usr/bin/env python
import json
import logging
from telegram import NullHandler
try:
from urllib.request import urlopen, Request
from urllib.parse import quote
from urllib.error import URLError, HTTPError
except ImportError:
from urllib2 import urlopen, Request
from urllib import quote
from urllib2 import URLError, HTTPError
H = NullHandler()
logging.getLogger(__name__).addHandler(H)
class Botan(object):
token = ''
url_template = 'https://api.botan.io/track?' \
'token={token}&uid={uid}&name={name}'
def __init__(self, token):
self.token = token
self.logger = logging.getLogger(__name__)
def track(self, message, event_name='event'):
try:
uid = message.chat_id
except AttributeError:
self.logger.warn('No chat_id in message')
return False
data = json.dumps(message.__dict__)
try:
url = self.url_template.format(token=str(self.token),
uid=str(uid),
name=quote(event_name))
request = Request(url,
data=data,
headers={'Content-Type': 'application/json'})
response = urlopen(request, json.dumps(data))
if response.getcode() != 200:
return False
return True
except HTTPError as error:
self.logger.warn('Botan track error ' +
str(error.code) + ':' + error.reason)
return False
except URLError as error:
self.logger.warn('Botan track error ' + error.reason)
return False