Reading messages from DB + clicking to place new messages

1 files changed, 14 insertions, 5 deletions

diff --git a/api/api.py b/api/api.py
index 64d9fe5..1006a2a 100644
--- a/api/api.py
+++ b/api/api.py
@@ -1,8 +1,17 @@
-import time
-from flask import Flask
+from flask import Flask, config
+from dotenv import dotenv_values
+from pymongo import MongoClient
+from bson.json_util import dumps
+
+config = dotenv_values(".env")
 
 app = Flask(__name__)
 
-@app.route('/api/time')
-def get_current_time():
-    return {'time': time.time()}
+app.mongoclient = MongoClient(config["DB_URI"])
+app.db = app.mongoclient[config["DB_NAME"]]
+print("Connected to MongoDB database")
+
+@app.route('/api/message')
+def get_messages():
+    messages = dumps(list(app.db["message"].find(limit=100)))
+    return messages