[Django]-Complete django DB reset

21๐Ÿ‘

โœ…

This Snippet gives you the code for a manage.py reset_db command you can install, itโ€™s the minimum change that solves your problem


That said, from comments below:

You might as well just install Django command extensions to get
reset_db and other goodies:
https://github.com/django-extensions/django-extensions

๐Ÿ‘คZach

1๐Ÿ‘

You want sqlreset:

% python manage.py help sqlreset
Usage: manage.py sqlreset [options] <appname appname ...>

Prints the DROP TABLE SQL, then the CREATE TABLE SQL, for the given app name(s).

Options:
  -v VERBOSITY, --verbosity=VERBOSITY
                        Verbosity level; 0=minimal output, 1=normal output,
                        2=all output
  --settings=SETTINGS   The Python path to a settings module, e.g.
                        "myproject.settings.main". If this isn't provided, the
                        DJANGO_SETTINGS_MODULE environment variable will be
                        used.
  --pythonpath=PYTHONPATH
                        A directory to add to the Python path, e.g.
                        "/home/djangoprojects/myproject".
  --traceback           Print traceback on exception
  --version             show program's version number and exit
  -h, --help            show this help message and exit

Just like when you modify a model, Django will not automatically do this for you. It will only output the commands for you to copy and paste.

๐Ÿ‘คjathanism

0๐Ÿ‘

Just assign a new database and drop this db from the db console. Seems to me to be the simplest.

๐Ÿ‘คlprsd

0๐Ÿ‘

Hm, maybe you lie to manage.py, pretending to make fixtures, but only to look for apps:

apps=$(python manage.py makefixture 2>&1 | egrep -v โ€˜(^Error|^django)โ€™|awk -F . โ€˜{print $2}โ€™|uniq); for i in $apps; do python manage.py sqlreset $i; done| grep DROP

That prints out a list of DROP TABLE statements for all apps tables of your project, excluding django tables itself. If you want to include them, remove the |^django pattern vom egrep.

But how to feed the correct database backend? sed/awk-ing through settings.conf? Or better by utilizing a little settings.conf-reading python script itself.

๐Ÿ‘คinitall

0๐Ÿ‘

Assuming you are (luckily) in a Linux environment as well as:

  • using MySQL and know user/password to access projectโ€™s db
  • bash, mysql, readlink are installed
  • settings.py is settings.py
  • dbname is the same as your django project name (see MYAPP var below)
  • all the APPS you want to reset are listed in INSTALLED_APPS as strings with fully qualified module names (starting with the name of your django project)

Change your current folder to your project code where manage.py and settings.py reside. Then run the following:

MYAPP=$(basename $(dirname `readlink -m settings.py`));DJANGOAPPS=$(echo 'import settings; print " ".join([ str(app).lstrip("'${MYAPP}'.") for app in settings.INSTALLED_APPS if "'${MYAPP}'" in app ])' | python - ); python manage.py sqlreset $DJANGOAPPS | mysql -u root --password ${MYAPP}

If you understand what is going in this bash oneliner โ€“ it should not be a problem to build upon this idea (e.g. put it to Makefile).

0๐Ÿ‘

In Django 3.1.1

if you want to totally reset the database and the schema then put this code in any app like this

app/management/commands/resetdb.py

import os
import glob
import shutil
from django.conf import settings
from django.core.management.base import BaseCommand
from django.db import connection


class Command(BaseCommand):
    help = 'Resets the database'

    def handle(self, *args, **options):
        dbname = settings.DATABASES["default"]["NAME"]
        with connection.cursor() as cursor:
            cursor.execute("DROP DATABASE %s" % dbname)
            cursor.execute("CREATE DATABASE %s" % dbname)

        base = str(settings.BASE_DIR)
        migrations = glob.glob(os.path.join(base, "*", "migrations"))

        for migration in migrations:
            shutil.rmtree(migration)

        apps = [migration.split("\\")[-2] for migration in migrations]
        for app in apps:
            os.system("python manage.py makemigrations %s" % app)
        os.system("python manage.py migrate")

now reset it using :

python manage.py resetdb

this will

  • drop the database
  • delete the migrations folders
  • make migrations
  • create a new db
  • migrate the fresh tables
๐Ÿ‘คAl Mahdi

-2๐Ÿ‘

take a look at reset command in djangoโ€™s code, and write your own which drops/creates DB first.

Leave a comment