Ticket #5062: mssql.r5986.diff

File mssql.r5986.diff, 32.5 KB (added by gregoire@…, 17 years ago)

Patch for r5986

  • db/models/base.py

     
    246246                    (qn(self._meta.db_table), qn(self._meta.order_with_respect_to.column)))
    247247                db_values.append(getattr(self, self._meta.order_with_respect_to.attname))
    248248            if db_values:
     249                if pk_set and (settings.DATABASE_ENGINE=="ado_mssql" or settings.DATABASE_ENGINE=="mssql"):
     250                    # You can't insert an auto value into a column unless you do
     251                    # this in MSSQL
     252                    # TODO: Only works for auto-id's... how chek it properly?
     253                    if self._meta.pk.column == 'id':
     254                        cursor.execute("SET IDENTITY_INSERT %s ON" % \
     255                            backend.quote_name(self._meta.db_table))
     256                   
    249257                cursor.execute("INSERT INTO %s (%s) VALUES (%s)" % \
    250258                    (qn(self._meta.db_table), ','.join(field_names),
    251259                    ','.join(placeholders)), db_values)
     260                   
     261                if pk_set and (settings.DATABASE_ENGINE=="ado_mssql" or settings.DATABASE_ENGINE=="mssql"):
     262                    if self._meta.pk.column == 'id':
     263                        cursor.execute("SET IDENTITY_INSERT %s OFF" %\
     264                            backend.quote_name(self._meta.db_table))
    252265            else:
    253266                # Create a new record with defaults for everything.
    254267                cursor.execute("INSERT INTO %s (%s) VALUES (%s)" %
  • db/models/fields/__init__.py

     
    223223                value = int(value)
    224224            except ValueError:
    225225                raise ValueError("The __year lookup type requires an integer argument")
    226             return ['%s-01-01 00:00:00' % value, '%s-12-31 23:59:59.999999' % value]
     226            return ['%s-01-01 00:00:00' % value, '%s-12-31 23:59:59.99' % value]
    227227        raise TypeError("Field has invalid lookup: %s" % lookup_type)
    228228
    229229    def has_default(self):
     
    574574        if value is not None:
    575575            # MySQL will throw a warning if microseconds are given, because it
    576576            # doesn't support microseconds.
    577             if settings.DATABASE_ENGINE == 'mysql' and hasattr(value, 'microsecond'):
     577            if settings.DATABASE_ENGINE in ('mysql', 'ado_mssql','mssql') and hasattr(value, 'microsecond'):
    578578                value = value.replace(microsecond=0)
    579579            value = smart_unicode(value)
    580580        return Field.get_db_prep_save(self, value)
    581581
    582582    def get_db_prep_lookup(self, lookup_type, value):
     583        # MSSQL doesn't like microseconds.
     584        if settings.DATABASE_ENGINE in ('mysql', 'ado_mssql','mssql') and hasattr(value, 'microsecond'):
     585            value = value.replace(microsecond=0)       
    583586        if lookup_type == 'range':
    584587            value = [smart_unicode(v) for v in value]
    585588        else:
     
    966969        # Casts dates into string format for entry into database.
    967970        if value is not None:
    968971            # MySQL will throw a warning if microseconds are given, because it
    969             # doesn't support microseconds.
    970             if settings.DATABASE_ENGINE == 'mysql' and hasattr(value, 'microsecond'):
     972            # doesn't support microseconds. Ditto MSSQL
     973            if settings.DATABASE_ENGINE in ('mysql', 'ado_mssql','mssql') \
     974                            and hasattr(value, 'microsecond'):
    971975                value = value.replace(microsecond=0)
    972976            if settings.DATABASE_ENGINE == 'oracle':
    973977                # cx_Oracle expects a datetime.datetime to persist into TIMESTAMP field.
  • db/backends/mssql/base.py

     
     1"""
     2Alpha Multi-plataform MSSQL database backend for Django.
     3
     4Requires pymssql >= v0.8.0: http://pymssql.sourceforge.net/
     5"""
     6
     7import datetime
     8from django.db.backends import BaseDatabaseWrapper, BaseDatabaseFeatures, BaseDatabaseOperations, util
     9from django.core.exceptions import ImproperlyConfigured
     10from django.utils.datastructures import SortedDict
     11
     12try:
     13    import pymssql as Database
     14except ImportError, e:
     15    raise ImproperlyConfigured("Error loading pymssql module: %s" % e)
     16
     17try:
     18    import mx
     19except ImportError:
     20    mx = None
     21
     22try:
     23    # Only exists in Python 2.4+
     24    from threading import local
     25except ImportError:
     26    # Import copy of _thread_local.py from Python 2.4
     27    from django.utils._threading_local import local
     28
     29DatabaseError = Database.DatabaseError
     30IntegrityError = Database.IntegrityError
     31
     32class DatabaseFeatures(BaseDatabaseFeatures):
     33        pass
     34
     35class DatabaseOperations(BaseDatabaseOperations):
     36    def last_insert_id(self, cursor, table_name, pk_name):
     37            cursor.execute("SELECT %s FROM %s WHERE %s = IDENT_CURRENT('%s')" % (pk_name, table_name, pk_name,table_name))
     38            return cursor.fetchone()[0]
     39           
     40    def query_set_class(self, DefaultQuerySet):
     41            "Create a custom QuerySet class for SqlServer."
     42       
     43            from django.db import connection
     44            from django.db.models.query import EmptyResultSet, GET_ITERATOR_CHUNK_SIZE, quote_only_if_word
     45       
     46            class SqlServerQuerySet(DefaultQuerySet):
     47       
     48                def iterator(self):
     49                    "Performs the SELECT database lookup of this QuerySet."
     50       
     51                    from django.db.models.query import get_cached_row
     52       
     53                    # self._select is a dictionary, and dictionaries' key order is
     54                    # undefined, so we convert it to a list of tuples.
     55                    extra_select = self._select.items()
     56       
     57                    full_query = None
     58       
     59                    try:
     60                        try:
     61                            select, sql, params, full_query = self._get_sql_clause(get_full_query=True)
     62                        except TypeError:
     63                            select, sql, params = self._get_sql_clause()
     64                    except EmptyResultSet:
     65                        raise StopIteration
     66                    if not full_query:
     67                        full_query = "SELECT %s%s\n%s" % \
     68                                     ((self._distinct and "DISTINCT " or ""),
     69                                      ', '.join(select), sql)
     70       
     71                    cursor = connection.cursor()
     72                    cursor.execute(full_query, params)
     73       
     74                    fill_cache = self._select_related
     75                    fields = self.model._meta.fields
     76                    index_end = len(fields)
     77       
     78                    # so here's the logic;
     79                    # 1. retrieve each row in turn
     80                    # 2. convert NCLOBs
     81       
     82                    while 1:
     83                        rows = cursor.fetchmany(GET_ITERATOR_CHUNK_SIZE)
     84                        if not rows:
     85                            raise StopIteration
     86                        for row in rows:
     87                            row = self.resolve_columns(row, fields)
     88                            if fill_cache:
     89                                obj, index_end = get_cached_row(klass=self.model, row=row,
     90                                                                index_start=0, max_depth=self._max_related_depth)
     91                            else:
     92                                obj = self.model(*row[:index_end])
     93                            for i, k in enumerate(extra_select):
     94                                setattr(obj, k[0], row[index_end+i])
     95                            yield obj
     96       
     97       
     98                def _get_sql_clause(self, get_full_query=False):
     99                    from django.db.models.query import fill_table_cache, \
     100                        handle_legacy_orderlist, orderfield2column
     101       
     102                    opts = self.model._meta
     103                    qn = connection.ops.quote_name
     104       
     105                    # Construct the fundamental parts of the query: SELECT X FROM Y WHERE Z.
     106                    select = ["%s.%s" % (qn(opts.db_table), qn(f.column)) for f in opts.fields]
     107                    tables = [quote_only_if_word(t) for t in self._tables]
     108                    joins = SortedDict()
     109                    where = self._where[:]
     110                    params = self._params[:]
     111       
     112                    # Convert self._filters into SQL.
     113                    joins2, where2, params2 = self._filters.get_sql(opts)
     114                    joins.update(joins2)
     115                    where.extend(where2)
     116                    params.extend(params2)
     117       
     118                    # Add additional tables and WHERE clauses based on select_related.
     119                    if self._select_related:
     120                        fill_table_cache(opts, select, tables, where, opts.db_table, [opts.db_table])
     121       
     122                    # Add any additional SELECTs.
     123                    if self._select:
     124                        select.extend(['(%s) AS %s' % (quote_only_if_word(s[1]), qn(s[0])) for s in self._select.items()])
     125       
     126                    # Start composing the body of the SQL statement.
     127                    sql = [" FROM", qn(opts.db_table)]
     128       
     129                    # Compose the join dictionary into SQL describing the joins.
     130                    if joins:
     131                        sql.append(" ".join(["%s %s %s ON %s" % (join_type, table, alias, condition)
     132                                        for (alias, (table, join_type, condition)) in joins.items()]))
     133       
     134                    # Compose the tables clause into SQL.
     135                    if tables:
     136                        sql.append(", " + ", ".join(tables))
     137       
     138                    # Compose the where clause into SQL.
     139                    if where:
     140                        sql.append(where and "WHERE " + " AND ".join(where))
     141                   
     142                    #copy a version suitable for LIMIT
     143                    sql2=[]
     144                    [sql2.append(x) for x in sql]           
     145                    # ORDER BY clause
     146                    order_by = []
     147                    if self._order_by is not None:
     148                        ordering_to_use = self._order_by
     149                    else:
     150                        ordering_to_use = opts.ordering
     151                    for f in handle_legacy_orderlist(ordering_to_use):
     152                        if f == '?': # Special case.
     153                            order_by.append(connection.ops.get_random_function_sql())
     154                        else:
     155                            if f.startswith('-'):
     156                                col_name = f[1:]
     157                                order = "DESC"
     158                            else:
     159                                col_name = f
     160                                order = "ASC"
     161                            if "." in col_name:
     162                                table_prefix, col_name = col_name.split('.', 1)
     163                                table_prefix = qn(table_prefix) + '.'
     164                            else:
     165                                # Use the database table as a column prefix if it wasn't given,
     166                                # and if the requested column isn't a custom SELECT.
     167                                if "." not in col_name and col_name not in (self._select or ()):
     168                                    table_prefix = qn(opts.db_table) + '.'
     169                                else:
     170                                    table_prefix = ''
     171                            order_by.append('%s%s %s' % (table_prefix, qn(orderfield2column(col_name, opts)), order))
     172                    if order_by:
     173                        sql.append("ORDER BY " + ", ".join(order_by))
     174       
     175                    # Look for column name collisions in the select elements
     176                    # and fix them with an AS alias.  This allows us to do a
     177                    # SELECT * later in the paging query.
     178                    cols = [clause.split('.')[-1] for clause in select]
     179                    for index, col in enumerate(cols):
     180                        if cols.count(col) > 1:
     181                            col = '%s%d' % (col.replace('[', '').replace(']',''), index)
     182                            cols[index] = qn(col)
     183                            select[index] = '%s AS %s' % (select[index], qn(col))
     184       
     185                    # LIMIT and OFFSET clauses
     186                    # To support limits and offsets, SqlServer requires some funky rewriting of an otherwise normal looking query.
     187                    select_clause = ",".join(select)
     188                    distinct = (self._distinct and "DISTINCT " or "")           
     189                    full_query = None
     190                   
     191                    if self._limit is None:
     192                        assert self._offset is None, "'offset' is not allowed without 'limit'"
     193       
     194                    if self._offset is not None:
     195                        offset = int(self._offset)
     196                    else:
     197                        offset = 0
     198       
     199                    if self._limit is not None:
     200                        limit = int(self._limit)
     201                    else:
     202                        limit = None
     203                   
     204                    limit_and_offset_clause = ''
     205       
     206                    if limit is not None:
     207                        limit_and_offset_clause = True
     208                    elif offset:
     209                        limit_and_offset_clause = True
     210                   
     211                    if limit_and_offset_clause:
     212                        #Django give:
     213                        # Offset : Start row
     214                        # Limit : How much aditional rows fetch
     215                       
     216                        # This must be transformed to Sql2005 to:
     217                        # Offset : First Row
     218                        # Limit : EndRow
     219                        StartRow = offset + 1
     220                        EndRow = StartRow + limit - 1
     221                        # and for Sql2000
     222                        # Offset : Top rows
     223                        # Limit: From where
     224                        limit = limit + offset
     225                        if offset==0:
     226                            offset = limit
     227                        else:
     228                            offset = offset + 1
     229                        #Must use a order. If not specified, use Id.
     230                        if len(order_by)==0:
     231                            order_by.append('%s.%s ASC' %
     232                                (qn(opts.db_table),
     233                                qn(opts.fields[0].db_column or opts.fields[0].column)
     234                                )
     235                            )
     236       
     237                        order_by_clause = ", ".join(order_by)
     238                        order_by_clauseReverse = ""
     239       
     240                        #For Sql2005+ use native implementation...
     241                        if version()>8:
     242                            fmt = \
     243"""
     244SELECT *
     245FROM (
     246    SELECT %(distinc)s TOP %(EndRow)s
     247        %(fields)s, ROW_NUMBER()
     248        OVER(
     249            ORDER BY  %(orderby)s
     250        ) AS row
     251    %(sql)s ORDER BY %(orderby)s
     252    ) AS x
     253    WHERE x.row BETWEEN %(StartRow)s AND %(EndRow)s
     254"""
     255                        else:   
     256                            #Is necesary reverse all the second order by for the trick to work...
     257                            order_by_clauseReverse= ", ".join(self.change_order_direction(order_by))
     258                           
     259                            fmt = \
     260"""
     261SELECT * FROM (
     262  SELECT TOP %(offset)s * FROM (
     263    SELECT TOP %(limit)s %(distinc)s%(fields)s
     264        %(sql)s   
     265    ORDER BY %(orderby)s
     266  ) AS %(table)s
     267  ORDER BY %(orderbyReverse)s) AS %(table)s
     268ORDER BY %(orderby)s
     269"""
     270       
     271                        full_query = fmt % {'distinc':distinct, 'fields':select_clause,
     272                                                'sql':" ".join(sql2),'orderby':order_by_clause,
     273                                                'orderbyReverse':order_by_clauseReverse,
     274                                                'table':qn(opts.db_table),
     275                                                'offset':offset,'limit':limit,
     276                                                'StartRow':StartRow,'EndRow':EndRow}
     277                   
     278                    print full_query
     279                    if get_full_query:
     280                        return select, " ".join(sql), params, full_query
     281                    else:
     282                        return select, " ".join(sql), params
     283       
     284       
     285                def change_order_direction(self,order_by):
     286                    newOrder=[]
     287                   
     288                    for order in order_by:
     289                        if order.find(' ASC'):
     290                            newOrder.append(order.replace(' ASC',' DESC'))
     291                        else:
     292                            newOrder.append(order.replace(' DESC',' ASC'))
     293                   
     294                    return newOrder
     295       
     296                def resolve_columns(self, row, fields=()):
     297                    from django.db.models.fields import DateField, DateTimeField, \
     298                        TimeField, BooleanField, NullBooleanField, DecimalField, Field
     299                    values = []
     300                    for value, field in map(None, row, fields):
     301                        # Convert 1 or 0 to True or False
     302                        if value in (1, 0) and isinstance(field, (BooleanField, NullBooleanField)):
     303                            value = bool(value)
     304                        # Convert floats to decimals
     305                        elif value is not None and isinstance(field, DecimalField):
     306                            value = util.typecast_decimal(field.format_number(value))
     307                        values.append(value)
     308                    return values
     309       
     310            return SqlServerQuerySet
     311   
     312    def date_extract_sql(self, lookup_type, field_name):
     313                # lookup_type is 'year', 'month', 'day'
     314                return "DATEPART(%s, %s)" % (lookup_type, table_name)
     315
     316    def date_trunc_sql(self, lookup_type, field_name):
     317                # lookup_type is 'year', 'month', 'day'
     318                if lookup_type=='year':
     319                        return "Convert(datetime, Convert(varchar, DATEPART(year, %s)) + '/01/01')" % field_name
     320                if lookup_type=='month':
     321                        return "Convert(datetime, Convert(varchar, DATEPART(year, %s)) + '/' + Convert(varchar, DATEPART(month, %s)) + '/01')" % (field_name, field_name)
     322                if lookup_type=='day':
     323                        return "Convert(datetime, Convert(varchar(12), %s))" % field_name
     324
     325    def limit_offset_sql(self, limit, offset=None):
     326                # Limits and offset are too complicated to be handled here.
     327                # Look for a implementation similar to SqlServer backend
     328                return ""
     329
     330    def quote_name(self, name):
     331        if name.startswith('[') and name.endswith(']'):
     332                return name # Quoting once is enough.
     333        return '[%s]' % name
     334
     335    def random_function_sql(self):
     336        return "RAND()"
     337       
     338    def tablespace_sql(self, tablespace, inline=False):
     339     return "ON %s" % quote_name(tablespace)
     340
     341    def sql_flush(self, style, tables, sequences):
     342                """Return a list of SQL statements required to remove all data from
     343                all tables in the database (without actually removing the tables
     344                themselves) and put the database in an empty 'initial' state
     345                """
     346                # Cannot use TRUNCATE on tables that are referenced by a FOREIGN KEY
     347                # So must use the much slower DELETE
     348                sql_list = ['%s %s %s;' % \
     349                                        (style.SQL_KEYWORD('DELETE'),
     350                                        style.SQL_KEYWORD('FROM'),
     351                                        style.SQL_FIELD(self.quote_name(table))
     352                                        )  for table in tables]
     353                #The reset the counters on each table.
     354                sql_list.extend(['%s %s %s %s %s %s %s;' % (
     355                        style.SQL_KEYWORD('DBCC'),
     356                        style.SQL_KEYWORD('CHECKIDENT'),
     357                        style.SQL_FIELD(self.quote_name(seq["table"])),
     358                        style.SQL_KEYWORD('RESEED'),
     359                        style.SQL_FIELD('1'),
     360                        style.SQL_KEYWORD('WITH'),
     361                        style.SQL_KEYWORD('NO_INFOMSGS'),
     362                        ) for seq in sequences])
     363               
     364                return sql_list
     365
     366
     367
     368def complain(*args, **kwargs):
     369    raise ImproperlyConfigured("You haven't set the DATABASE_ENGINE setting yet.")
     370
     371def ignore(*args, **kwargs):
     372    pass
     373
     374class DatabaseError(Exception):
     375    pass
     376
     377class IntegrityError(DatabaseError):
     378    pass
     379
     380class DatabaseWrapper(BaseDatabaseWrapper):
     381    features = DatabaseFeatures()
     382    ops = DatabaseOperations()
     383    operators = {
     384            'exact': '= %s',
     385            'iexact': 'LIKE %s',
     386            'contains': 'LIKE %s',
     387            'icontains': 'LIKE %s',
     388            'gt': '> %s',
     389            'gte': '>= %s',
     390            'lt': '< %s',
     391            'lte': '<= %s',
     392            'startswith': 'LIKE %s',
     393            'endswith': 'LIKE %s',
     394            'istartswith': 'LIKE %s',
     395            'iendswith': 'LIKE %s',
     396        }
     397    def __init__(self, **kwargs):
     398        self.connection = None
     399        self.queries = []
     400
     401    def cursor(self):
     402        from django.conf import settings
     403        if self.connection is None:
     404            if settings.DATABASE_NAME == '' or settings.DATABASE_USER == '':
     405                raise ImproperlyConfigured("You need to specify both DATABASE_NAME and DATABASE_USER in your Django settings file.")
     406            if not settings.DATABASE_HOST:
     407                settings.DATABASE_HOST = "127.0.0.1"
     408
     409            if settings.DATABASE_PORT:
     410                hostStr = '%s:%s' % ( settings.DATABASE_HOST ,settings.DATABASE_PORT)
     411            else:
     412                hostStr = settings.DATABASE_HOST
     413                                                       
     414            self.connection = Database.connect(host=hostStr,user=settings.DATABASE_USER,password=settings.DATABASE_PASSWORD,database=settings.DATABASE_NAME)
     415       
     416        self.connection.cursor().execute("SET DATEFORMAT ymd\nGO")
     417       
     418        cursor = self.connection.cursor()
     419        if settings.DEBUG:
     420            return util.CursorDebugWrapper(cursor, self)
     421        return cursor
     422
     423    def _commit(self):
     424        if self.connection is not None:
     425            return self.connection.commit()
     426
     427    def _rollback(self):
     428        if self.connection is not None:
     429            return self.connection.rollback()
     430
     431    def close(self):
     432        if self.connection is not None:
     433            self.connection.close()
     434            self.connection = None
     435
     436'''
     437    Return the major version of the server. 7=Sql 7,8=Sql2000,9=Sql2005
     438'''
     439def version():
     440    cur = DatabaseWrapper().cursor()
     441    cur.execute("SELECT SERVERPROPERTY('ProductVersion')")
     442   
     443    return int(cur.fetchone()[0].split('.')[0])
     444
     445
     446if __name__ == '__main__':
     447    from mysite.polls.models import Poll, Choice
     448    from datetime import datetime
     449   
     450    #Poll.objects.all().delete()
     451    #i =0
     452    #for i in range(i,150):
     453    #    p = Poll(question="%s" % i, pub_date=datetime.now())
     454    #    p.save()
     455
     456    for poll in Poll.objects.all()[:10]:
     457        print poll
     458 Pas de fin de ligne
  • db/backends/mssql/client.py

     la fin du fichier
     
     1from django.conf import settings
     2import os
     3import sys
     4
     5def runshell():
     6    if os.name=='nt':
     7        args = ['']
     8        db = settings.DATABASE_OPTIONS.get('db', settings.DATABASE_NAME)
     9        user = settings.DATABASE_OPTIONS.get('user', settings.DATABASE_USER)
     10        passwd = settings.DATABASE_OPTIONS.get('passwd', settings.DATABASE_PASSWORD)
     11        host = settings.DATABASE_OPTIONS.get('host', settings.DATABASE_HOST)
     12        port = settings.DATABASE_OPTIONS.get('port', settings.DATABASE_PORT)
     13        defaults_file = settings.DATABASE_OPTIONS.get('read_default_file')
     14        # Seems to be no good way to set sql_mode with CLI
     15       
     16        if defaults_file:
     17            args += ["-i %s" % defaults_file]
     18        if user:
     19            args += ["-U %s" % user]
     20        if passwd:
     21            args += ["-P %s" % passwd]
     22        if host:
     23            args += ["-E %s" % host]
     24        if db:
     25            args += ["-d %s" % db]
     26   
     27        cmd = "osql %s" % ' '.join(args)
     28   
     29        rv = os.system(cmd)
     30        if (rv):
     31           print "Error al ejecutar %s " % rv
     32           sys.exit(rv)
     33    else:
     34        raise NotImplementedError
     35
     36   
     37   
  • db/backends/mssql/__init__.py

     
     1# placeholder
     2 Pas de fin de ligne
  • db/backends/mssql/introspection.py

     la fin du fichier
     
     1def get_table_list(cursor):
     2    "Returns a list of table names in the current database."
     3    cursor.execute("SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_TYPE = 'BASE TABLE'")
     4    return [row[2] for row in cursor.fetchall()]
     5
     6def _is_auto_field(cursor, table_name, column_name):
     7    cursor.execute("SELECT COLUMNPROPERTY( OBJECT_ID('%s'),'%s','IsIdentity')" % (table_name, column_name))
     8    return cursor.fetchall()[0][0]
     9
     10def get_table_description(cursor, table_name, identity_check=True):
     11    """Returns a description of the table, with the DB-API cursor.description interface.
     12
     13    The 'auto_check' parameter has been added to the function argspec.
     14    If set to True, the function will check each of the table's fields for the
     15    IDENTITY property (the IDENTITY property is the MSSQL equivalent to an AutoField).
     16
     17    When a field is found with an IDENTITY property, it is given a custom field number
     18    of -777, which maps to the 'AutoField' value in the DATA_TYPES_REVERSE dict.
     19    """   
     20    cursor.execute("SELECT TOP 1 * FROM %s" % table_name)
     21    cursor.nextset()
     22    items = []
     23    if identity_check:
     24        for data in cursor.description:
     25            if _is_auto_field(cursor, table_name, data[0]):
     26                data = list(data)
     27                data[1] = -777
     28            items.append(list(data))
     29    else:
     30        items = cursor.description
     31    return items
     32
     33def _name_to_index(cursor, table_name):
     34    """
     35    Returns a dictionary of {field_name: field_index} for the given table.
     36    Indexes are 0-based.
     37    """
     38    return dict([(d[0], i) for i, d in enumerate(get_table_description(cursor, table_name, identity_check=False))])
     39
     40def get_relations(cursor, table_name):
     41    """
     42    Returns a dictionary of {field_index: (field_index_other_table, other_table)}
     43    representing all relationships to the given table. Indexes are 0-based.   
     44    """
     45    table_index = _name_to_index(cursor, table_name)
     46    sql = """SELECT e.COLUMN_NAME AS column_name,
     47                    c.TABLE_NAME AS referenced_table_name,
     48                    d.COLUMN_NAME AS referenced_column_name
     49                    FROM INFORMATION_SCHEMA.TABLE_CONSTRAINTS AS a
     50                        INNER JOIN INFORMATION_SCHEMA.REFERENTIAL_CONSTRAINTS AS b
     51                              ON a.CONSTRAINT_NAME = b.CONSTRAINT_NAME
     52                        INNER JOIN INFORMATION_SCHEMA.CONSTRAINT_TABLE_USAGE AS c
     53                              ON b.UNIQUE_CONSTRAINT_NAME = c.CONSTRAINT_NAME
     54                        INNER JOIN INFORMATION_SCHEMA.CONSTRAINT_COLUMN_USAGE AS d
     55                              ON c.CONSTRAINT_NAME = d.CONSTRAINT_NAME
     56                        INNER JOIN INFORMATION_SCHEMA.CONSTRAINT_COLUMN_USAGE AS e
     57                              ON a.CONSTRAINT_NAME = e.CONSTRAINT_NAME
     58                    WHERE a.TABLE_NAME = ? AND
     59                          a.CONSTRAINT_TYPE = 'FOREIGN KEY'"""
     60    cursor = Cursor(cursor.db.connection)
     61    cursor.execute(sql, (table_name,))
     62    return dict([(table_index[item[0]], (_name_to_index(cursor, item[1])[item[2]], item[1]))
     63                  for item in cursor.fetchall()])
     64   
     65def get_indexes(cursor, table_name):
     66    """
     67    Returns a dictionary of fieldname -> infodict for the given table,
     68    where each infodict is in the format:
     69        {'primary_key': boolean representing whether it's the primary key,
     70         'unique': boolean representing whether it's a unique index}
     71    """
     72    sql = """SELECT b.COLUMN_NAME, a.CONSTRAINT_TYPE
     73               FROM INFORMATION_SCHEMA.TABLE_CONSTRAINTS AS a INNER JOIN
     74                    INFORMATION_SCHEMA.CONSTRAINT_COLUMN_USAGE AS b
     75                    ON a.CONSTRAINT_NAME = b.CONSTRAINT_NAME AND
     76                       a.TABLE_NAME = b.TABLE_NAME
     77               WHERE a.TABLE_NAME = ? AND
     78                     (CONSTRAINT_TYPE = 'PRIMARY KEY' OR
     79                      CONSTRAINT_TYPE = 'UNIQUE')"""
     80    field_names = [item[0] for item in get_table_description(cursor, table_name, identity_check=False)]
     81    cursor = Cursor(cursor.db.connection)
     82    cursor.execute(sql, (table_name,))
     83    indexes = {}
     84    results = {}
     85    data = cursor.fetchall()
     86    if data:
     87        results.update(data)
     88    for field in field_names:
     89        val = results.get(field, None)
     90        indexes[field] = dict(primary_key=(val=='PRIMARY KEY'), unique=(val=='UNIQUE'))
     91    return indexes
     92
     93# A reference for the values below:
     94# http://msdn.microsoft.com/library/default.asp?url=/library/en-us/ado270/htm/mdcstdatatypeenum.asp
     95DATA_TYPES_REVERSE = {
     96# 8192 : Array ,
     97# 128 : Binary ,
     98# 9 : IDispatch ,
     99# 12 : Variant ,
     100# 13 : IUnknown ,
     101# 21  : UnsignedBigInt,
     102# 132 : UserDefined ,
     103# 0   : Empty ,
     104# 136 : Chapter ,
     105# 138 : PropVariant ,
     106# 204 : VarBinary ,
     107# 205 : LongVarBinary ,
     108-777: 'AutoField',                  # Custom number used to identify AutoFields
     1092   : 'SmallIntegerField',          # SmallInt
     1103   : 'IntegerField',               # Integer
     1114   : 'FloatField',                 # Single
     1125   : 'FloatField',                 # Decimal
     1136   : 'FloatField',                 # Currency
     1147   : 'DateField',                  # Date
     1158   : 'CharField',                  # BSTR
     11610  : 'IntegerField',               # Error
     11711  : 'BooleanField',               # Boolean
     11814  : 'FloatField',                 # Decimal
     11916  : 'SmallIntegerField',          # TinyInt
     12017  : 'PositiveSmallIntegerField',  # UnsignedTinyInt
     12118  : 'PositiveSmallIntegerField',  # UnsignedSmallInt
     12219  : 'PositiveIntegerField',       # UnsignedInt
     12320  : 'IntegerField',               # BigInt
     12464  : 'DateTimeField',              # FileTime
     12572  : 'CharField',                  # GUID
     126129 : 'CharField',                  # Char
     127130 : 'CharField',                  # WChar
     128131 : 'FloatField',                 # Numeric
     129133 : 'DateField',                  # DBDate
     130134 : 'TimeField',                  # DBTime
     131135 : 'DateTimeField',              # DBTimeStamp
     132139 : 'FloatField',                 # VarNumeric
     133200 : 'CharField',                  # VarChar
     134201 : 'TextField',                  # LongVarChar
     135202 : 'CharField',                  # VarWChar
     136203 : 'TextField',                  # LongVarWChar
     137}
     138 Pas de fin de ligne
  • db/backends/mssql/creation.py

     la fin du fichier
     
     1DATA_TYPES = {
     2    'AutoField':         'int IDENTITY (1, 1)',
     3    'BooleanField':      'bit',
     4    'CharField':         'varchar(%(max_length)s)',
     5    'CommaSeparatedIntegerField': 'varchar(%(max_length)s)',
     6    'DateField':         'datetime',
     7    'DateTimeField':     'datetime',
     8    'DecimalField':      'numeric(%(max_digits)s, %(decimal_places)s)',
     9    'FileField':         'varchar(254)',
     10    'FilePathField':     'varchar(254)',
     11    'FloatField':        'double precision',
     12    'ImageField':        'varchar(254)',
     13    'IntegerField':      'int',
     14    'IPAddressField':    'char(15)',
     15    'ManyToManyField':   None,
     16    'NullBooleanField':  'bit',
     17    'OneToOneField':     'int',
     18    'PhoneNumberField':  'varchar(20)',
     19    #The check must be unique in for the database. Put random so the regresion test not complain about duplicate names
     20    'PositiveIntegerField': 'int CONSTRAINT [CK_int_pos_%(creation_counter)s_%(column)s] CHECK ([%(column)s] > 0)',   
     21    'PositiveSmallIntegerField': 'smallint CONSTRAINT [CK_smallint_pos_%(creation_counter)s_%(column)s] CHECK ([%(column)s] > 0)',
     22    'SlugField':         'varchar(%(max_length)s)',
     23    'SmallIntegerField': 'smallint',
     24    'TextField':         'text',
     25    'TimeField':         'datetime',
     26    'USStateField':      'varchar(2)',
     27}
  • db/backends/util.py

     
     1from django.conf import settings
    12import datetime
    23import md5
    34from time import time
  • contrib/sessions/middleware.py

     
    6161                self._session_cache = {}
    6262            else:
    6363                try:
     64                    datenow = datetime.datetime.now()
     65                    if hasattr(datenow, 'microsecond'):
     66                        datenow = datenow.replace(microsecond=0)
    6467                    s = Session.objects.get(session_key=self.session_key,
    65                         expire_date__gt=datetime.datetime.now())
     68                        expire_date__gt=datenow)
    6669                    self._session_cache = s.get_decoded()
    6770                except (Session.DoesNotExist, SuspiciousOperation):
    6871                    self._session_cache = {}
Back to Top