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

refactor: extract global endpoint

Browse Source
This commit is contained in:
arkohut 2024-06-04 16:48:44 +08:00
parent 981793165d
commit e4fa1693d9
1 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
memos/commands.py
View File

@ -15,6 +15,7 @@ app = typer.Typer()
lib_app = typer.Typer() lib_app = typer.Typer()
app.add_typer(lib_app, name="lib") app.add_typer(lib_app, name="lib")
BASE_URL = "http://localhost:8080"
def format_timestamp(timestamp): def format_timestamp(timestamp):
if isinstance(timestamp, str): if isinstance(timestamp, str):
@ -45,7 +46,7 @@ def serve():
@lib_app.command("ls") @lib_app.command("ls")
def ls(): def ls():
response = httpx.get("http://localhost:8080/libraries") response = httpx.get(f"{BASE_URL}/libraries")
libraries = response.json() libraries = response.json()
display_libraries(libraries) display_libraries(libraries)
@ -55,7 +56,7 @@ def add(name: str, folders: List[str]):
absolute_folders = [str(Path(folder).resolve()) for folder in folders] absolute_folders = [str(Path(folder).resolve()) for folder in folders]
response = httpx.post( response = httpx.post(
"http://localhost:8080/libraries", f"{BASE_URL}/libraries",
json={"name": name, "folders": absolute_folders}, json={"name": name, "folders": absolute_folders},
) )
if 200 <= response.status_code < 300: if 200 <= response.status_code < 300:
@ -66,7 +67,7 @@ def add(name: str, folders: List[str]):
@lib_app.command("show") @lib_app.command("show")
def show(library_id: int): def show(library_id: int):
response = httpx.get(f"http://localhost:8080/libraries/{library_id}") response = httpx.get(f"{BASE_URL}/libraries/{library_id}")
if response.status_code == 200: if response.status_code == 200:
library = response.json() library = response.json()
display_libraries([library]) display_libraries([library])
@ -77,7 +78,7 @@ def show(library_id: int):
@lib_app.command("scan") @lib_app.command("scan")
def scan(library_id: int): def scan(library_id: int):
response = httpx.get(f"http://localhost:8080/libraries/{library_id}") response = httpx.get(f"{BASE_URL}/libraries/{library_id}")
if response.status_code != 200: if response.status_code != 200:
print(f"Failed to retrieve library: {response.status_code} - {response.text}") print(f"Failed to retrieve library: {response.status_code} - {response.text}")
return return
@ -129,7 +130,7 @@ def scan(library_id: int):
# Check if the entity already exists # Check if the entity already exists
get_response = httpx.get( get_response = httpx.get(
f"http://localhost:8080/libraries/{library_id}/entities/by-filepath", f"{BASE_URL}/libraries/{library_id}/entities/by-filepath",
params={ params={
"filepath": str(relative_file_path) "filepath": str(relative_file_path)
}, # Use relative path }, # Use relative path
@ -161,7 +162,7 @@ def scan(library_id: int):
) )
# Update the existing entity # Update the existing entity
update_response = httpx.put( update_response = httpx.put(
f"http://localhost:8080/libraries/{library_id}/entities/{existing_entity['id']}", f"{BASE_URL}/libraries/{library_id}/entities/{existing_entity['id']}",
json=new_entity, json=new_entity,
) )
if 200 <= update_response.status_code < 300: if 200 <= update_response.status_code < 300:
@ -179,7 +180,7 @@ def scan(library_id: int):
# Add the new entity # Add the new entity
post_response = httpx.post( post_response = httpx.post(
f"http://localhost:8080/libraries/{library_id}/entities", f"{BASE_URL}/libraries/{library_id}/entities",
json=new_entity, json=new_entity,
) )
if 200 <= post_response.status_code < 300: if 200 <= post_response.status_code < 300:
@ -192,11 +193,11 @@ def scan(library_id: int):
file_count += 1 file_count += 1
# Check for deleted files # Check for deleted files
limit = 200 # Adjust the limit as needed limit = 200
offset = 0 offset = 0
while True: while True:
existing_files_response = httpx.get( existing_files_response = httpx.get(
f"http://localhost:8080/libraries/{library_id}/folders/{folder['id']}/entities", f"{BASE_URL}/libraries/{library_id}/folders/{folder['id']}/entities",
params={"limit": limit, "offset": offset} params={"limit": limit, "offset": offset}
) )
if existing_files_response.status_code != 200: if existing_files_response.status_code != 200:
@ -213,7 +214,7 @@ def scan(library_id: int):
if existing_file["filepath"] not in scanned_files: if existing_file["filepath"] not in scanned_files:
# File has been deleted # File has been deleted
delete_response = httpx.delete( delete_response = httpx.delete(
f"http://localhost:8080/libraries/{library_id}/entities/{existing_file['id']}" f"{BASE_URL}/libraries/{library_id}/entities/{existing_file['id']}"
) )
if 200 <= delete_response.status_code < 300: if 200 <= delete_response.status_code < 300:
tqdm.write(f"Deleted file from library: {existing_file['filepath']}") tqdm.write(f"Deleted file from library: {existing_file['filepath']}")