GithubBackup/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-07-08 00:01:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix site userdata

Browse Source
This commit is contained in:
jxxghp
2024-09-12 15:44:31 +08:00
parent 53195457c7
commit 73407825f5
4 changed files with 47 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
app/db/models/siteuserdata.py
View File

@@ -52,7 +52,14 @@ class SiteUserData(Base):
@staticmethod
@db_query
def get_by_domain(db: Session, domain: str):
def get_by_domain(db: Session, domain: str, workdate: str = None, worktime: str = None):
if workdate and worktime:
return db.query(SiteUserData).filter(SiteUserData.domain == domain,
SiteUserData.updated_day == workdate,
SiteUserData.updated_time == worktime).all()
elif workdate:
return db.query(SiteUserData).filter(SiteUserData.domain == domain,
SiteUserData.updated_day == workdate).all()
return db.query(SiteUserData).filter(SiteUserData.domain == domain).all()
@staticmethod

23
app/db/site_oper.py
View File

@@ -105,22 +105,29 @@ class SiteOper(DbOper):
"""
更新站点用户数据
"""
site = Site.get_by_domain(self._db, domain)
if not site:
return False, "站点不存在"
# 当前系统日期
current_day = datetime.now().strftime('%Y-%m-%d')
current_time = datetime.now().strftime('%H:%M:%S')
payload.update({
"domain": domain,
"updated_day": datetime.now().strftime('%Y-%m-%d'),
"updated_time": datetime.now().strftime('%H:%M:%S')
"updated_day": current_day,
"updated_time": current_time
})
SiteUserData.update(self._db, payload)
siteuserdata = SiteUserData.get_by_domain(self._db, domain=domain,
workdate=current_day, worktime=current_time)
if siteuserdata:
# 存在则更新
SiteUserData.update(self._db, payload)
else:
# 不存在则插入
SiteUserData(**payload).create(self._db)
return True, "更新站点用户数据成功"
def get_userdata_by_domain(self, domain: str) -> List[SiteUserData]:
def get_userdata_by_domain(self, domain: str, workdate: str = None) -> List[SiteUserData]:
"""
获取站点用户数据
"""
return SiteUserData.get_by_domain(self._db, domain)
return SiteUserData.get_by_domain(self._db, domain=domain, workdate=workdate)
def get_userdata_by_date(self, date: str) -> List[SiteUserData]:
"""