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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
|
# Copyright 2010 Canonical Ltd. This software is licensed under the
# GNU Affero General Public License version 3 (see the file LICENSE).
# Parts of this module are based on code from the Django project.
# Please see the license file in the third-party/django directory
import os
import sys
from django.core.management.base import BaseCommand
from django.core.management.color import no_style
from django.utils.translation import ugettext as _
class Command(BaseCommand):
fixture_cache = {}
help = _('Installs the named fixture(s) in the database.')
args = _("fixture [fixture ...]")
def handle(self, *fixture_labels, **options):
from django.db.models import get_apps
from django.core import serializers
from django.db import connection, transaction
from django.conf import settings
self.style = no_style()
show_traceback = options.get('traceback', False)
# commit is a stealth option - it isn't really useful as
# a command line option, but it can be useful when invoking
# loaddata from within another script.
# If commit=True, loaddata will use its own transaction;
# if commit=False, the data load SQL will become part of
# the transaction in place when loaddata was invoked.
commit = options.get('commit', True)
# Keep a count of the installed objects and fixtures
fixture_count = 0
object_count = 0
models = set()
humanize = lambda dirname: dirname and "'%s'" % dirname or 'absolute path'
# Get a cursor (even though we don't need one yet). This has
# the side effect of initializing the test database (if
# it isn't already initialized).
cursor = connection.cursor()
# Start transaction management. All fixtures are installed in a
# single transaction to ensure that all references are resolved.
if commit:
transaction.commit_unless_managed()
transaction.enter_transaction_management()
transaction.managed(True)
app_fixtures = [os.path.join(os.path.dirname(app.__file__), 'fixtures')
for app in get_apps()]
for fixture_label in fixture_labels:
if fixture_label in Command.fixture_cache:
fixture = Command.fixture_cache[fixture_label]
cursor.execute(fixture['sql'])
object_count += 1
models.update(fixture['models'])
continue
else:
current_query = len(connection.queries)
new_fixture = {'models': set()}
parts = fixture_label.split('.')
if len(parts) == 1:
fixture_name = fixture_label
formats = serializers.get_public_serializer_formats()
else:
fixture_name, format = '.'.join(parts[:-1]), parts[-1]
if format in serializers.get_public_serializer_formats():
formats = [format]
else:
formats = []
if not formats:
sys.stderr.write(
self.style.ERROR(_("Problem installing fixture "
"'%(fixture_name)s': %(format)s is "
"not a known serialization format.") %
{'fixture_name': fixture_name,
'format': format}))
transaction.rollback()
transaction.leave_transaction_management()
return
if os.path.isabs(fixture_name):
fixture_dirs = [fixture_name]
else:
fixture_dirs = (app_fixtures +
list(settings.FIXTURE_DIRS) + [''])
for fixture_dir in fixture_dirs:
label_found = False
for format in formats:
serializer = serializers.get_serializer(format)
try:
full_path = os.path.join(
fixture_dir, '.'.join([fixture_name, format]))
fixture = open(full_path, 'r')
if label_found:
fixture.close()
print self.style.ERROR(
"Multiple fixtures named '%(fixture_name)s'"
"in %(fixture_dir)s. Aborting." %
{'fixture_name': fixture_name,
'fixture_dir': humanize(fixture_dir)})
transaction.rollback()
transaction.leave_transaction_management()
return
else:
fixture_count += 1
objects_in_fixture = 0
try:
objects = serializers.deserialize(format,
fixture)
for obj in objects:
objects_in_fixture += 1
models.add(obj.object.__class__)
new_fixture['models'].add(
obj.object.__class__)
obj.save()
object_count += objects_in_fixture
label_found = True
except (SystemExit, KeyboardInterrupt):
raise
except Exception:
import traceback
fixture.close()
transaction.rollback()
transaction.leave_transaction_management()
if show_traceback:
traceback.print_exc()
else:
sys.stderr.write(
self.style.ERROR(
"Problem installing fixture "
"'%(full_path)s': %(exception)s\n" %
{'full_path': full_path,
'exception': ''.join(traceback.format_exception(
sys.exc_type,
sys.exc_value,
sys.exc_traceback))}))
return
fixture.close()
all = connection.queries[current_query:]
queries = [x['sql'] for x in all
if not x['sql'].lower().startswith('select')]
new_fixture['sql'] = ';'.join(queries)
Command.fixture_cache[fixture_label] = new_fixture
# If the fixture we loaded contains 0 objects,
# assume that an error was encountered during
# fixture loading.
if objects_in_fixture == 0:
sys.stderr.write(
self.style.ERROR(
"No fixture data found for "
"'%(fixture_name)s'. (File format "
"may be invalid.)" %
{'fixture_name': fixture_name}))
transaction.rollback()
transaction.leave_transaction_management()
return
except:
pass
# If we found even one object in a fixture, we need to reset the
# database sequences.
if object_count > 0:
sequence_sql = connection.ops.sequence_reset_sql(self.style,
models)
if sequence_sql:
for line in sequence_sql:
cursor.execute(line)
if commit:
transaction.commit()
transaction.leave_transaction_management()
# Close the DB connection. This is required as a workaround for an
# edge case in MySQL: if the same connection is used to
# create tables, load data, and query, the query can return
# incorrect results. See Django #7572, MySQL #37735.
if commit:
connection.close()
|