Added mgmt command to add & remove all mtimes from the cache (faster for large deployments)

compressor/management/commands/mtime_cache.py

@@ -0,0 +1,41 @@
+from optparse import make_option
+import os
+
+from django.core.cache import cache
+from django.core.management.base import NoArgsCommand, CommandError
+
+from compressor.conf import settings
+from compressor.utils import get_mtime
+
+class Command(NoArgsCommand):
+    option_list = NoArgsCommand.option_list + (
+        make_option('--ignore-vcs', dest='ignore_vcs', action='store_true', help="Don't traverse into VCS directories"),
+        make_option('--follow-links', dest='follow_links', action='store_true', help='Follow symlinks when traversing the `COMPRESS_ROOT` (default `MEDIA_ROOT`). '
+        'Be aware that setting follow-links can lead to infinite recursion if a link points to a parent directory of itself.'),
+        make_option('--clean', dest='clean', action='store_true', help='Remove all items'),
+        make_option('--add', dest='add', action='store_true', help='Add all items'),
+    )
+    help = 'Add or remove all mtimes from the cache'
+
+    def handle_noargs(self, **options):
+        if (options['add'] and options['clean']) or (not options['add'] and not options['clean']):
+            raise CommandError, 'Please specify either "--add" or "--clean"'
+
+        for root,dirs,files in os.walk(settings.MEDIA_ROOT, followlinks=options['follow_links']):
+            if options['ignore_vcs']:
+                # no VCS (CVS, SVN, git, hg) directories
+                if 'CVS' in dirs:
+                    dirs.remove('CVS')
+                if '.svn' in dirs:
+                    dirs.remove('.svn')
+                if '.git' in dirs:
+                    dirs.remove('.git')
+                if '.hg' in dirs:
+                    dirs.remove('.hg')
+            for filename in files:
+                filename = os.path.join(root,filename)
+                key = "django_compressor.mtime.%s" % filename
+                cache.set(key, None, 10)
+                if options['add']:
+                    get_mtime(filename)
+