update date_created field default by local time

Cinnamon · Nov 11, 2024 · a5993f3 · a5993f3
1 parent c4a810e
commit a5993f3
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/libs/ktem/ktem/index/file/index.py b/libs/ktem/ktem/index/file/index.py
@@ -1,4 +1,5 @@
 import uuid
+from datetime import datetime
 from typing import Any, Optional, Type
 
 from ktem.components import filestorage_path, get_docstore, get_vectorstore
@@ -7,9 +8,9 @@
 from sqlalchemy import JSON, Column, DateTime, Integer, String, UniqueConstraint
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.ext.mutable import MutableDict
-from sqlalchemy.sql import func
 from theflow.settings import settings as flowsettings
 from theflow.utils.modules import import_dotted_string
+from tzlocal import get_localzone
 
 from kotaemon.storages import BaseDocumentStore, BaseVectorStore
 
@@ -73,7 +74,7 @@ def _setup_resources(self):
                     "path": Column(String),
                     "size": Column(Integer, default=0),
                     "date_created": Column(
-                        DateTime(timezone=True), server_default=func.now()
+                        DateTime(timezone=True), default=datetime.now(get_localzone())
                     ),
                     "user": Column(Integer, default=1),
                     "note": Column(
@@ -98,7 +99,7 @@ def _setup_resources(self):
                     "path": Column(String),
                     "size": Column(Integer, default=0),
                     "date_created": Column(
-                        DateTime(timezone=True), server_default=func.now()
+                        DateTime(timezone=True), default=datetime.now(get_localzone())
                     ),
                     "user": Column(Integer, default=1),
                     "note": Column(
@@ -126,7 +127,7 @@ def _setup_resources(self):
                 "__tablename__": f"index__{self.id}__group",
                 "id": Column(Integer, primary_key=True, autoincrement=True),
                 "date_created": Column(
-                    DateTime(timezone=True), server_default=func.now()
+                    DateTime(timezone=True), default=datetime.now(get_localzone())
                 ),
                 "name": Column(String, unique=True),
                 "user": Column(Integer, default=1),

diff --git a/scripts/migrate/migrate_chroma_db.py b/scripts/migrate/migrate_chroma_db.py
@@ -1,4 +1,5 @@
 import uuid
+from datetime import datetime
 
 import chromadb
 from ktem.index.models import Index
@@ -15,7 +16,7 @@
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.ext.mutable import MutableDict
 from sqlalchemy.orm import Session
-from sqlalchemy.sql import func
+from tzlocal import get_localzone
 
 
 def _init_resource(private: bool = True, id: int = 1):
@@ -41,7 +42,7 @@ def _init_resource(private: bool = True, id: int = 1):
                 "path": Column(String),
                 "size": Column(Integer, default=0),
                 "date_created": Column(
-                    DateTime(timezone=True), server_default=func.now()
+                    DateTime(timezone=True), default=datetime.now(get_localzone())
                 ),
                 "user": Column(Integer, default=1),
                 "note": Column(
@@ -66,7 +67,7 @@ def _init_resource(private: bool = True, id: int = 1):
                 "path": Column(String),
                 "size": Column(Integer, default=0),
                 "date_created": Column(
-                    DateTime(timezone=True), server_default=func.now()
+                    DateTime(timezone=True), default=datetime.now(get_localzone())
                 ),
                 "user": Column(Integer, default=1),
                 "note": Column(