def handle(self, *args, **options):
cmd = options['command']
database = options['database']
if database is None:
databases = [alias for alias in connections]
else:
databases = [database]
with atomic():
try:
if cmd == 'list':
self._list(databases)
elif cmd == 'create':
for alias, triggers in create_triggers(databases).items():
self.stdout.write("Database: {}".format(alias))
for trigger in triggers:
self.stdout.write(
" Created {0[2]} for {0[1]}".format(
trigger))
self.stdout.write('')
elif cmd == 'drop':
for alias, triggers in drop_triggers(*databases).items():
self.stdout.write("Database: {}".format(alias))
for trigger in triggers:
self.stdout.write(
" Dropped {0[2]}".format(trigger))
self.stdout.write('')
else: # pragma: no cover
raise Exception()
except ImproperlyConfigured as e: # pragma: no cover
self.stdout.write(self.style.ERROR(e))
def handle(self, *args, **options):
cmd = options['command']
database = options['database']
if database is None:
databases = [alias for alias in connections]
else:
databases = [database]
with atomic():
try:
if cmd == 'list':
self._list(databases)
elif cmd == 'create':
for alias, triggers in create_triggers(databases).items():
self.stdout.write("Database: {}".format(alias))
for trigger in triggers:
self.stdout.write(" Created {0[2]} for {0[1]}".format(trigger))
self.stdout.write('')
elif cmd == 'drop':
for alias, triggers in drop_triggers(*databases).items():
self.stdout.write("Database: {}".format(alias))
for trigger in triggers:
self.stdout.write(" Dropped {0[2]}".format(trigger))
self.stdout.write('')
else: # pragma: no cover
raise Exception()
except ImproperlyConfigured as e: # pragma: no cover
self.stdout.write(self.style.ERROR(e))
def test_drop_triggers(db):
conn = connections['default']
ret = drop_triggers('default')
assert sorted([i[0].__name__ for i in ret['default']]) == [
'DropTriggerConcurrentModel', 'TriggerConcurrentModel'
]
assert factory(conn).get_list() == []
def test_drop_triggers(db):
conn = connections['default']
ret = drop_triggers('default')
assert sorted([i[0].__name__ for i in ret['default']]) == ['DropTriggerConcurrentModel',
'TriggerConcurrentModel']
assert factory(conn).get_list() == []
