roulette-bot/api.py
2024-05-01 16:19:28 +03:00

25 lines
695 B
Python

from bson import ObjectId
from flask import Flask, request
from tools.mongo import mongo
app = Flask("roulette")
@app.route('/dialogs')
def main():
html = "<html><head></head><body>"
for d in mongo.dialogs_collection.find({}).sort([('started_at', -1)]):
html += f'<a href="/dialog?id={d["_id"]}">{d["_id"]}</a><br>'
html += "</body></html>"
return html
@app.route('/dialog')
def dialog():
html = "<html><head></head><body>"
for message in mongo.messages_collection.find({"dialog_id": ObjectId(request.args.get('dialog_id'))}).sort([('sent_at', 1)]):
html += f'{message["sender"]}: {message["text"]}<br>'
html += "</body></html>"
return html