-
-from django.db.models.signals import post_syncdb
-import forum.models
-
-def post_syncdb_callback(sender, **kwargs):
- # Import the needed libraries to use the database and detect the
- # DB engine/sever type being currently employed.
- from django.db import connection, connections
-
- verbosity = int(kwargs["verbosity"])
-
- # Get the DB engine being used for persistence for this app.
- current_db_engine = connections.databases[connection.alias]['ENGINE']
-
- # Make sure the updates are only executed for a MySQL DB.
- if current_db_engine.find("mysql") > 0:
- # Ok, mysql was found in the engine description. Go ahead
- # and attempt to execute the alter table statements.
- cursor = connection.cursor()
-
- # Pair the table names with the columns that need to be updated.
- updatable_table_columns = {
- "forum_tag": "name",
- "auth_user": "username"
- }
-
- # Update each column in turn.
- for table_name, column_name in updatable_table_columns.iteritems():
- alter_table_statement = "ALTER TABLE %(table_name)s MODIFY %(column_name)s varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL;" % {
- "table_name": table_name, "column_name": column_name}
- log_status(verbosity,"Updating MySQL column with this statement: " + alter_table_statement)
- cursor.execute(alter_table_statement)
-
-def log_status(verbosity, message):
- if verbosity == 2:
- print "[DEBUG] " + str(message)
-
-# Link the callback to the post_syncdb signal.
-post_syncdb.connect(post_syncdb_callback, sender=forum.models)