add type convesrions for legacy drivers from string to datetime

Author: michiya

sql_server/pyodbc/operations.py

@@ -1,6 +1,7 @@
 import datetime
 import time
 import uuid
+import warnings
 
 from django.conf import settings
 from django.db.backends.base.operations import BaseDatabaseOperations
@@ -33,6 +34,15 @@ def _get_utcoffset(self, tzname):
         delta = zone.localize(now, is_dst=False).utcoffset()
         return delta.days * 86400 + delta.seconds
 
+    def _warn_legacy_driver(self, sqltype):
+        warnings.warn(
+            'A %s value was received as a string. This is because you '
+            'are now using a legacy ODBC driver which does not support '
+            'this data type while your database has been migrated using it. '
+            'You should upgrade your ODBC driver for consistency with your '
+            'database migration.' % sqltype,
+            RuntimeWarning)
+
     def bulk_batch_size(self, fields, objs):
         """
         Returns the maximum allowed batch size for the backend. The fields
@@ -83,19 +93,40 @@ def combine_expression(self, connector, sub_expressions):
 
     def convert_datefield_value(self, value, expression, connection, context):
         if value is not None:
-            if self.connection.use_legacy_datetime:
+            # WDAC and old FreeTDS receive a date value as a string
+            # http://blogs.msdn.com/b/sqlnativeclient/archive/2008/02/27/microsoft-sql-server-native-client-and-microsoft-sql-server-2008-native-client.aspx
+            if isinstance(value, string_types):
+                self._warn_legacy_driver('date')
+                value = datetime.date(*map(lambda x: int(x), value.split('-')))
+            elif self.connection.use_legacy_datetime:
                 if isinstance(value, datetime.datetime):
                     value = value.date() # extract date
         return value
 
+    def convert_datetimefield_value(self, value, expression, connection, context):
+        if value is not None:
+            # WDAC and old FreeTDS receive a datetime2 value as a string
+            # http://blogs.msdn.com/b/sqlnativeclient/archive/2008/02/27/microsoft-sql-server-native-client-and-microsoft-sql-server-2008-native-client.aspx
+            if isinstance(value, string_types):
+                self._warn_legacy_driver('datetime2')
+                value = datetime.datetime.strptime(value[:26], '%Y-%m-%d %H:%M:%S.%f')
+                if settings.USE_TZ:
+                    value = timezone.make_aware(value, timezone.utc)
+        return value
+
     def convert_floatfield_value(self, value, expression, connection, context):
         if value is not None:
             value = float(value)
         return value
 
     def convert_timefield_value(self, value, expression, connection, context):
         if value is not None:
-            if self.connection.use_legacy_datetime:
+            # WDAC and old FreeTDS receive a time value as a string
+            # http://blogs.msdn.com/b/sqlnativeclient/archive/2008/02/27/microsoft-sql-server-native-client-and-microsoft-sql-server-2008-native-client.aspx
+            if isinstance(value, string_types):
+                self._warn_legacy_driver('time')
+                value = datetime.time(*map(lambda x: int(x), value[:15].replace('.', ':').split(':')))
+            elif self.connection.use_legacy_datetime:
                 if (isinstance(value, datetime.datetime) and value.year == 1900 and value.month == value.day == 1):
                     value = value.time() # extract time
         return value
@@ -194,6 +225,8 @@ def get_db_converters(self, expression):
         internal_type = expression.output_field.get_internal_type()
         if internal_type == 'DateField':
             converters.append(self.convert_datefield_value)
+        if internal_type == 'DateTimeField':
+            converters.append(self.convert_datetimefield_value)
         elif internal_type == 'FloatField':
             converters.append(self.convert_floatfield_value)
         elif internal_type == 'TimeField':
@@ -392,7 +425,7 @@ def value_to_db_datetime(self, value):
             return None
         if settings.USE_TZ and timezone.is_aware(value):
             # pyodbc donesn't support datetimeoffset
-            value = value.astimezone(timezone.utc)
+            value = value.astimezone(timezone.utc).replace(tzinfo=None)
         if not self.connection.features.supports_microsecond_precision:
             value = value.replace(microsecond=0)
         return value