bot.py

#!/usr/bin/env python3
"""Main bot file. It will communicate with Telegram API."""

import urllib
import json
import requests
import time

from dbhelper import DBHelper
from logger import Logger

__author__ = "S. Vahid Hosseini"
__copyright__ = "Copyright 2019, IOT Module Controller"
__credits__ = ["S. Vahid Hosseini"]
__license__ = "GPL-3.0"
__version__ = "0.1"
__maintainer__ = "S. Vahid Hosseini"
__email__ = "s.vahid.h@behmerd.ir"
__status__ = "Dev"


#global variables
TOKEN = "970592554:AAFMNLvMAShUMgjcw_XbKsN3ozI9psrEVAQ"
URL = "https://api.telegram.org/bot{}/".format(TOKEN)
database = DBHelper()
log = Logger()

#API handlers
def get_url(url):
    response = requests.get(url)
    content = response.content.decode("utf8")
    return content


def get_json_from_url(url):
    content = get_url(url)
    js = json.loads(content)
    return js


def get_updates(offset=None):
    url = URL + "getUpdates?timeout=100"
    if offset:
        url += "&offset={}".format(offset) 
    js = get_json_from_url(url)
    return js


def get_last_chat_id_and_text(updates):
    num_updates = len(updates["result"])
    last_update = num_updates -1
    text = updates["result"][last_update]["message"]["text"]
    chat_id = updates["result"][last_update]["message"]["chat"]["id"]
    return (text, chat_id)


def send_message(text, chat_id, reply_markup=None):
    text = urllib.parse.quote_plus(text)
    url = URL + "sendMessage?text={}&chat_id={}&parse_mode=Markdown".format(text, chat_id)
    if reply_markup:
        url += "&reply_markup={}".format(reply_markup)
    get_url(url)


def get_last_update_id(updates):
    update_ids = []
    for update in updates["result"]:
        update_ids.append(int(update["update_id"]))
    return max(update_ids)


def build_keyboard(items):
    keyboard = [[item] for item in items]
    reply_markup = {"keyboard":keyboard, "one_time_keyboard": True}
    return json.dumps(reply_markup)


def handle_updates(updates):
    for update in updates["result"]:
        try:
            text = update["message"]["text"]
            chat = update["message"]["chat"]["id"]
            user = update["message"]["from"]
            args = {"sender":"main", "user":user["username"]}
            log.out.info('update received: %s', text, extra=args)
            if text.startswith("/"):
                command = text.substing(1)
                if command == "start":
                    start_bot(name, user, chat)
            elif text.startswith("$"):
                command = text.substing(1)
                terminal(command, chat)
            else:
                pass
        except KeyError:
            pass


#internal commands
def terminal(command, chat):
    response = "`"
    if command in NA:
        response += "'{}' is not supported!".format(command)
    elif command in ROOT and login-check() == False:
        response += "Unauthorized access! Please login first!"
    else:
        response += excecute(command)

    response += "`"
    send_message(response, chat)


def excecute(command):
    try:
        if command == "login":
            args = command.split("=")
            if args[1] in ACCOUNTS:
                return "Insert your password:"
            else:
                return "Login failed!"
        else:
            pass
    except Exception as e:
        print(e)
        return str(e)


def login(username):
    send_message("**Login:**\n`Insert your password:`", chat)


#bot commands
def start_bot(name, user, chat):
	send_message("IOT module controller.\nWelcome {}!".format(name), chat)
    #login(chat)


#utils


#main
def main():
    last_update_id = None
    database.setup()
    while True:
        updates = get_updates(last_update_id)
        if len(updates["result"]) > 0:
            last_update_id = get_last_update_id(updates) + 1
            handle_updates(updates)
        time.sleep(0.5)


if __name__ == '__main__':
    main()