From a006c75df9ddbdf69a3c3ba70a29a1bf78ad826b Mon Sep 17 00:00:00 2001
From: meltland <meltland@icloud.com>
Date: Sat, 22 Mar 2025 15:57:13 -0400
Subject: delete/edit

---
 db.py | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

(limited to 'db.py')

diff --git a/db.py b/db.py
index 28f3ba8..01718be 100644
--- a/db.py
+++ b/db.py
@@ -159,6 +159,37 @@ class posts:
             return "fail"
         return True
 
+    def remove(post_id):
+        post = postsd.find_one({"_id": post_id})
+        if not post:
+            return "notExists"
+        try:
+            postsd.delete_one({"_id": post_id})
+            postsd.update_many(
+                {"replies": {"$elemMatch": {"_id": post_id}}},
+                {"$set": {"replies.$.content": "post deleted"}}
+            )
+        except:
+            return "fail"
+        return True
+    
+    def edit(post_id, content):
+        post = postsd.find_one({"_id": post_id})
+        if not post:
+            return "notExists"
+        try:
+            postsd.update_one(
+                {"_id": post_id},
+                {"$set": {"content": content}}
+            )
+            postsd.update_many(
+                {"replies": {"$elemMatch": {"_id": post_id}}},
+                {"$set": {"replies.$.content": content}}
+            )
+        except:
+            return "fail"
+        return True
+
 class inbox:
     def get_recent(amount=75):
         posts = list(inboxd.find().sort("created", -1).limit(amount))
-- 
cgit 1.4.1-2-gfad0