Ticket #5308: 0001-Re-5308-added-delete_user-hook.patch

mediagoblin/db/models.py

@@ -43 +43 @@
 from mediagoblin.tools.files import delete_media_files
 from mediagoblin.tools.common import import_component
 from mediagoblin.tools.routing import extract_url_arguments
+from mediagoblin.tools.pluginapi import hook_runall
 
 import six
 from six.moves.urllib.parse import urljoin
@@ -283 +284 @@
             # all related files/Comments
             media.delete(del_orphan_tags=False, commit=False)
 
+        #allow plugin-specific deletion
+        _log.debug("Running all 'delete_user' hooks")
+        hook_runall("delete_user",self.id)
+
         # Delete now unused tags
         # TODO: import here due to cyclic imports!!! This cries for refactoring
         from mediagoblin.db.util import clean_orphan_tags

mediagoblin/media_types/blog/__init__.py

@@ -23 +23 @@
 
 from mediagoblin.tools import pluginapi
 
+from mediagoblin.db.base import Session
+
 PLUGIN_DIR = os.path.dirname(__file__)
 MEDIA_TYPE = 'mediagoblin.media_types.blogpost'
 
@@ -111 +113 @@
         context['blogs'] = None
     return context
 
+def delete_user_hook(user_id):
+    _log.debug("Running blog deletion hook on user with id {0}".format(user_id))
+    blogs = Session.query(Blog).filter(Blog.author == user_id)
+    for blog in blogs:
+        #SQLAlchemy will automatically delete blogposts that are connected to the deleted blogs
+        blog.delete()
+
 
 hooks = {
     'setup': setup_plugin,
     ('media_manager', MEDIA_TYPE): lambda: BlogPostMediaManager,
     # Inject blog context on user profile page
     ("mediagoblin.user_pages.user_home",
-     "mediagoblin/user_pages/user.html"): add_to_user_home_context
+     "mediagoblin/user_pages/user.html"): add_to_user_home_context,
+    "delete_user": delete_user_hook
 }