- January 6, 2011
- custom-sql south initial-data post_migrate
- 0 (after 0 ratings)
I found that South doesn't execute initial sql file after a migration, like what Django can do after syncdb. So here's the workaround by using post_migrate signal.
Usage: Put your SQL files the same way as you would if you were to use Django to load your initial SQL file (follow Django doc). The only difference is that, the SQL filename needs to in this format: <app_label>.sql OR <app_label>.<backend>.sql This is done this way since the migration is run per app.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26
def run_initial_sql(sender, **kwargs): app_label = kwargs.get('app') import os from django.db import connection, transaction app_dir = os.path.normpath(os.path.join(os.path.dirname(models.get_app(app_label).__file__), 'sql')) backend_name = connection.settings_dict['ENGINE'].split('.')[-1] sql_files = [os.path.join(app_dir, "%s.%s.sql" % (app_label, backend_name)), os.path.join(app_dir, "%s.sql" % app_label)] cursor = connection.cursor() for sql_file in sql_files: try: if os.path.exists(sql_file): print "Loading initial SQL data from '%s'" % sql_file f = open(sql_file) sql = f.read() f.close() cursor.execute(sql) except Exception, e: sys.stderr.write("Failed to install custom SQL file '%s': %s\n" % \ (sql_file, e)) import traceback traceback.print_exc() transaction.rollback_unless_managed() else: transaction.commit_unless_managed() post_migrate.connect(run_initial_sql)
More like this
- Loading initial data per model at table creation (useful with migrations) by GigiusB 2 years ago
- Proper fixtures loading in south data migrations by JustDelight 3 years, 5 months ago
- South unran migration check by rednaw 3 years ago
- Load customized SQL by roppert 7 years, 5 months ago
- immitating 'real' post_syncdb signal by jango 6 years, 3 months ago