src/mtxbot.py

   1 #!/usr/bin/env python3
   2
   3 import asyncio
   4 import aiofiles
   5 import json
   6 import logging
   7 import nio
   8 import os
   9 import signal
  10 import stat
  11 import time
  12
  13 import sys
  14 assert sys.version_info >= (3, 5)
  15
  16 from presence import Presence
  17
  18 def handleInterrupt(sig, frame):
  19     global stop
  20     if os.getpid() == mainPid:
  21         logging.info( "Got stop signal." )
  22         stop = True
  23         for presence in presences:
  24             presence.stop()
  25
  26 async def main():
  27     client = nio.AsyncClient(config['server']['url'], config['server']['user'])
  28
  29     try:
  30         await client.login(config['server']['password'])
  31
  32         presence_tasks = []
  33         for presence in presences:
  34             presence_tasks.append( asyncio.create_task(presence.run(client)) )
  35
  36         done, pending = await asyncio.wait( presence_tasks,
  37                                             return_when = asyncio.FIRST_EXCEPTION )
  38
  39         for task in done:
  40             try:
  41                 task.result()
  42             except Exception as e:
  43                 logging.error( "Error running task: %s" % str(e) )
  44
  45     finally:
  46         await client.close()
  47
  48 logging.basicConfig(format='[%(asctime)s] %(levelname)s: %(message)s',
  49                     level=logging.INFO,
  50                     datefmt='%m/%d/%Y %H:%M:%S')
  51
  52 mainPid = os.getpid()
  53 signal.signal(signal.SIGINT, handleInterrupt)
  54
  55 stop = False
  56
  57 if len(sys.argv) != 2:
  58     print("Usage: %s <config json>" % sys.argv[0])
  59     sys.exit(1)
  60
  61 with open(sys.argv[1]) as configFile:
  62     config = json.load(configFile)
  63
  64 fifo_dir = os.getenv('MTXBOT_FIFO_DIR', '/run/mtxbot')
  65
  66 while not stop:
  67     presences = []
  68     for entry in os.listdir(fifo_dir):
  69         fullpath = "%s/%s" % (fifo_dir, entry)
  70         if stat.S_ISFIFO(os.stat(fullpath).st_mode):
  71             logging.info("Creating presence for %s..." % entry)
  72             presences.append( Presence(entry, fullpath) )
  73     if len(presences) == 0:
  74         logging.error("No fifos could be found!")
  75         break
  76     asyncio.run(main())
  77     if not stop:
  78         logging.warning("Main loop exited!")
  79         logging.info("Restarting after grace period...")
  80         time.sleep(3.0)