forked from retoor/devplacepy
Compare commits
1
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
7332e4dd59 |
File diff suppressed because one or more lines are too long
+3
-11
@@ -15,27 +15,19 @@ def avatar_seed(user) -> str:
|
|||||||
return user.get("avatar_seed") or user.get("username") or ""
|
return user.get("avatar_seed") or user.get("username") or ""
|
||||||
|
|
||||||
|
|
||||||
def _svg_with_size(svg: str, size: int) -> str:
|
def generate_avatar_svg(seed: str) -> str:
|
||||||
size_attr = f' width="{size}" height="{size}"'
|
|
||||||
if svg.startswith("<svg"):
|
|
||||||
tag_end = svg.index(">")
|
|
||||||
return svg[:tag_end] + size_attr + svg[tag_end:]
|
|
||||||
return svg
|
|
||||||
|
|
||||||
|
|
||||||
def generate_avatar_svg(seed: str, size: int = 128) -> str:
|
|
||||||
try:
|
try:
|
||||||
from multiavatar.multiavatar import multiavatar
|
from multiavatar.multiavatar import multiavatar
|
||||||
|
|
||||||
svg = multiavatar(seed, None, None)
|
svg = multiavatar(seed, None, None)
|
||||||
if svg and svg.strip().startswith("<svg"):
|
if svg and svg.strip().startswith("<svg"):
|
||||||
return _svg_with_size(svg, size)
|
return svg
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.exception(e)
|
logger.exception(e)
|
||||||
logger.warning(f"Avatar generation failed for {seed}: {e}")
|
logger.warning(f"Avatar generation failed for {seed}: {e}")
|
||||||
initial = seed[:1].upper() if seed else "?"
|
initial = seed[:1].upper() if seed else "?"
|
||||||
return (
|
return (
|
||||||
f'<svg xmlns="http://www.w3.org/2000/svg" width="{size}" height="{size}" viewBox="0 0 100 100">'
|
'<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 100 100">'
|
||||||
f'<rect width="100" height="100" rx="50" fill="#ff6b35"/>'
|
f'<rect width="100" height="100" rx="50" fill="#ff6b35"/>'
|
||||||
f'<text x="50" y="65" text-anchor="middle" fill="white" font-size="40" font-weight="700" font-family="sans-serif">{initial}</text></svg>'
|
f'<text x="50" y="65" text-anchor="middle" fill="white" font-size="40" font-weight="700" font-family="sans-serif">{initial}</text></svg>'
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -439,8 +439,8 @@ class VoteForm(BaseModel):
|
|||||||
@field_validator("value")
|
@field_validator("value")
|
||||||
@classmethod
|
@classmethod
|
||||||
def valid_value(cls, value):
|
def valid_value(cls, value):
|
||||||
if value not in (1, -1):
|
if value not in (1, -1, 0):
|
||||||
raise ValueError("value must be 1 or -1")
|
raise ValueError("value must be 1, -1, or 0")
|
||||||
return value
|
return value
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -17,16 +17,14 @@ _CACHE_CONTROL = f"public, max-age={SECONDS_PER_DAY}, immutable"
|
|||||||
|
|
||||||
@router.get("/{style}/{seed}")
|
@router.get("/{style}/{seed}")
|
||||||
async def avatar_proxy(request: Request, style: str, seed: str, size: int = 128):
|
async def avatar_proxy(request: Request, style: str, seed: str, size: int = 128):
|
||||||
size = max(16, min(512, size))
|
etag = '"' + hashlib.md5(f"{seed}:{size}".encode("utf-8")).hexdigest() + '"'
|
||||||
cache_key = f"{seed}:{size}"
|
|
||||||
etag = '"' + hashlib.md5(cache_key.encode("utf-8")).hexdigest() + '"'
|
|
||||||
headers = {"ETag": etag, "Cache-Control": _CACHE_CONTROL}
|
headers = {"ETag": etag, "Cache-Control": _CACHE_CONTROL}
|
||||||
|
|
||||||
if request.headers.get("if-none-match") == etag:
|
if request.headers.get("if-none-match") == etag:
|
||||||
return Response(status_code=304, headers=headers)
|
return Response(status_code=304, headers=headers)
|
||||||
|
|
||||||
svg = _cache.get(cache_key)
|
svg = _cache.get(seed)
|
||||||
if svg is None:
|
if svg is None:
|
||||||
svg = generate_avatar_svg(seed, size)
|
svg = generate_avatar_svg(seed)
|
||||||
_cache.set(cache_key, svg)
|
_cache.set(seed, svg)
|
||||||
return Response(content=svg, media_type="image/svg+xml", headers=headers)
|
return Response(content=svg, media_type="image/svg+xml", headers=headers)
|
||||||
|
|||||||
@@ -22,7 +22,7 @@ ENGAGEMENT_ACTIONS: tuple[Action, ...] = (
|
|||||||
),
|
),
|
||||||
body(
|
body(
|
||||||
"value",
|
"value",
|
||||||
"Vote value: 1 to upvote, -1 to downvote (re-send to remove).",
|
"Vote value: 1 to upvote, -1 to downvote, 0 to retract.",
|
||||||
required=True,
|
required=True,
|
||||||
),
|
),
|
||||||
),
|
),
|
||||||
|
|||||||
+1
-29
@@ -1,6 +1,7 @@
|
|||||||
# retoor <retoor@molodetz.nl>
|
# retoor <retoor@molodetz.nl>
|
||||||
|
|
||||||
import requests
|
import requests
|
||||||
|
from devplacepy.avatar import avatar_url, generate_avatar_svg
|
||||||
from tests.conftest import BASE_URL
|
from tests.conftest import BASE_URL
|
||||||
|
|
||||||
|
|
||||||
@@ -9,32 +10,3 @@ def test_avatar_endpoint_serves_svg(app_server):
|
|||||||
assert r.status_code == 200
|
assert r.status_code == 200
|
||||||
assert "svg" in r.headers.get("content-type", "").lower()
|
assert "svg" in r.headers.get("content-type", "").lower()
|
||||||
assert "<svg" in r.text
|
assert "<svg" in r.text
|
||||||
|
|
||||||
|
|
||||||
def test_avatar_respects_size(app_server):
|
|
||||||
r = requests.get(f"{BASE_URL}/avatar/multiavatar/alice_test?size=32")
|
|
||||||
assert r.status_code == 200
|
|
||||||
assert 'width="32"' in r.text
|
|
||||||
assert 'height="32"' in r.text
|
|
||||||
|
|
||||||
|
|
||||||
def test_avatar_different_sizes_different_response(app_server):
|
|
||||||
r32 = requests.get(f"{BASE_URL}/avatar/multiavatar/bob_test?size=32")
|
|
||||||
r128 = requests.get(f"{BASE_URL}/avatar/multiavatar/bob_test?size=128")
|
|
||||||
assert r32.status_code == 200
|
|
||||||
assert r128.status_code == 200
|
|
||||||
etag32 = r32.headers.get("etag", "")
|
|
||||||
etag128 = r128.headers.get("etag", "")
|
|
||||||
assert etag32 != etag128, "same seed with different sizes must produce different ETags"
|
|
||||||
|
|
||||||
|
|
||||||
def test_avatar_size_clamped_low(app_server):
|
|
||||||
r = requests.get(f"{BASE_URL}/avatar/multiavatar/clamp_low?size=1")
|
|
||||||
assert r.status_code == 200
|
|
||||||
assert 'width="16"' in r.text
|
|
||||||
|
|
||||||
|
|
||||||
def test_avatar_size_clamped_high(app_server):
|
|
||||||
r = requests.get(f"{BASE_URL}/avatar/multiavatar/clamp_high?size=9999")
|
|
||||||
assert r.status_code == 200
|
|
||||||
assert 'width="512"' in r.text
|
|
||||||
|
|||||||
@@ -135,3 +135,13 @@ def test_non_ajax_vote_redirects_to_referer(app_server):
|
|||||||
)
|
)
|
||||||
assert r.status_code == 302
|
assert r.status_code == 302
|
||||||
assert r.headers["location"] == "/gists"
|
assert r.headers["location"] == "/gists"
|
||||||
|
|
||||||
|
|
||||||
|
def test_explicit_zero_retracts_vote(app_server):
|
||||||
|
s_a, a_name = _session_votes()
|
||||||
|
s_b, _ = _session_votes()
|
||||||
|
post_uid = _make_post_votes(_uid_votes(a_name))
|
||||||
|
s_b.post(f"{BASE_URL}/votes/post/{post_uid}", data={"value": "1"}, headers=AJAX_votes)
|
||||||
|
r = s_b.post(f"{BASE_URL}/votes/post/{post_uid}", data={"value": "0"}, headers=AJAX_votes)
|
||||||
|
assert r.json()["value"] == 0
|
||||||
|
assert r.json()["net"] == 0
|
||||||
|
|||||||
Reference in New Issue
Block a user