mirror of
https://github.com/m-labs/artiq.git
synced 2024-12-19 08:26:30 +08:00
81 lines
2.4 KiB
Python
Executable File
81 lines
2.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
import asyncio
|
|
import struct
|
|
import logging
|
|
import re
|
|
|
|
from artiq.tools import *
|
|
from artiq.protocols.pc_rpc import Server
|
|
from artiq.protocols.logging import log_with_name
|
|
from artiq.coredevice.comm_mgmt import Request, Reply
|
|
|
|
|
|
def get_argparser():
|
|
parser = argparse.ArgumentParser(
|
|
description="ARTIQ controller for core device logs")
|
|
simple_network_args(parser, 1068)
|
|
parser.add_argument("core_addr",
|
|
help="hostname or IP address of the core device")
|
|
return parser
|
|
|
|
|
|
class PingTarget:
|
|
def ping(self):
|
|
return True
|
|
|
|
|
|
async def get_logs(host):
|
|
reader, writer = await asyncio.open_connection(host, 1380)
|
|
writer.write(b"ARTIQ management\n")
|
|
writer.write(struct.pack("B", Request.PullLog.value))
|
|
await writer.drain()
|
|
|
|
while True:
|
|
length, = struct.unpack(">l", await reader.readexactly(4))
|
|
log = await reader.readexactly(length)
|
|
|
|
for line in log.decode("utf-8").splitlines():
|
|
m = re.match(r"^\[.+?\] (TRACE|DEBUG| INFO| WARN|ERROR)\((.+?)\): (.+)$", line)
|
|
if m.group(1) == 'TRACE':
|
|
continue
|
|
elif m.group(1) == 'DEBUG':
|
|
level = logging.DEBUG
|
|
elif m.group(1) == ' INFO':
|
|
level = logging.INFO
|
|
elif m.group(1) == ' WARN':
|
|
level = logging.WARN
|
|
elif m.group(1) == 'ERROR':
|
|
level = logging.ERROR
|
|
name = 'firmware.' + m.group(2).replace('::', '.')
|
|
text = m.group(3)
|
|
log_with_name(name, level, text)
|
|
|
|
|
|
def main():
|
|
args = get_argparser().parse_args()
|
|
|
|
loop = asyncio.get_event_loop()
|
|
try:
|
|
get_logs_task = asyncio.ensure_future(get_logs(args.core_addr))
|
|
try:
|
|
server = Server({"corelog": PingTarget()}, None, True)
|
|
loop.run_until_complete(server.start(bind_address_from_args(args), args.port))
|
|
try:
|
|
multiline_log_config(logging.DEBUG)
|
|
loop.run_until_complete(server.wait_terminate())
|
|
finally:
|
|
loop.run_until_complete(server.stop())
|
|
finally:
|
|
get_logs_task.cancel()
|
|
try:
|
|
loop.run_until_complete(get_logs_task)
|
|
except asyncio.CancelledError:
|
|
pass
|
|
finally:
|
|
loop.close()
|
|
|
|
if __name__ == "__main__":
|
|
main()
|