From 2d1b1491b0a107a19264979dbf8f5ab23adfacbd Mon Sep 17 00:00:00 2001 From: julius Date: Sun, 26 Jan 2025 11:38:16 +0100 Subject: [PATCH] fix: utctime for `Chemistry` --- main.py | 23 ++++++++++++++++------- 1 file changed, 16 insertions(+), 7 deletions(-) diff --git a/main.py b/main.py index b4414bc..4a7c0bd 100644 --- a/main.py +++ b/main.py @@ -1,7 +1,16 @@ from datetime import datetime, timezone -from fastapi import FastAPI, status +from fastapi import APIRouter, FastAPI, status from fastapi.staticfiles import StaticFiles -from sqlmodel import ARRAY, Column, SQLModel, Field, Session, String, create_engine +from sqlmodel import ( + ARRAY, + Column, + SQLModel, + Field, + Session, + String, + create_engine, + select, +) from fastapi.middleware.cors import CORSMiddleware with open("db.secrets", "r") as f: @@ -26,19 +35,19 @@ app.add_middleware( ) +def utctime(): + return datetime.now(tz=timezone.utc) + + class Chemistry(SQLModel, table=True): id: int | None = Field(default=None, primary_key=True) - time: datetime | None = Field(default_factory=datetime.now) + time: datetime | None = Field(default_factory=utctime) user: str love: list[str] = Field(sa_column=Column(ARRAY(String))) hate: list[str] = Field(sa_column=Column(ARRAY(String))) undecided: list[str] = Field(sa_column=Column(ARRAY(String))) -def utctime(): - return datetime.now(tz=timezone.utc) - - class MVPRanking(SQLModel, table=True): id: int | None = Field(default=None, primary_key=True) time: datetime | None = Field(default_factory=utctime)