This repository has been archived by the owner on Jan 10, 2019. It is now read-only.
/
symlinkmedia.py
53 lines (44 loc) · 2.33 KB
/
symlinkmedia.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
from django.core.management.base import NoArgsCommand
from django.db.models import get_apps
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
print "creating symlinks for app media under %s" % media_path
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
apps = get_apps()
for app in apps:
app_file = os.path.abspath(app.__file__)
if os.path.splitext(app_file)[0].endswith('/__init__'):
# models are an folder, go one level up
app_file = os.path.dirname(app_file)
app_path = os.path.dirname(app_file)
if 'media' in os.listdir(app_path) and os.path.isdir(os.path.join(app_path,'media')):
module = app.__name__
app_name = module.split('.')[-2]
app_media = os.path.join(app_path, "media", app_name)
if not os.path.isdir(app_media):
app_media = os.path.join(app_path, "media")
try:
os.symlink(app_media, os.path.join(media_path, app_name))
print " + added %s as %s" % (os.path.join(app_media), os.path.join(media_path, app_name))
except OSError, e:
if e.errno == 17:
pass
print " o skipping %s" % app_media
else:
raise
# 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]))