from django.core.management.base import NoArgsCommand
class Command(NoArgsCommand):
help = """
Removes all symlinks in MEDIA_ROOT and then scans all installed applications for a media folder to symlink to MEDIA_ROOT.
If installed app has a media folder, it first attempts to symlink the contents
ie: app/media/app_name -> MEDIA_ROOT/app_name
If the symlink name already exists, it assumes the media directory is not subfoldered and attempts:
ie: app/media -> MEDIA_ROOT/app_name"""
def handle_noargs(self, **options):
from django.conf import settings
import os
media_path = settings.MEDIA_ROOT
for d in os.listdir(media_path):
path = os.path.join(media_path,d)
if os.path.islink(path):
os.remove(os.path.join(path))
print " - removed %s" % path
for app in settings.INSTALLED_APPS:
app_path = __import__(app, {}, {}, [app.split('.')[-1]]).__path__
if 'media' in os.listdir(app_path[0]) and os.path.isdir(os.path.join(app_path[0],'media')):
app_media = os.path.join(app_path[0],'media')
try:
for node in os.listdir(app_media):
os.symlink(os.path.join(app_media,node), os.path.join(media_path,node))
print " + added %s as %s" % (os.path.join(app_media,node), os.path.join(media_path,node))
except OSError, e:
try:
os.symlink(app_media, os.path.join(media_path,app.split('.')[-1]))
print " + added %s as %s" % (app_media, os.path.join(media_path,app.split('.')[-1]))
except OSError,e:
print "ERROR: cannot add media from %s" % app
Comments