From 5b42882f7092a32696f0ec7879f997e2a91c79f0 Mon Sep 17 00:00:00 2001
From: Ethan Dalool <git@voussoir.net>
Date: Sat, 9 Jan 2021 15:45:06 -0800
Subject: [PATCH] Add User.delete.

---
 etiquette/exceptions.py |  3 +++
 etiquette/objects.py    | 56 +++++++++++++++++++++++++++++++++++++++++
 2 files changed, 59 insertions(+)

diff --git a/etiquette/exceptions.py b/etiquette/exceptions.py
index 0597a15..474827e 100644
--- a/etiquette/exceptions.py
+++ b/etiquette/exceptions.py
@@ -130,6 +130,9 @@ class TagTooShort(EtiquetteException):
 class AlreadySignedIn(EtiquetteException):
     error_message = 'You\'re already signed in.'
 
+class CantDeleteUser(EtiquetteException):
+    error_message = '{} can\'t be deleted because they still have possessions.'
+
 class InvalidPassword(EtiquetteException):
     error_message = 'Password is invalid.'
 
diff --git a/etiquette/objects.py b/etiquette/objects.py
index c1857c6..b64c778 100644
--- a/etiquette/objects.py
+++ b/etiquette/objects.py
@@ -1693,6 +1693,42 @@ class User(ObjectBase):
 
         return display_name
 
+    def _uncache(self):
+        self.photodb.caches['user'].remove(self.id)
+
+    @decorators.required_feature('user.edit')
+    @decorators.transaction
+    def delete(self, *, disown_authored_things):
+        '''
+        If disown_authored_things is True, then all of this user's albums,
+        bookmarks, photos, and tags will have their author_id set to None.
+
+        If disown_authored_things is False, and the user has any belongings,
+        exceptions.CantDeleteUser is raised.
+
+        You should delete those objects first. Since each object type has
+        different options while deleting, that functionality is not provided
+        here.
+        '''
+        if disown_authored_things:
+            pairs = {'author_id': (self.id, None)}
+            self.photodb.sql_update(table='albums', pairs=pairs, where_key='author_id')
+            self.photodb.sql_update(table='bookmarks', pairs=pairs, where_key='author_id')
+            self.photodb.sql_update(table='photos', pairs=pairs, where_key='author_id')
+            self.photodb.sql_update(table='tags', pairs=pairs, where_key='author_id')
+        else:
+            fail = (
+                self.has_any_albums() or
+                self.has_any_bookmarks() or
+                self.has_any_photos() or
+                self.has_any_tags()
+            )
+            if fail:
+                raise exceptions.CantDeleteUser(self)
+        self.photodb.sql_delete(table='users', pairs={'id': self.id})
+        self._uncache()
+        self.deleted = True
+
     @property
     def display_name(self):
         if self._display_name is None:
@@ -1736,6 +1772,26 @@ class User(ObjectBase):
             [self.id]
         )
 
+    def has_any_albums(self):
+        query = f'SELECT 1 FROM albums WHERE author_id == ? LIMIT 1'
+        row = self.photodb.sql_select_one(query, [self.id])
+        return row is not None
+
+    def has_any_bookmarks(self):
+        query = f'SELECT 1 FROM bookmarks WHERE author_id == ? LIMIT 1'
+        row = self.photodb.sql_select_one(query, [self.id])
+        return row is not None
+
+    def has_any_photos(self):
+        query = f'SELECT 1 FROM photos WHERE author_id == ? LIMIT 1'
+        row = self.photodb.sql_select_one(query, [self.id])
+        return row is not None
+
+    def has_any_tags(self):
+        query = f'SELECT 1 FROM tags WHERE author_id == ? LIMIT 1'
+        row = self.photodb.sql_select_one(query, [self.id])
+        return row is not None
+
     def jsonify(self):
         j = {
             'type': 'user',