from django.core.management.base import LabelCommand

class Command(LabelCommand):

    help = "Creates the table needed to use the SQL cache backend."

    args = "<tablename>"

    label = 'tablename'

    requires_model_validation = False

    def handle_label(self, tablename, **options):

        from django.db import connection, transaction, models

        fields = (

            # "key" is a reserved word in MySQL, so use "cache_key" instead.

            models.CharField(name='cache_key', max_length=255, unique=True, primary_key=True),

            models.TextField(name='value'),

            models.DateTimeField(name='expires', db_index=True),

        )

        table_output = []

        index_output = []

        qn = connection.ops.quote_name

        for f in fields:

            field_output = [qn(f.name), f.db_type()]

            field_output.append("%sNULL" % (not f.null and "NOT " or ""))

            if f.primary_key:

                field_output.append("PRIMARY KEY")

            elif f.unique:

                field_output.append("UNIQUE")

            if f.db_index:

                unique = f.unique and "UNIQUE " or ""

                index_output.append("CREATE %sINDEX %s_%s ON %s (%s);" % \

                    (unique, tablename, f.name, qn(tablename),

                    qn(f.name)))

            table_output.append(" ".join(field_output))

        full_statement = ["CREATE TABLE %s (" % qn(tablename)]

        for i, line in enumerate(table_output):

            full_statement.append('    %s%s' % (line, i < len(table_output)-1 and ',' or ''))

        full_statement.append(');')

        curs = connection.cursor()

        curs.execute("\n".join(full_statement))

        for statement in index_output:

            curs.execute(statement)

        transaction.commit_unless_managed()