matrix_fridge_bot/bot.py

   1 '''
   2 Prefix: .fridge
   3
   4 Synax:
   5 .fridge COMMAND ARGS
   6
   7 Commands:
   8 add - add an item
   9 list - print the list of fresh items
  10 remove - remove an item
  11 to.toss - print the list of items to toss
  12 filter TERM - list all items that include TERM
  13 '''
  14
  15 from datetime import datetime as dt
  16 import logging
  17 import os
  18
  19 import simplematrixbotlib as botlib
  20
  21 from .db import Data
  22
  23 logger = logging.getLogger(__name__)
  24 logging.getLogger('nio').setLevel(logging.WARNING)
  25
  26 MODULE='fridge'
  27
  28 class FridgeBot():
  29     def __init__(self, matrix_server,
  30                  bot_name,
  31                  bot_pwd,
  32                  db_file=None,
  33                  bot_prefix='.fridge'):
  34
  35         logging.basicConfig(filename=f'matrix-{MODULE}-bot.log', encoding='utf-8', level=logging.DEBUG)
  36
  37
  38         creds = botlib.Creds(matrix_server, bot_name, bot_pwd)
  39         self.bot = botlib.Bot(creds)
  40         self.prefix = bot_prefix
  41         self.data = Data(db_file=db_file)
  42
  43     def run(self):
  44         self.bot.run()
  45
  46     async def handle_message(self, room, message):
  47         match = botlib.MessageMatch(room, message, self.bot, self.prefix)
  48
  49         if not match.is_not_from_this_bot() and match.prefix():
  50             return
  51
  52         if match.prefix():
  53             try:
  54                 result_lines = await self._handle_match(match)
  55
  56                 message_lines = []
  57                 for line in result_lines:
  58                     message_lines.append(str(line))
  59
  60                 message = '\n'.join(message_lines)
  61
  62                 await self.bot.api.send_text_message(
  63                     room.room_id, message)
  64
  65             except Exception as ex:
  66                 line = f'Ye flask must be load bearing, or something... {str(ex)}'
  67                 await self.bot.api.send_text_message(
  68                     room.room_id, str(line))
  69
  70
  71     async def _handle_match(self, match):
  72
  73         if match.command('list'):
  74             results = ['## Listing fresh items...']
  75             for item in self.data.list():
  76                 results.append(item)
  77             results.append('-- End of List --')
  78             return results
  79
  80
  81         if match.command('toss'):
  82             results = ['## Listing items to toss...']
  83             for item in self.data.to_toss():
  84                 results.append(item)
  85             results.append('-- End of List --')
  86             return results
  87
  88
  89         if match.command('filter'):
  90             term = ' '.join(match.args()[1:])
  91             results = [f'## List matching {term}']
  92             for item in self.data.filter(term):
  93                 results.append(item)
  94             results.append('-- End of List --')
  95             return results
  96
  97         if match.command('setup'):
  98             results = []
  99             results.append('Doing fridge setup...')
 100             result = self.data.setup()
 101             results.append(str(result))
 102             return results
 103
 104         if match.command('add'):
 105             new_fridge = ' '.join(match.args()[1:])
 106             result = self.data.add(new_fridge)
 107             return [result]
 108
 109
 110         if match.command('remove'):
 111             fridge_id = match.args()[1]
 112             result = self.data.remove(fridge_id)
 113             return [result]
 114
 115         return [__doc__]
 116