intermediate commit, testing a bug fix, reinstating get_sql_create_template_values

Author: sp-mmartin

django_celery_beat/models.py

@@ -1,6 +1,6 @@
 """Database models."""
 from __future__ import absolute_import, unicode_literals
-import logging
+
 from datetime import timedelta
 import dateutil
 
@@ -49,9 +49,11 @@ def _is_mysql_backend(self, connection):
 
     def _create_sql(self, model, schema_editor, using=''):
         sql_create_index = 'CREATE INDEX %(name)s ON %(table)s (%(columns)s(%(size)d))%(extra)s'
-        sql_parameters = super(CeleryMySQLIndex, self).create_sql(model, schema_editor, using=using)
-        logging.warning('creat index content - {c}'.format(c=sql_parameters))
-        print(sql_parameters)
+        sql_parameters = self.get_sql_create_template_values(
+            model,
+            schema_editor,
+            using
+        )
         sql_parameters['size'] = getattr(
             settings,
             'DJANGO_CELERY_BEAT_NAME_MAX_LENGTH',
@@ -66,6 +68,22 @@ def create_sql(self, model, schema_editor, using=''):
         else:
             return super(CeleryMySQLIndex, self).create_sql(model, schema_editor, using=using)
 
+    def get_sql_create_template_values(self, model, schema_editor, using):
+        fields = [model._meta.get_field(field_name) for field_name, order in self.fields_orders]
+        tablespace_sql = schema_editor._get_index_tablespace_sql(model, fields)
+        quote_name = schema_editor.quote_name
+        columns = [
+            ('%s %s' % (quote_name(field.column), order)).strip()
+            for field, (field_name, order) in zip(fields, self.fields_orders)
+        ]
+        return {
+            'table': quote_name(model._meta.db_table),
+            'name': quote_name(self.name),
+            'columns': ', '.join(columns),
+            'using': using,
+            'extra': tablespace_sql,
+        }
+
 
 @python_2_unicode_compatible
 class SolarSchedule(models.Model):