codfrm/MoviePilot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix site

Browse Source
This commit is contained in:
jxxghp
2023-07-07 09:54:12 +08:00
parent ecec348886
commit 1b42a382a1
2 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
app/api/endpoints/site.py
View File

@ -29,10 +29,10 @@ async def read_sites(db: Session = Depends(get_db),
""" """
获取站点列表 获取站点列表
""" """
return Site.list(db) return Site.list_order_by_pri(db)
@router.put("/", summary="更新站点", response_model=schemas.Site) @router.put("/", summary="更新站点", response_model=schemas.Response)
async def update_site( async def update_site(
*, *,
db: Session = Depends(get_db), db: Session = Depends(get_db),
@ -44,12 +44,9 @@ async def update_site(
""" """
site = Site.get(db, site_in.id) site = Site.get(db, site_in.id)
if not site: if not site:
raise HTTPException( return schemas.Response(success=False, message="站点不存在")
status_code=404, site.update(db, site_in.dict())
detail=f"站点 {site_in.id} 不存在", return schemas.Response(success=True)
)
site.update(db, **site_in.dict())
return site
@router.delete("/", summary="删除站点", response_model=schemas.Response) @router.delete("/", summary="删除站点", response_model=schemas.Response)

6
app/db/models/site.py
View File

@ -18,7 +18,7 @@ class Site(Base):
# 站点地址 # 站点地址
url = Column(String, nullable=False) url = Column(String, nullable=False)
# 站点优先级 # 站点优先级
pri = Column(Integer) pri = Column(Integer, default=1)
# RSS地址，未启用 # RSS地址，未启用
rss = Column(String) rss = Column(String)
# Cookie # Cookie
@ -53,3 +53,7 @@ class Site(Base):
@staticmethod @staticmethod
def get_actives(db: Session): def get_actives(db: Session):
return db.query(Site).filter(Site.is_active == 1).all() return db.query(Site).filter(Site.is_active == 1).all()
@staticmethod
def list_order_by_pri(db: Session):
return db.query(Site).order_by(Site.pri).all()