2009-12-22 23:18:51 +08:00
|
|
|
from optparse import make_option
|
|
|
|
|
2007-08-16 22:34:01 +08:00
|
|
|
from django.core.management.base import NoArgsCommand
|
2009-12-22 23:18:51 +08:00
|
|
|
from django.core.management.sql import sql_flush
|
|
|
|
from django.db import connections, DEFAULT_DB_ALIAS
|
2007-08-16 14:06:55 +08:00
|
|
|
|
2007-08-16 22:34:01 +08:00
|
|
|
class Command(NoArgsCommand):
|
2007-08-16 14:06:55 +08:00
|
|
|
help = "Returns a list of the SQL statements required to return all tables in the database to the state they were in just after they were installed."
|
|
|
|
|
2009-12-22 23:18:51 +08:00
|
|
|
option_list = NoArgsCommand.option_list + (
|
|
|
|
make_option('--database', action='store', dest='database',
|
|
|
|
default=DEFAULT_DB_ALIAS, help='Nominates a database to print the '
|
|
|
|
'SQL for. Defaults to the "default" database.'),
|
|
|
|
)
|
|
|
|
|
2007-08-16 14:06:55 +08:00
|
|
|
output_transaction = True
|
|
|
|
|
2007-08-16 22:34:01 +08:00
|
|
|
def handle_noargs(self, **options):
|
2011-10-23 11:43:43 +08:00
|
|
|
return u'\n'.join(sql_flush(self.style, connections[options.get('database')], only_django=True)).encode('utf-8')
|