fix(site): resolve site user data update failure

This commit is contained in:
InfinityPacer
2024-10-17 15:00:18 +08:00
parent 52a15086cb
commit b83279b05a

View File

@@ -1,5 +1,5 @@
from datetime import datetime from datetime import datetime
from typing import Tuple, List from typing import List, Tuple
from app.db import DbOper from app.db import DbOper
from app.db.models import SiteIcon from app.db.models import SiteIcon
@@ -117,10 +117,10 @@ class SiteOper(DbOper):
"updated_time": current_time "updated_time": current_time
}) })
# 按站点+天判断是否存在数据 # 按站点+天判断是否存在数据
siteuserdata = SiteUserData.get_by_domain(self._db, domain=domain, workdate=current_day) siteuserdatas = SiteUserData.get_by_domain(self._db, domain=domain, workdate=current_day)
if siteuserdata: if siteuserdatas:
# 存在则更新 # 存在则更新
SiteUserData.update(self._db, payload) siteuserdatas[0].update(self._db, payload)
else: else:
# 不存在则插入 # 不存在则插入
SiteUserData(**payload).create(self._db) SiteUserData(**payload).create(self._db)