| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364 |
- # -*- coding: utf-8 -*-
- import os
- import sys
- import shutil
- from django.core.management.base import AppCommand
- from django.core.management.color import color_style
- from django_extensions.management.utils import _make_writeable, signalcommand
- class Command(AppCommand):
- help = "Creates a Django jobs command directory structure for the given app name in the current directory."
- requires_system_checks = False
- # Can't import settings during this command, because they haven't
- # necessarily been created.
- can_import_settings = True
- @signalcommand
- def handle_app_config(self, app, **options):
- copy_template('jobs_template', app.path, **options)
- def copy_template(template_name, copy_to, **options):
- """Copy the specified template directory to the copy_to location"""
- import django_extensions
- style = color_style()
- ERROR = getattr(style, 'ERROR', lambda x: x)
- SUCCESS = getattr(style, 'SUCCESS', lambda x: x)
- template_dir = os.path.join(django_extensions.__path__[0], 'conf', template_name)
- verbosity = options["verbosity"]
- # walks the template structure and copies it
- for d, subdirs, files in os.walk(template_dir):
- relative_dir = d[len(template_dir) + 1:]
- if relative_dir and not os.path.exists(os.path.join(copy_to, relative_dir)):
- os.mkdir(os.path.join(copy_to, relative_dir))
- for i, subdir in enumerate(subdirs):
- if subdir.startswith('.'):
- del subdirs[i]
- for f in files:
- if f.endswith('.pyc') or f.startswith('.DS_Store'):
- continue
- path_old = os.path.join(d, f)
- path_new = os.path.join(copy_to, relative_dir, f).rstrip(".tmpl")
- if os.path.exists(path_new):
- if verbosity > 1:
- print(ERROR("%s already exists" % path_new))
- continue
- if verbosity > 1:
- print(SUCCESS("%s" % path_new))
- with open(path_old, 'r') as fp_orig:
- with open(path_new, 'w') as fp_new:
- fp_new.write(fp_orig.read())
- try:
- shutil.copymode(path_old, path_new)
- _make_writeable(path_new)
- except OSError:
- sys.stderr.write("Notice: Couldn't set permission bits on %s. You're probably using an uncommon filesystem setup. No problem.\n" % path_new)
|
