# HG changeset patch
# User Brian Neal <bgneal@gmail.com>
# Date 1369513062 18000
# Node ID 644f69e1c1e1e9a6c94c6ff55a31f822b991fc54
# Parent  264b08bce8b8d16c88a055854f49c256249bef44
For issue #42, create avatar cleanup script.

diff -r 264b08bce8b8 -r 644f69e1c1e1 bio/management/commands/avatar_cleanup.py
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/bio/management/commands/avatar_cleanup.py	Sat May 25 15:17:42 2013 -0500
@@ -0,0 +1,80 @@
+"""avatar_cleanup - a management command to remove orphaned avatar files."""
+
+from optparse import make_option
+import os
+
+from django.core.management.base import NoArgsCommand
+from django.conf import settings
+
+from bio.models import UserProfile
+
+
+class Command(NoArgsCommand):
+    help = "Finds and optionally deletes orphan avatar files"
+    option_list = NoArgsCommand.option_list + (
+            make_option('--delete',
+                action='store_true',
+                dest='delete',
+                default=False,
+                help='Delete orphan files'),
+    )
+
+    def handle_noargs(self, **options):
+        """Find and optionally delete orphan avatar files."""
+
+        delete = options.get('delete', False)
+
+        orphans = set()
+        missing = []
+
+        avatar_dir = os.path.join(settings.MEDIA_ROOT, settings.AVATAR_DIR)
+        default_avatar = os.path.join(avatar_dir,
+                settings.AVATAR_DEFAULT_URL.split('/')[-1])
+
+        # find the set of all files in the downloads area
+        for root, dirs, files in os.walk(avatar_dir, followlinks=True):
+            for name in files:
+                orphans.add(unicode(os.path.join(root, name), 'utf-8'))
+
+        try:
+            orphans.remove(default_avatar)
+        except KeyError:
+            self.stderr.write("Default avatar file missing!  - {}\n".format(
+                default_avatar))
+
+        # examine the UserProfile objects
+        for profile in UserProfile.objects.iterator():
+            if profile.avatar:
+                try:
+                    orphans.remove(profile.avatar.path)
+                except KeyError:
+                    missing.append(profile)
+
+        if orphans and delete:
+            for path in orphans:
+                self.stdout.write("Deleting: {}\n".format(path))
+                os.remove(path)
+        elif orphans:
+            self.stdout.write("Orphan files:\n")
+            for orphan in orphans:
+                self.stdout.write("{}\n".format(orphan))
+
+        if missing:
+            self.stdout.write("Users with missing avatar files:\n")
+            for profile in missing:
+                self.stdout.write("{}\n".format(profile.user.username))
+
+        empty_dirs = []
+        # check for empty directories after deletions
+        for root, dirs, files in os.walk(avatar_dir, followlinks=True):
+            if not len(dirs) and not len(files):
+                empty_dirs.append(root)
+
+        if empty_dirs and delete:
+            for path in empty_dirs:
+                self.stdout.write("Deleting empty dir: {}\n".format(path))
+                os.removedirs(path)
+        elif empty_dirs:
+            self.stdout.write("Empty directories:\n")
+            for d in empty_dirs:
+                self.stdout.write("{}\n".format(d))