jmbwell/rsbbs
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

prevent deleting others' messages

Browse Source
This commit is contained in:
John Burwell 2023-04-29 20:24:30 -05:00
parent 5bf004a609
commit 246edcd5f0
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
rsbbs/plugins/delete/plugin.py
View File

@ -44,11 +44,23 @@ class Plugin():
def delete(self, number) -> None: def delete(self, number) -> None:
with self.api.controller.session() as session: with self.api.controller.session() as session:
try: try:
message = session.get(Message, number) statement = sqlalchemy.delete(Message).where(
session.delete(message) sqlalchemy.and_(
Message.recipient == self.api.config.calling_station,
Message.id == number,
)).returning(Message)
result = session.execute(
statement,
execution_options={"prebuffer_rows": True})
session.commit() session.commit()
results = result.all()
count = len(results)
if count > 0:
self.api.write_output(f"Deleted message #{number}") self.api.write_output(f"Deleted message #{number}")
logging.info(f"deleted message {number}") logging.info(f"deleted message {number}")
else:
self.api.write_output("A message with that ID addressed "
"to you was not found.")
except sqlalchemy.exc.NoResultFound: except sqlalchemy.exc.NoResultFound:
self.api.write_output(f"Message not found.") self.api.write_output(f"Message not found.")
except Exception as e: except Exception as e: