]> git.treefish.org Git - mtxbot.git/blob - src/mtxbot.py
fix fifo check
[mtxbot.git] / 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)