tcsenpai/pensieve
1
0
Fork 0
mirror of https://github.com/tcsenpai/pensieve.git synced 2025-06-08 20:25:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: support indexing for specific folders

Browse Source
This commit is contained in:
arkohut 2024-07-26 16:27:01 +08:00
parent 506e5eedd0
commit 4e1ce5ef96
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
memos/commands.py
View File

@ -352,7 +352,10 @@ async def update_entity(
@lib_app.command("index") @lib_app.command("index")
def index(library_id: int): def index(
library_id: int,
folders: List[int] = typer.Option(None, "--folder", "-f"),
):
print(f"Indexing library {library_id}") print(f"Indexing library {library_id}")
# Get the library # Get the library
@ -364,8 +367,16 @@ def index(library_id: int):
library = response.json() library = response.json()
scanned_entities = set() scanned_entities = set()
# Filter folders if the folders parameter is provided
if folders:
library_folders = [
folder for folder in library["folders"] if folder["id"] in folders
]
else:
library_folders = library["folders"]
# Iterate through folders # Iterate through folders
for folder in library["folders"]: for folder in library_folders:
tqdm.write(f"Processing folder: {folder['id']}") tqdm.write(f"Processing folder: {folder['id']}")
# List all entities in the folder # List all entities in the folder