mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-06-28 23:52:09 -04:00
Tal.Yuan
8066a8e0cd
Move the gallery route domain into routes/gallery/ while preserving backward-compatible legacy import shims. - app imports the canonical gallery route module - canonical gallery route code imports canonical gallery helpers - legacy gallery route/helper paths remain compatibility aliases - add shim regression coverage for module identity and monkeypatch behavior - repoint gallery source-introspection tests to the canonical paths No intended behavior change.
43 lines
1.0 KiB
Python
43 lines
1.0 KiB
Python
import ast
|
|
from pathlib import Path
|
|
|
|
|
|
GATED_IMAGE_FUNCTIONS = {
|
|
"gallery_ai_upscale",
|
|
"gallery_style_transfer",
|
|
"inpaint_proxy",
|
|
"harmonize_image",
|
|
"denoise_image",
|
|
"upscale_image_local",
|
|
"remove_background",
|
|
"enhance_face",
|
|
}
|
|
|
|
|
|
def _gallery_source():
|
|
return Path("routes/gallery/gallery_routes.py").read_text(encoding="utf-8")
|
|
|
|
|
|
def _function_sources(source):
|
|
tree = ast.parse(source)
|
|
return {
|
|
node.name: ast.get_source_segment(source, node) or ""
|
|
for node in ast.walk(tree)
|
|
if isinstance(node, (ast.FunctionDef, ast.AsyncFunctionDef))
|
|
}
|
|
|
|
|
|
def test_image_generation_endpoints_require_image_privilege():
|
|
source = _gallery_source()
|
|
functions = _function_sources(source)
|
|
|
|
for name in GATED_IMAGE_FUNCTIONS:
|
|
assert name in functions
|
|
assert 'require_privilege(request, "can_generate_images")' in functions[name]
|
|
|
|
|
|
def test_gallery_routes_imports_privilege_helper():
|
|
source = _gallery_source()
|
|
assert "get_current_user" in source
|
|
assert "require_privilege" in source
|