Add a small poezio_logs script to parse logfiles

2 files changed, 102 insertions, 1 deletions

diff --git a/scripts/poezio_logs b/scripts/poezio_logs
new file mode 100755
index 00000000..59d397da
--- /dev/null
+++ b/scripts/poezio_logs
@@ -0,0 +1,101 @@
+#!/usr/bin/env python3
+"""
+A simple script to parse and output logs from a poezio logfile
+"""
+
+from poezio.logger import LogInfo, LogMessage, parse_message_line
+from functools import singledispatch
+from poezio import poopt
+import argparse
+import datetime
+import sys
+
+INFO_COLOR = '\033[35;2m'
+NICK_COLOR = '\033[36;2m'
+NO_COLOR = '\033[0m'
+TIME_COLOR = '\033[33;2m'
+
+SHOW_INFO = True
+SHOW_TIME = True
+
+@singledispatch
+def print_log(log_object, additional_lines=None):
+    time = log_object.time.strftime('%Y-%m-%d %H:%M:%S')
+
+    nick = log_object.nick
+    offset = ((poopt.wcswidth(time) +1) if SHOW_TIME else 0) + 2 + poopt.wcswidth(nick)
+    pad = ' ' * offset
+
+    if additional_lines:
+        more = ''.join([(pad + line) for line in additional_lines])
+    else:
+        more = ''
+
+    if SHOW_TIME:
+        print(('%s%s%s %s%s%s> %s\n' % (TIME_COLOR, time, NO_COLOR, NICK_COLOR, nick, NO_COLOR, log_object.text)) + more, end='')
+    else:
+        print(('%s%s%s> %s\n' % (NICK_COLOR, nick, NO_COLOR, log_object.text)) + more, end='')
+
+@print_log.register(type(None))
+def _(log_object, additional_lines=None):
+    return
+
+@print_log.register(LogInfo)
+def _(log_object, additional_lines=None):
+    if not SHOW_INFO: return
+    time = log_object.time.strftime('%Y-%m-%d %H:%M:%S') + ' '
+
+    offset = (poopt.wcswidth(time) + 1) if SHOW_TIME else 0
+    pad = ' ' * offset
+
+    if additional_lines:
+        more = ''.join([(pad + line) for line in additional_lines])
+    else:
+        more = ''
+
+    if SHOW_TIME:
+        print(('%s%s%s %s%s\n' % (TIME_COLOR, time, NO_COLOR, INFO_COLOR, log_object.text)) + more, end='')
+    else:
+        print(('%s%s\n' % (INFO_COLOR, log_object.text)) + more, end='')
+
+def parse_messages(fd):
+    in_text = False
+    more_lines = []
+    current_log = None
+    for line in fd:
+        if in_text and not line.startswith(' '):
+            print_log(current_log, more_lines)
+            more_lines = []
+            in_text = False
+            current_log = None
+        elif in_text:
+            more_lines.append(line[1:])
+            continue
+        current_log = parse_message_line(line.rstrip())
+        in_text = True
+    print_log(current_log, more_lines)
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser('poezio_logs', description="""
+Show the logs stored in poezio format in a more human-readable way.
+""")
+    parser.add_argument('-i', '--hide-info', dest='hide_info',
+                        action='store_true', default=False,
+                        help='Hide info lines')
+    parser.add_argument('-t', '--hide-time', dest='hide_time',
+                        action='store_true', default=False,
+                        help='Hide timestamps')
+    parser.add_argument('-c', '--no-color', dest='no_color',
+                        action='store_true', default=False,
+                        help='Remove color')
+    parser.add_argument('log_file', type=argparse.FileType('r'))
+    result = parser.parse_args()
+    SHOW_INFO = not result.hide_info
+    SHOW_TIME = not result.hide_time
+    if result.no_color:
+        INFO_COLOR = ''
+        NICK_COLOR = ''
+        NO_COLOR = ''
+        TIME_COLOR = ''
+
+    parse_messages(result.log_file)
diff --git a/setup.py b/setup.py
index fb491bbd..303c789b 100755
--- a/setup.py
+++ b/setup.py
@@ -95,7 +95,7 @@ setup(name="poezio",
                    'poezio_plugins': 'plugins',
                    'poezio_themes': 'data/themes'},
       package_data={'poezio': ['default_config.cfg']},
-      scripts=['scripts/poezio_gpg_export'],
+      scripts=['scripts/poezio_gpg_export', 'scripts/poezio_logs'],
       entry_points={'console_scripts': ['poezio = poezio.__main__:run']},
       data_files=([('share/man/man1/', ['data/poezio.1']),
                    ('share/poezio/', ['README.rst', 'COPYING', 'CHANGELOG'])]