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() == []