Просмотр исходного кода

Use the naive datetime format for MySQL as well

See the SQLAlchemy docs for more information:
http://docs.sqlalchemy.org/en/latest/dialects/mysql.html#sqlalchemy.dial
ects.mysql.DATETIME
sh4nks 8 лет назад
Родитель
Сommit
5837df594f
1 измененных файлов с 3 добавлено и 3 удалено
  1. 3 3
      flaskbb/utils/database.py

+ 3 - 3
flaskbb/utils/database.py

@@ -36,8 +36,8 @@ class UTCDateTime(db.TypeDecorator):
     def process_bind_param(self, value, dialect):
         """Way into the database."""
         if value is not None:
-            # store naive datetime for sqlite
-            if dialect.name == "sqlite":
+            # store naive datetime for sqlite and mysql
+            if dialect.name in ("sqlite", "mysql"):
                 return value.replace(tzinfo=None)
 
             return value.astimezone(pytz.UTC)
@@ -45,7 +45,7 @@ class UTCDateTime(db.TypeDecorator):
     def process_result_value(self, value, dialect):
         """Way out of the database."""
         # convert naive datetime to non naive datetime
-        if dialect.name == "sqlite" and value is not None:
+        if dialect.name in ("sqlite", "mysql") and value is not None:
             return value.replace(tzinfo=pytz.UTC)
 
         # other dialects are already non-naive