11from typing import Optional , Dict , Any , List
22from sqlmodel import Session , select
33from sqlalchemy .exc import IntegrityError
4- from datetime import datetime
4+ from datetime import datetime , timezone
55
66from app .models import Credential , CredsCreate , CredsUpdate
77from app .core .providers import (
@@ -35,7 +35,7 @@ def set_creds_for_org(*, session: Session, creds_add: CredsCreate) -> List[Crede
3535 provider = provider ,
3636 credential = encrypted_credentials ,
3737 )
38- credential .inserted_at = datetime .utcnow ( )
38+ credential .inserted_at = datetime .now ( timezone . utc )
3939 try :
4040 session .add (credential )
4141 session .commit ()
@@ -140,7 +140,7 @@ def update_creds_for_org(
140140 raise ValueError (f"No credentials found for provider { creds_in .provider } " )
141141
142142 creds .credential = encrypted_credentials
143- creds .updated_at = datetime .utcnow ( )
143+ creds .updated_at = datetime .now ( timezone . utc )
144144 session .add (creds )
145145 session .commit ()
146146 session .refresh (creds )
@@ -166,7 +166,7 @@ def remove_provider_credential(
166166
167167 # Soft delete by setting is_active to False
168168 creds .is_active = False
169- creds .updated_at = datetime .utcnow ( )
169+ creds .updated_at = datetime .now ( timezone . utc )
170170
171171 try :
172172 session .add (creds )
@@ -191,7 +191,7 @@ def remove_creds_for_org(
191191
192192 for cred in creds :
193193 cred .is_active = False
194- cred .updated_at = datetime .utcnow ( )
194+ cred .updated_at = datetime .now ( timezone . utc )
195195 session .add (cred )
196196
197197 session .commit ()
0 commit comments