from __future__ import absolute_import, division, print_function

from .compatibility import basestring

from .dispatch import dispatch

try:

import sqlalchemy as sa

from sqlalchemy import Table

except ImportError:

Table = type(None)

__all__ = ()

@dispatch(Table, basestring)

def create_index(s, column, name=None, unique=False):

if name is None:

raise ValueError('SQL indexes must have a name')

sa.Index(name, getattr(s.c, column), unique=unique).create(s.bind)

@dispatch(Table, list)

def create_index(s, columns, name=None, unique=False):

if name is None:

raise ValueError('SQL indexes must have a name')

args = name,

args += tuple(getattr(s.c, column) for column in columns)

sa.Index(*args, unique=unique).create(s.bind)