mirror of
https://github.com/fergalmoran/dss.git
synced 2026-02-13 03:23:59 +00:00
45 lines
1.7 KiB
Python
Executable File
45 lines
1.7 KiB
Python
Executable File
from django.conf import settings
|
|
|
|
from django.core.management.base import NoArgsCommand
|
|
|
|
|
|
class Command(NoArgsCommand):
|
|
help = "Drop and re-create the database"
|
|
|
|
def handle_noargs(self, **options):
|
|
if settings.DATABASES['default']['ENGINE'] == 'django.db.backends.postgresql_psycopg2':
|
|
self.pgsql_handle_noargs(**options)
|
|
else:
|
|
self.mysql_handle_noargs(**options)
|
|
|
|
def pgsql_handle_noargs(self, **options):
|
|
import psycopg2
|
|
|
|
db = psycopg2.connect(database='postgres', #settings.DATABASES['default']['NAME'],
|
|
host=settings.DATABASES['default']['HOST'],
|
|
user=settings.DATABASES['default']['USER'],
|
|
password=settings.DATABASES['default']['PASSWORD'])
|
|
|
|
cur = db.cursor()
|
|
cur.execute("drop database %s; create database %s;" % (
|
|
settings.DATABASES['default']['NAME'], settings.DATABASES['default']['NAME']))
|
|
|
|
print "Dropped"
|
|
|
|
def mysql_handle_noargs(self, **options):
|
|
import MySQLdb
|
|
|
|
print "Connecting..."
|
|
db = MySQLdb.connect(
|
|
host=settings.DATABASES['default']['HOST'] or "localhost",
|
|
user=settings.DATABASES['default']['USER'],
|
|
passwd=settings.DATABASES['default']['PASSWORD'],
|
|
port=int(settings.DATABASES['default']['PORT'] or 3306))
|
|
|
|
cursor = db.cursor()
|
|
print "Dropping database %s" % settings.DATABASES['default']['NAME']
|
|
cursor.execute("drop database %s; create database %s;" % (
|
|
settings.DATABASES['default']['NAME'], settings.DATABASES['default']['NAME']))
|
|
print "Dropped"
|
|
|