tcsenpai/pensieve
1
0
Fork 0
mirror of https://github.com/tcsenpai/pensieve.git synced 2025-06-07 03:35:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(scan): support scan any sub path

Browse Source
This commit is contained in:
arkohut 2024-10-25 11:40:27 +08:00
parent cc24ccf11e
commit a7ce861751
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
memos/cmds/library.py
View File

@ -362,10 +362,15 @@ async def loop_files(library_id, folder, folder_path, force, plugins):
@lib_app.command("scan") @lib_app.command("scan")
def scan( def scan(
library_id: int, library_id: int,
path: str = typer.Argument(None, help="Path to scan within the library"),
force: bool = False, force: bool = False,
plugins: List[int] = typer.Option(None, "--plugin", "-p"), plugins: List[int] = typer.Option(None, "--plugin", "-p"),
folders: List[int] = typer.Option(None, "--folder", "-f"), folders: List[int] = typer.Option(None, "--folder", "-f"),
): ):
# Check if both path and folders are provided
if path and folders:
print("Error: You cannot specify both a path and folders at the same time.")
return
response = httpx.get(f"{BASE_URL}/libraries/{library_id}") response = httpx.get(f"{BASE_URL}/libraries/{library_id}")
if response.status_code != 200: if response.status_code != 200:
@ -385,6 +390,20 @@ def scan(
else: else:
library_folders = library["folders"] library_folders = library["folders"]
# Check if a specific path is provided
if path:
path = Path(path).expanduser().resolve()
# Check if the path is a folder or a subdirectory of a library folder
folder = next(
(folder for folder in library_folders if path.is_relative_to(Path(folder["path"]).resolve())),
None
)
if not folder:
print(f"Error: The path {path} is not part of any folder in the library.")
return
# Only scan the specified path
library_folders = [{"id": folder["id"], "path": str(path)}]
for folder in library_folders: for folder in library_folders:
folder_path = Path(folder["path"]) folder_path = Path(folder["path"])
if not folder_path.exists() or not folder_path.is_dir(): if not folder_path.exists() or not folder_path.is_dir():