""" NooLite API wrapper """ import requests from requests.auth import HTTPBasicAuth from requests.exceptions import ConnectTimeout, ConnectionError import xml.etree.ElementTree as ET class NooLiteSens: """    ,         """ def __init__(self, temperature, humidity, state): self.temperature = float(temperature.replace(',', '.')) if temperature != '-' else None self.humidity = int(humidity) if humidity != '-' else None self.state = state class NooLiteApi: """     NooLite""" def __init__(self, login, password, base_api_url, request_timeout=10): self.login = login self.password = password self.base_api_url = base_api_url self.request_timeout = request_timeout def get_sens_data(self): """   xml    :return:  NooLiteSens     :rtype: list """ response = self._send_request('{}/sens.xml'.format(self.base_api_url)) sens_states = { 0: ' ,   ', 1: '  ', 2: '   ', 3: '     ' } response_xml_root = ET.fromstring(response.text) sens_list = [] for sens_number in range(4): sens_list.append(NooLiteSens( response_xml_root.find('snst{}'.format(sens_number)).text, response_xml_root.find('snsh{}'.format(sens_number)).text, sens_states.get(int(response_xml_root.find('snt{}'.format(sens_number)).text)) )) return sens_list def send_command_to_channel(self, data): """   NooLite    NooLite  url   data :param data: url  :type data: dict :return: response """ return self._send_request('{}/api.htm'.format(self.base_api_url), params=data) def _send_request(self, url, **kwargs): """   NooLite        url    kwargs :param url: url   :type url: str :return: response  NooLite   """ try: response = requests.get(url, auth=HTTPBasicAuth(self.login, self.password), timeout=self.request_timeout, **kwargs) except ConnectTimeout as e: print(e) raise NooLiteConnectionTimeout('Connection timeout: {}'.format(self.request_timeout)) except ConnectionError as e: print(e) raise NooLiteConnectionError('Connection timeout: {}'.format(self.request_timeout)) if response.status_code != 200: raise NooLiteBadResponse('Bad response: {}'.format(response)) else: return response #   NooLiteConnectionTimeout = type('NooLiteConnectionTimeout', (Exception,), {}) NooLiteConnectionError = type('NooLiteConnectionError', (Exception,), {}) NooLiteBadResponse = type('NooLiteBadResponse', (Exception,), {}) NooLiteBadRequestMethod = type('NooLiteBadRequestMethod', (Exception,), {})
      
      

        
        
        
      

    
        
        
        
      
      

        
        
        
      

    
     

 import os import logging import functools import yaml import requests import telnetlib from requests.exceptions import ConnectionError from telegram import ReplyKeyboardMarkup, ParseMode from telegram.ext import Updater, CommandHandler, Filters, MessageHandler, Job from noolite_api import NooLiteApi, NooLiteConnectionTimeout,\ NooLiteConnectionError, NooLiteBadResponse #      config = yaml.load(open('conf.yaml')) #    logger = logging.getLogger() logger.setLevel(logging.INFO) formatter = logging.Formatter( '%(asctime)s - %(filename)s:%(lineno)s - %(levelname)s - %(message)s' ) stream_handler = logging.StreamHandler() stream_handler.setFormatter(formatter) logger.addHandler(stream_handler) #     NooLite updater = Updater(config['telegtam']['token']) noolite_api = NooLiteApi( config['noolite']['login'], config['noolite']['password'], config['noolite']['api_url'] ) job_queue = updater.job_queue def auth_required(func): """ """ @functools.wraps(func) def wrapped(bot, update): if update.message.chat_id not in config['telegtam']['authenticated_users']: bot.sendMessage( chat_id=update.message.chat_id, text=" .\n   /auth password." ) else: return func(bot, update) return wrapped def log(func): """ """ @functools.wraps(func) def wrapped(bot, update): logger.info('Received message: {}'.format( update.message.text if update.message else update.callback_query.data) ) func(bot, update) logger.info('Response was sent') return wrapped def start(bot, update): """    """ bot.sendMessage( chat_id=update.message.chat_id, text="    .\n" "   /auth password." ) def auth(bot, update): """    ,         """ if config['telegtam']['password'] in update.message.text: if update.message.chat_id not in config['telegtam']['authenticated_users']: config['telegtam']['authenticated_users'].append(update.message.chat_id) custom_keyboard = [ ['/_', '/_'], ['/_', '/_'], ['/'] ] reply_markup = ReplyKeyboardMarkup(custom_keyboard) bot.sendMessage( chat_id=update.message.chat_id, text=" .", reply_markup=reply_markup ) else: bot.sendMessage(chat_id=update.message.chat_id, text=" .") def send_command_to_noolite(command): """   NooLite.  .   ,      . """ try: logger.info('Send command to noolite: {}'.format(command)) response = noolite_api.send_command_to_channel(command) except NooLiteConnectionTimeout as e: logger.info(e) return None, "* !*\n`{}`".format(e) except NooLiteConnectionError as e: logger.info(e) return None, "*!*\n`{}`".format(e) except NooLiteBadResponse as e: logger.info(e) return None, "*   !*\n`{}`".format(e) return response.text, None # ========================== Commands ================================ @log @auth_required def outdoor_light_on(bot, update): """  """ response, error = send_command_to_noolite({'ch': 2, 'cmd': 2}) logger.info('Send message: {}'.format(response or error)) bot.sendMessage(chat_id=update.message.chat_id, text="{}".format(response or error)) @log @auth_required def outdoor_light_off(bot, update): """  """ response, error = send_command_to_noolite({'ch': 2, 'cmd': 0}) logger.info('Send message: {}'.format(response or error)) bot.sendMessage(chat_id=update.message.chat_id, text="{}".format(response or error)) @log @auth_required def heaters_on(bot, update): """ """ response, error = send_command_to_noolite({'ch': 0, 'cmd': 2}) logger.info('Send message: {}'.format(response or error)) bot.sendMessage(chat_id=update.message.chat_id, text="{}".format(response or error)) @log @auth_required def heaters_off(bot, update): """ """ response, error = send_command_to_noolite({'ch': 0, 'cmd': 0}) logger.info('Send message: {}'.format(response or error)) bot.sendMessage(chat_id=update.message.chat_id, text="{}".format(response or error)) @log @auth_required def send_temperature(bot, update): """   """ try: sens_list = noolite_api.get_sens_data() except NooLiteConnectionTimeout as e: logger.info(e) bot.sendMessage( chat_id=update.message.chat_id, text="* !*\n`{}`".format(e), parse_mode=ParseMode.MARKDOWN ) return except NooLiteBadResponse as e: logger.info(e) bot.sendMessage( chat_id=update.message.chat_id, text="*   !*\n`{}`".format(e), parse_mode=ParseMode.MARKDOWN ) return except NooLiteConnectionError as e: logger.info(e) bot.sendMessage( chat_id=update.message.chat_id, text="*   noolite!*\n`{}`".format(e), parse_mode=ParseMode.MARKDOWN ) return if sens_list[0].temperature and sens_list[0].humidity: message = ": *{}C*\n: *{}%*".format( sens_list[0].temperature, sens_list[0].humidity ) else: message = "   : {}".format(sens_list[0].state) logger.info('Send message: {}'.format(message)) bot.sendMessage(chat_id=update.message.chat_id, text=message, parse_mode=ParseMode.MARKDOWN) @log @auth_required def send_log(bot, update): """   """ bot.sendDocument( chat_id=update.message.chat_id, document=open('/var/log/telegram_bot/err.log', 'rb') ) @log def unknown(bot, update): """ """ bot.sendMessage(chat_id=update.message.chat_id, text="    ") def power_restore(bot, job): """     """ for user_chat in config['telegtam']['authenticated_users']: bot.sendMessage(user_chat, '  ') def check_temperature(bot, job): """     E  ,    -     """ try: sens_list = noolite_api.get_sens_data() except NooLiteConnectionTimeout as e: print(e) return except NooLiteConnectionError as e: print(e) return except NooLiteBadResponse as e: print(e) return if sens_list[0].temperature and \ sens_list[0].temperature < config['noolite']['temperature_alert']: for user_chat in config['telegtam']['authenticated_users']: bot.sendMessage( chat_id=user_chat, parse_mode=ParseMode.MARKDOWN, text='*  {} : {}!*'.format( config['noolite']['temperature_alert'], sens_list[0].temperature ) ) def check_internet_connection(bot, job): """               -    telnet     .         -  Raspberry Pi """ try: requests.get('http://ya.ru') config['noolite']['internet_connection_counter'] = 0 except ConnectionError: if config['noolite']['internet_connection_counter'] == 2: tn = telnetlib.Telnet(config['router']['ip']) tn.read_until(b"login: ") tn.write(config['router']['login'].encode('ascii') + b"\n") tn.read_until(b"Password: ") tn.write(config['router']['password'].encode('ascii') + b"\n") tn.write(b"reboot\n") elif config['noolite']['internet_connection_counter'] == 4: os.system("sudo reboot") else: config['noolite']['internet_connection_counter'] += 1 dispatcher = updater.dispatcher dispatcher.add_handler(CommandHandler('start', start)) dispatcher.add_handler(CommandHandler('auth', auth)) dispatcher.add_handler(CommandHandler('', send_temperature)) dispatcher.add_handler(CommandHandler('_', heaters_on)) dispatcher.add_handler(CommandHandler('_', heaters_off)) dispatcher.add_handler(CommandHandler('_', outdoor_light_on)) dispatcher.add_handler(CommandHandler('_', outdoor_light_off)) dispatcher.add_handler(CommandHandler('log', send_log)) dispatcher.add_handler(MessageHandler([Filters.command], unknown)) job_queue.put(Job(check_internet_connection, 60*5), next_t=60*5) job_queue.put(Job(check_temperature, 60*30), next_t=60*6) job_queue.put(Job(power_restore, 60, repeat=False)) updater.start_polling(bootstrap_retries=-1)