51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
import argparse
|
|
import logging
|
|
from pathlib import Path
|
|
import logging
|
|
|
|
from . import (format_messages, merge_adjacent_messages,
|
|
parse_messages_in_chat_folder, Message, filter_useless_messages)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
def group_messages_by_period(messages: list[Message]) -> dict[str, list[Message]]:
|
|
by_period: dict[str,list[Message]] = {}
|
|
for msg in messages:
|
|
period_key = f'{msg.sent_at.year}-{msg.sent_at.month:02}'
|
|
by_period.setdefault(period_key, []).append(msg)
|
|
del msg
|
|
return by_period
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('path', type=Path)
|
|
parser.add_argument('--output', type=Path)
|
|
return parser.parse_args()
|
|
|
|
|
|
def main():
|
|
logging.basicConfig()
|
|
logging.getLogger().setLevel('INFO')
|
|
args = parse_args()
|
|
|
|
server = args.path.parent.name
|
|
receipient = args.path.name
|
|
|
|
all_messages = parse_messages_in_chat_folder(args.path)
|
|
all_messages = filter_useless_messages(all_messages)
|
|
all_messages = merge_adjacent_messages(all_messages)
|
|
|
|
messages_by_period = group_messages_by_period(all_messages)
|
|
|
|
for period_key, messages in messages_by_period.items():
|
|
output_file = args.output / f'{server} - {receipient} - {period_key}.md'
|
|
logger.info("Writing % 5d messages to %s", len(messages), output_file)
|
|
with open(output_file, 'w') as f:
|
|
f.write(format_messages(messages, title = f'{server} - {receipient} - {period_key}'))
|
|
|
|
del period_key, messages, output_file
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|