diff --git a/main.py b/main.py
index f1147be..ce08d41 100644
--- a/main.py
+++ b/main.py
@@ -29,7 +29,9 @@ security = HTTPBasic()
expected_password: str | None = None
-async def get_current_username(credentials: Annotated[HTTPBasicCredentials, Depends(security)]) -> str:
+async def get_current_username(
+ credentials: Annotated[HTTPBasicCredentials, Depends(security)],
+) -> str:
"""Verify Basic Authentication credentials"""
if expected_password is not None and credentials.password != expected_password:
raise HTTPException(
@@ -187,7 +189,9 @@ async def root(username: str = Depends(get_current_username)):
@app.get("/{order}/{delay}")
-async def order_delay(order: str, delay: int, username: str = Depends(get_current_username)):
+async def order_delay(
+ order: str, delay: int, username: str = Depends(get_current_username)
+):
"""Redirect to random file with order and delay"""
random_hash = _get_random_hash()
return RedirectResponse(
@@ -195,13 +199,27 @@ async def order_delay(order: str, delay: int, username: str = Depends(get_curren
)
-def _get_navigation_data(file_hash: str):
- """Get navigation data for a file hash"""
+def _get_navigation_data(file_hash: str, order: str | None = None):
+ """Get navigation data for a file hash.
+
+ Args:
+ file_hash: The current file's hash.
+ order: Navigation order - 'next' for sequential, 'random' for random,
+ or None for default browse mode.
+
+ Returns:
+ Dictionary with navigation hashes and filename.
+ """
keys = list(file_mapping.keys())
idx = keys.index(file_hash)
- next_hash = keys[(idx + 1) % len(keys)]
- prev_hash = keys[idx - 1] if idx > 0 else keys[-1]
- next_random_hash = _get_random_hash()
+
+ if order == "random":
+ next_hash = _get_random_hash()
+ prev_hash = _get_random_hash()
+ else:
+ next_hash = keys[(idx + 1) % len(keys)]
+ prev_hash = keys[idx - 1] if idx > 0 else keys[-1]
+
indexer = _find_indexer_for_hash(file_hash)
filename = indexer.get_filename_by_hash(file_hash) if indexer else ""
@@ -209,7 +227,6 @@ def _get_navigation_data(file_hash: str):
"file_hash": file_hash,
"next_hash": next_hash,
"prev_hash": prev_hash,
- "next_random_hash": next_random_hash,
"filename": filename,
}
@@ -248,10 +265,7 @@ def _render_page(
content = template.substitute(
img_url="/api/{file_hash}/data".format(file_hash=navigation_data["file_hash"]),
- image_click_url=image_click_url
- or "/{next_random_hash}".format(
- next_random_hash=navigation_data["next_random_hash"]
- ),
+ image_click_url=image_click_url or _get_random_hash(),
next_url=next_url,
prev_url=prev_url,
filename=navigation_data["filename"],
@@ -268,7 +282,7 @@ async def hash_page(file_hash: str, username: str = Depends(get_current_username
if file_hash not in file_mapping:
raise HTTPException(status_code=404, detail="File not found")
- navigation_data = _get_navigation_data(file_hash)
+ navigation_data = _get_navigation_data(file_hash, order=None)
play_button = '⏵'.format(
file_hash=file_hash
)
@@ -278,7 +292,12 @@ async def hash_page(file_hash: str, username: str = Depends(get_current_username
@app.get("/{order}/{delay}/{file_hash}")
-async def hash_page_with_refresh(order: str, delay: int, file_hash: str, username: str = Depends(get_current_username)):
+async def hash_page_with_refresh(
+ order: str,
+ delay: int,
+ file_hash: str,
+ username: str = Depends(get_current_username),
+):
"""Serve a page for a specific file hash with auto-refresh navigation"""
if file_hash not in file_mapping:
raise HTTPException(status_code=404, detail="File not found")
@@ -288,18 +307,11 @@ async def hash_page_with_refresh(order: str, delay: int, file_hash: str, usernam
status_code=400, detail="Invalid order. Must be 'next' or 'random'"
)
- navigation_data = _get_navigation_data(file_hash)
+ navigation_data = _get_navigation_data(file_hash, order=order)
- if order == "next":
- refresh_url = "/{order}/{delay}/{next_hash}".format(
- order=order, delay=delay, next_hash=navigation_data["next_hash"]
- )
- else:
- refresh_url = "/{order}/{delay}/{next_random_hash}".format(
- order=order,
- delay=delay,
- next_random_hash=navigation_data["next_random_hash"],
- )
+ refresh_url = "/{order}/{delay}/{next_hash}".format(
+ order=order, delay=delay, next_hash=navigation_data["next_hash"]
+ )
refresh_meta = f''
image_click_url = "/{file_hash}".format(file_hash=file_hash)
@@ -347,7 +359,9 @@ if __name__ == "__main__":
parser.add_argument(
"--salt", type=str, default=None, help="Salt for hashing file paths"
)
- parser.add_argument("--password", type=str, default=None, help="Password for Basic Authentication")
+ parser.add_argument(
+ "--password", type=str, default=None, help="Password for Basic Authentication"
+ )
args = parser.parse_args()
initialize_server(args)