diff --git a/src/snek/dump.py b/src/snek/dump.py index cf3d417..1b7eb6b 100644 --- a/src/snek/dump.py +++ b/src/snek/dump.py @@ -12,20 +12,18 @@ async def fix_message(message): ) user = await app.services.user.get(uid=message['user_uid']) message['user'] = user and user['username'] or None - return message + return (message['user'] or '') + ': ' + (message['text'] or '') async def dump_public_channels(): - result = {'channels':{}} + result = [] for channel in app.db['channel'].find(is_private=False,is_listed=True,tag='public'): print(f"Dumping channel: {channel['label']}.") - result['channels'][channel['label']] = dict(channel) - result['channels'][channel['label']]['messages'] = [await fix_message(record) for record in app.db['channel_message'].find(channel_uid=channel['uid'],order_by='created_at')] + result += [await fix_message(record) for record in app.db['channel_message'].find(channel_uid=channel['uid'],order_by='created_at')] print("Dump succesfull!") print("Converting to json.") - data = json.dumps(result, indent=4,default=str) print("Converting succesful, now writing to dump.json") - with open("dump.json","w") as f: - f.write(data) + with open("dump.txt","w") as f: + f.write('\n\n'.join(result)) print("Dump written to dump.json")