def init():
"""
Creates tables if they do not exists
"""
if not Chat.table_exists():
Chat.create_table()
if not Metrika.table_exists():
Metrika.create_table()
if not Metrika_tokens.table_exists():
Metrika_tokens.create_table()
if not Users.table_exists():
Users.create_table()
if not Github_repositories.table_exists():
Github_repositories.create_table()
data = {
'url': TELEGRAM_CALLBACK_URL
}
if not CConfig.CERTIFICATE:
query = 'https://api.telegram.org/bot%s/setWebhook?%s' % (CConfig.API_TOKEN, urlencode(data))
result = requests.get(query)
else:
query = 'https://api.telegram.org/bot%s/setWebhook?%s' % (CConfig.API_TOKEN, urlencode(data))
def add(self, chat_id, token_id, counter_id):
# TODO: Params as hash id in cache
try:
token = Metrika_tokens.get(Metrika_tokens.id == token_id)
try:
counter = Metrika.select().where((Metrika.counter_id == counter_id) & (Metrika.chat_id == chat_id)).get()
send_to_chat("Счетчик <%s> уже прикреплен к данному чату." % counter.counter_name, chat_id)
except Metrika.DoesNotExist:
counter_name = CConfig.CACHE['metrika_%s_%s' % (token_id, counter_id)]
metrika = Metrika.create(oauth_token=token.token,
counter_id=counter_id,
chat_id=chat_id,
counter_name=counter_name,
client_id=CConfig.MODULES['metrika']['ID'])
metrika.save()
send_to_chat("Готово! Сайт <%s> успешно подключен." % counter_name, chat_id)
except Metrika_tokens.DoesNotExist:
send_to_chat("Ошибка токена <%s>." % token_id, chat_id)
def delete(self, chat_id, token_id, counter_id):
try:
token = Metrika_tokens.get(Metrika_tokens.id == token_id)
try:
counter = Metrika.select().where((Metrika.counter_id == counter_id) & (Metrika.chat_id == chat_id)).get()
counter.delete_instance()
counter_name = CConfig.CACHE['metrika_%s_%s' % (token_id, counter_id)]
send_to_chat("Готово! Сайт <%s> отключен от модуля." % counter_name, chat_id)
except Metrika.DoesNotExist:
send_to_chat("Сайт не найден.", chat_id)
except Metrika_tokens.DoesNotExist:
send_to_chat("Ошибка токена <%s>." % token_id, chat_id)
def get(self, chat_id, cmd):
metrikas = Metrika.select().where(Metrika.chat_id == chat_id)
message = "%s\n\n" % self.stats(cmd)
for metrika in metrikas:
try:
metrikaAPI = MetrikaAPI(metrika.oauth_token, metrika.counter_id, chat_id)
result = metrikaAPI.get_visit_statistics(cmd)
if result:
users, hits = result
message += "%s:\n%d уникальных посетителей\n%d просмотров\n\n" % (metrika.counter_name,
users, hits)
except Exception as e:
logging.warning("Metrika API exception: %s" % e)
send_to_chat(message, chat_id)
def help(self, chat_id):
msg = "Этот модуль поможет вам следить за статистикой сайта. Возможности модуля: \n\n" \
"- моментальное получение текущих значений счетчиков (DAU, просмотры, источники) за период (день, неделя, месяц)\n" \
"- уведомление о достижении целей (например, бот сообщит о достижении показателя в 10k уникальных посетителей)"
metrikas = Metrika.select().where(Metrika.chat_id == chat_id)
if not len(metrikas):
msg += "\n\nВ данный момент модуль не активирован.\n\nДля настройки модуля, используйте команду /metrika_start"
else:
msg += "\n\nПодключенные сайты.\n\n"
for metrika in metrikas:
msg += "%s\n" % metrika.counter_name
msg += "\nДля отключения счетчика используйте команду /metrika_stop\n" \
"Подключить еще один сайт можно с помощью команды /metrika_start\n\n" \
"Меню модуля: /metrika_help"
return msg
def metrika_stop(self, chat_id):
try:
token = Metrika_tokens.get(Metrika_tokens.chat_id == chat_id)
counters = Metrika.select().where(Metrika.chat_id == chat_id)
if len(counters):
buttons = []
buttons_row = []
for counter in counters:
buttons_row.append(
{'text': counter.counter_name, 'callback_data': "/metrika_%s%s#%s" % ("stop", token.id, counter.counter_id)})
CConfig.CACHE['metrika_%s_%s' % (token.id, counter.counter_id)] = counter.counter_name
if len(buttons_row) == 2:
buttons.append(buttons_row[:])
buttons_row = []
if len(buttons_row):
buttons.append(buttons_row[:])
send_object_to_chat("Выберите сайт, который хотите отключить.\n", json.dumps({
'inline_keyboard': buttons
}), chat_id)
else:
send_to_chat("Подключенных счетчиков не найдено.", chat_id)
except Metrika_tokens.DoesNotExist:
send_to_chat("Токен не найден.", chat_id)
