1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
import sys
import asyncio
import argparse
import pdb
import contextlib
import yaml
import logging.config
import atexit
import pkg_resources
import hatter.json_validator
from hatter.backend import Backend
from hatter.server import create_web_server
def main():
args = _create_parser().parse_args()
with open(args.conf, encoding='utf-8') as conf_file:
conf = yaml.safe_load(conf_file)
hatter.json_validator.validate(conf, 'hatter://server.yaml#')
if 'log' in conf:
logging.config.dictConfig(conf['log'])
if args.web_path:
web_path = args.web_path
else:
atexit.register(pkg_resources.cleanup_resources)
web_path = pkg_resources.resource_filename('hatter', 'web')
_run_until_complete_without_interrupt(async_main(conf, web_path))
async def async_main(conf, web_path):
backend = None
web_server = None
try:
backend = Backend(conf.get('db_path', 'hatter.db'))
web_server = await create_web_server(
backend, conf.get('host', '0.0.0.0'), conf.get('port', 24000),
conf.get('webhook_path', '/webhook'), web_path)
await asyncio.Future()
except asyncio.CancelledError:
pass
except Exception as e:
pdb.set_trace()
raise
finally:
if web_server:
await web_server.async_close()
if backend:
await backend.async_close()
await asyncio.sleep(0.5)
def _create_parser():
parser = argparse.ArgumentParser(prog='hatter')
parser.add_argument(
'--web-path', default=None, metavar='path', dest='web_path',
help="web ui directory path")
named_arguments = parser.add_argument_group('required named arguments')
named_arguments.add_argument(
'-c', '--conf', required=True, metavar='path', dest='conf',
help='configuration path')
return parser
def _run_until_complete_without_interrupt(future):
async def ping_loop():
with contextlib.suppress(asyncio.CancelledError):
while True:
await asyncio.sleep(1)
task = asyncio.ensure_future(future)
if sys.platform == 'win32':
ping_loop_task = asyncio.ensure_future(ping_loop())
with contextlib.suppress(KeyboardInterrupt):
asyncio.get_event_loop().run_until_complete(task)
asyncio.get_event_loop().call_soon(task.cancel)
if sys.platform == 'win32':
asyncio.get_event_loop().call_soon(ping_loop_task.cancel)
while not task.done():
with contextlib.suppress(KeyboardInterrupt):
asyncio.get_event_loop().run_until_complete(task)
if sys.platform == 'win32':
while not ping_loop_task.done():
with contextlib.suppress(KeyboardInterrupt):
asyncio.get_event_loop().run_until_complete(ping_loop_task)
return task.result()
if __name__ == '__main__':
sys.exit(main())
|
