fix(auth): resolve conflicts

This commit is contained in:
InfinityPacer
2024-10-19 18:03:18 +08:00
7 changed files with 27 additions and 10 deletions

View File

@@ -41,6 +41,7 @@ async def login_access_token(
),
token_type="bearer",
super_user=user_or_message.is_superuser,
user_id=user_or_message.id,
user_name=user_or_message.name,
avatar=user_or_message.avatar,
level=level

View File

@@ -56,9 +56,9 @@ def save(name: str,
@router.post("/list", summary="所有目录和文件", response_model=List[schemas.FileItem])
def list(fileitem: schemas.FileItem,
sort: str = 'updated_at',
_: User = Depends(get_current_active_superuser)) -> Any:
def list_files(fileitem: schemas.FileItem,
sort: str = 'updated_at',
_: User = Depends(get_current_active_superuser)) -> Any:
"""
查询当前目录下所有目录和文件
:param fileitem: 文件项

View File

@@ -174,7 +174,7 @@ def set_config(key: str, value: Union[list, dict, bool, int, str] = None,
@router.delete("/id/{user_id}", summary="删除用户", response_model=schemas.Response)
def delete_user_from_user_id(
def delete_user_by_id(
*,
db: Session = Depends(get_db),
user_id: int,
@@ -191,7 +191,7 @@ def delete_user_from_user_id(
@router.delete("/name/{user_name}", summary="删除用户", response_model=schemas.Response)
def delete_user_from_user_id(
def delete_user_by_name(
*,
db: Session = Depends(get_db),
user_name: str,
@@ -208,7 +208,7 @@ def delete_user_from_user_id(
@router.get("/{username}", summary="用户详情", response_model=schemas.User)
def read_user_by_id(
def read_user_by_name(
username: str,
current_user: User = Depends(get_current_active_user),
db: Session = Depends(get_db),

View File

@@ -1106,7 +1106,9 @@ class FileManagerModule(_ModuleBase):
continue
media_files = self.list_files(fileitem, True)
if media_files:
ret_fileitems.extend(media_files)
for media_file in media_files:
if f".{media_file.extension.lower()}" in settings.RMT_MEDIAEXT:
ret_fileitems.append(media_file)
return ret_fileitems
def media_exists(self, mediainfo: MediaInfo, **kwargs) -> Optional[ExistMediaInfo]:

View File

@@ -159,6 +159,7 @@ class AliPan(StorageBase):
"updateTime": time.time(),
})
self.__update_params(data)
self.__update_drives()
self.__init_aligo()
except Exception as e:
return {}, f"bizExt 解码失败:{str(e)}"
@@ -181,6 +182,17 @@ class AliPan(StorageBase):
"""
return self.aligo.get_user()
def __update_drives(self):
"""
更新用户存储根目录
"""
drivers = self.aligo.list_my_drives()
for driver in drivers:
if driver.category == "resource":
self.__update_params({"resourceDriveId": driver.drive_id})
elif driver.category == "backup":
self.__update_params({"backDriveId": driver.drive_id})
def __get_fileitem(self, fileinfo: BaseFile, parent: str = "/") -> schemas.FileItem:
"""
获取文件信息
@@ -231,7 +243,7 @@ class AliPan(StorageBase):
return [
schemas.FileItem(
storage=self.schema.value,
fileid=fileitem.fileid,
fileid="root",
drive_id=self.__auth_params.get("resourceDriveId"),
parent_fileid="root",
type="dir",
@@ -241,7 +253,7 @@ class AliPan(StorageBase):
),
schemas.FileItem(
storage=self.schema.value,
fileid=fileitem.fileid,
fileid="root",
drive_id=self.__auth_params.get("backDriveId"),
parent_fileid="root",
type="dir",

View File

@@ -335,7 +335,7 @@ class Rclone(StorageBase):
ret = subprocess.run(
[
'rclone', 'about',
'MP:/', '--json'
'/', '--json'
],
capture_output=True,
startupinfo=self.__get_hidden_shell()

View File

@@ -10,6 +10,8 @@ class Token(BaseModel):
token_type: str
# 超级用户
super_user: bool
# 用户ID
user_id: int
# 用户名
user_name: str
# 头像