You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
110 lines
3.1 KiB
110 lines
3.1 KiB
#!/usr/bin/python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import datetime
|
|
import os.path
|
|
import asyncio
|
|
import logging
|
|
import argparse
|
|
import websockets
|
|
import socketserver
|
|
import configparser
|
|
import os
|
|
from collections import deque
|
|
from urllib.parse import urlparse, parse_qs
|
|
from ansi2html import Ansi2HTMLConverter
|
|
|
|
current_dir = os.getcwd()
|
|
config = configparser.ConfigParser()
|
|
config.read(current_dir + '/logtailer.ini')
|
|
|
|
|
|
|
|
# init
|
|
logging.basicConfig(format='%(asctime)s %(levelname)s %(message)s', level=logging.INFO)
|
|
conv = Ansi2HTMLConverter(inline=True)
|
|
|
|
@asyncio.coroutine
|
|
def view_log(websocket, path):
|
|
global config
|
|
logging.info('Connected, remote={}, path={}'.format(websocket.remote_address, path))
|
|
|
|
try:
|
|
try:
|
|
parse_result = urlparse(path)
|
|
except Exception:
|
|
raise ValueError('Fail to parse URL')
|
|
|
|
NUM_LINES = int(config['MMDVMHost']['Num_Lines'])
|
|
path = os.path.abspath(parse_result.path)
|
|
now = datetime.datetime.now()
|
|
year = str(now.year)
|
|
month = str(now.month)
|
|
if len(month) == 1:
|
|
month = "0" + month
|
|
day = str(now.day)
|
|
if len(day) == 1:
|
|
day = "0" + day
|
|
|
|
file_path = ""
|
|
if path == "/MMDVM":
|
|
file_path = config['MMDVMHost']['Logdir']+config['MMDVMHost']['Prefix']+"-"+year+"-"+month+"-"+day+".log"
|
|
elif path == "/DAPNET":
|
|
file_path = config['DAPNETGateway']['Logdir']+config['DAPNETGateway']['Prefix']+"-"+year+"-"+month+"-"+day+".log"
|
|
NUM_LINES = 0
|
|
logging.info(file_path)
|
|
if not os.path.isfile(file_path):
|
|
raise ValueError('Not found')
|
|
|
|
with open(file_path, newline = '\n') as f:
|
|
|
|
content = ''.join(deque(f, NUM_LINES))
|
|
content = conv.convert(content, full=False)
|
|
lines = content.split("\n")
|
|
for line in lines:
|
|
yield from websocket.send(line)
|
|
|
|
while True:
|
|
content = f.read()
|
|
if content:
|
|
content = conv.convert(content, full=False)
|
|
yield from websocket.send(content)
|
|
else:
|
|
yield from asyncio.sleep(1)
|
|
|
|
except ValueError as e:
|
|
try:
|
|
yield from websocket.send('<font color="red"><strong>{}</strong></font>'.format(e))
|
|
yield from websocket.close()
|
|
except Exception:
|
|
pass
|
|
|
|
log_close(websocket, path, e)
|
|
|
|
except Exception as e:
|
|
log_close(websocket, path, e)
|
|
|
|
else:
|
|
log_close(websocket, path)
|
|
|
|
|
|
def log_close(websocket, path, exception=None):
|
|
message = 'Closed, remote={}, path={}'.format(websocket.remote_address, path)
|
|
if exception is not None:
|
|
message += ', exception={}'.format(exception)
|
|
logging.info(message)
|
|
|
|
|
|
def websocketserver():
|
|
start_server = websockets.serve(view_log, config['DEFAULT']['Host'], config['DEFAULT']['Port'])
|
|
asyncio.get_event_loop().run_until_complete(start_server)
|
|
asyncio.get_event_loop().run_forever()
|
|
|
|
|
|
def main():
|
|
websocketserver()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|