Hopiu/django-dbtemplates
1
0
Fork 0
mirror of https://github.com/jazzband/django-dbtemplates.git synced 2026-03-16 22:20:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Extended sync_templates command to make exporting changes out to file easier.

Browse source
This commit is contained in:
olivergeorge 2011-06-09 22:24:09 +10:00 committed by Jannis Leidel
parent 7473b54bb1
commit fb1fdcbd84
3 changed files with 40 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
dbtemplates/management/commands/sync_templates.py
View file

@ -25,13 +25,16 @@ class Command(NoArgsCommand):
"files from database templates"),
make_option("-a", "--app-first", action="store_true", dest="app_first",
default=False, help="look for templates in applications "
"directories before project templates"))
"directories before project templates"),
make_option("-d", "--delete", action="store_true", dest="delete",
default=False, help="Delete templates after syncing"))
def handle_noargs(self, **options):
extension = options.get('ext')
force = options.get('force')
overwrite = options.get('overwrite')
app_first = options.get('app_first')
delete = options.get('delete')
if not extension.startswith("."):
extension = ".%s" % extension
@ -84,16 +87,23 @@ class Command(NoArgsCommand):
path, t.__repr__()))
else:
confirm = overwrite
if confirm == '' or confirm in (
FILES_TO_DATABASE, DATABASE_TO_FILES):
if confirm in ('', FILES_TO_DATABASE, DATABASE_TO_FILES):
if confirm == FILES_TO_DATABASE:
t.content = codecs.open(path, 'r').read()
t.save()
t.sites.add(site)
if delete:
try:
os.remove(path)
except OSError:
raise CommandError(
u"Couldn't delete %s" % path)
elif confirm == DATABASE_TO_FILES:
try:
f = codecs.open(path, 'w')
f.write(t.content)
finally:
f.close()
if delete:
t.delete()
break

1
dbtemplates/templates/dbtemplates/tests/test.html
View file

@ -1 +0,0 @@
This is just a test template.

33
dbtemplates/tests.py
View file

@ -1,5 +1,8 @@
from __future__ import with_statement
import codecs
import os
import shutil
import tempfile
from django.core.management import call_command
from django.template import loader, Context
@ -10,7 +13,8 @@ from django.contrib.sites.models import Site
from dbtemplates.conf import settings
from dbtemplates.models import Template
from dbtemplates.utils.template import get_template_source
from dbtemplates.management.commands.sync_templates import FILES_TO_DATABASE
from dbtemplates.management.commands.sync_templates import (FILES_TO_DATABASE,
DATABASE_TO_FILES)
class DbTemplatesTestCase(TestCase):
def setUp(self):
@ -60,13 +64,30 @@ class DbTemplatesTestCase(TestCase):
def test_sync_templates(self):
old_template_dirs = settings.TEMPLATE_DIRS
temp_template_dir = tempfile.mkdtemp('dbtemplates')
last_path_part = temp_template_dir.split('/')[-1]
temp_template_path = os.path.join(temp_template_dir, 'temp_test.html')
temp_template = codecs.open(temp_template_path, 'w')
try:
self.assertFalse(Template.objects.filter(name='dbtemplates/tests/test.html').exists())
settings.TEMPLATE_DIRS = (
os.path.join(os.path.dirname(__file__), 'templates'),
)
temp_template.write('temp test')
settings.TEMPLATE_DIRS = (temp_template_dir,)
self.assertFalse(Template.objects.filter(name='temp_test.html').exists())
call_command('sync_templates',
force=True, verbosity=0, overwrite=FILES_TO_DATABASE)
self.assertTrue(Template.objects.filter(name='dbtemplates/tests/test.html').exists())
self.assertTrue(Template.objects.filter(name='temp_test.html').exists())
t = Template.objects.get(name='temp_test.html')
t.content = 'temp test modified'
t.save()
call_command('sync_templates',
force=True, verbosity=0, overwrite=DATABASE_TO_FILES)
self.assertTrue('modified' in codecs.open(temp_template_path).read())
call_command('sync_templates',
force=True, verbosity=0, delete=True, overwrite=DATABASE_TO_FILES)
self.assertTrue(os.path.exists(temp_template_path))
self.assertFalse(Template.objects.filter(name='temp_test.html').exists())
finally:
temp_template.close()
settings.TEMPLATE_DIRS = old_template_dirs
shutil.rmtree(temp_template_dir)