forked from 5l1v3r1/fastocloud_admin
-
Notifications
You must be signed in to change notification settings - Fork 0
/
server.py
executable file
·40 lines (27 loc) · 1.13 KB
/
server.py
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
#!/usr/bin/env python3
import logging
import argparse
import gevent
from gevent.pywsgi import WSGIServer
from geventwebsocket.handler import WebSocketHandler
from app import app, servers_manager
PROJECT_NAME = 'fastocloud_iptv_admin'
LOGS_PATH = PROJECT_NAME + '.log'
def servers_refresh():
servers_manager.refresh()
def main():
parser = argparse.ArgumentParser(prog=PROJECT_NAME, usage='%(prog)s [options]')
parser.add_argument('--logs_path', help='logs path (default: {0})'.format(LOGS_PATH), default=LOGS_PATH)
argv = parser.parse_args()
logging.basicConfig(filename=argv.logs_path, level=logging.DEBUG,
format='%(asctime)s.%(msecs)03d [%(levelname)s] %(message)s', datefmt='%H:%M:%S')
http_server = WSGIServer((servers_manager.host, servers_manager.port), app, handler_class=WebSocketHandler)
srv_greenlet = gevent.spawn(http_server.serve_forever)
alarm_greenlet = gevent.spawn(servers_refresh)
try:
gevent.joinall([srv_greenlet, alarm_greenlet])
except KeyboardInterrupt:
servers_manager.stop()
http_server.stop()
if __name__ == '__main__':
main()