schema.py 2.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. from django.db.backends.schema import BaseDatabaseSchemaEditor
  2. from django.db.models import NOT_PROVIDED
  3. class DatabaseSchemaEditor(BaseDatabaseSchemaEditor):
  4. sql_rename_table = "RENAME TABLE %(old_table)s TO %(new_table)s"
  5. sql_alter_column_null = "MODIFY %(column)s %(type)s NULL"
  6. sql_alter_column_not_null = "MODIFY %(column)s %(type)s NOT NULL"
  7. sql_alter_column_type = "MODIFY %(column)s %(type)s"
  8. sql_rename_column = "ALTER TABLE %(table)s CHANGE %(old_column)s %(new_column)s %(type)s"
  9. sql_delete_unique = "ALTER TABLE %(table)s DROP INDEX %(name)s"
  10. sql_create_fk = "ALTER TABLE %(table)s ADD CONSTRAINT %(name)s FOREIGN KEY (%(column)s) REFERENCES %(to_table)s (%(to_column)s)"
  11. sql_delete_fk = "ALTER TABLE %(table)s DROP FOREIGN KEY %(name)s"
  12. sql_delete_index = "DROP INDEX %(name)s ON %(table)s"
  13. sql_delete_pk = "ALTER TABLE %(table)s DROP PRIMARY KEY"
  14. alter_string_set_null = 'MODIFY %(column)s %(type)s NULL;'
  15. alter_string_drop_null = 'MODIFY %(column)s %(type)s NOT NULL;'
  16. sql_create_pk = "ALTER TABLE %(table)s ADD CONSTRAINT %(name)s PRIMARY KEY (%(columns)s)"
  17. sql_delete_pk = "ALTER TABLE %(table)s DROP PRIMARY KEY"
  18. def quote_value(self, value):
  19. # Inner import to allow module to fail to load gracefully
  20. import MySQLdb.converters
  21. return MySQLdb.escape(value, MySQLdb.converters.conversions)
  22. def skip_default(self, field):
  23. """
  24. MySQL doesn't accept default values for longtext and longblob
  25. and implicitly treats these columns as nullable.
  26. """
  27. return field.db_type(self.connection) in {'longtext', 'longblob'}
  28. def add_field(self, model, field):
  29. super(DatabaseSchemaEditor, self).add_field(model, field)
  30. # Simulate the effect of a one-off default.
  31. if self.skip_default(field) and field.default not in {None, NOT_PROVIDED}:
  32. effective_default = self.effective_default(field)
  33. self.execute('UPDATE %(table)s SET %(column)s = %%s' % {
  34. 'table': self.quote_name(model._meta.db_table),
  35. 'column': self.quote_name(field.column),
  36. }, [effective_default])