Ruff format/ Cleanup

This commit is contained in:
Vladless
2026-04-14 07:19:14 +00:00
parent 0e8b0d04f0
commit 39dd2432cc
224 changed files with 1583 additions and 1182 deletions
+4 -4
View File
@@ -22,7 +22,7 @@ def encode_referral_code(user_id: int) -> str:
raw = int(user_id).to_bytes(8, byteorder="big", signed=False)
secret = _secret_bytes()
mask = hmac.new(secret, b"ref-mask-v1", hashlib.sha256).digest()[:8]
obfuscated = bytes(a ^ b for a, b in zip(raw, mask))
obfuscated = bytes(a ^ b for a, b in zip(raw, mask, strict=False))
signature = hmac.new(secret, b"ref-sign-v1:" + obfuscated, hashlib.sha256).digest()[:6]
payload = base64.urlsafe_b64encode(obfuscated + signature).decode("ascii").rstrip("=")
return f"r1_{payload}"
@@ -34,7 +34,7 @@ def encode_partner_code(user_id: int) -> str:
raw = int(user_id).to_bytes(8, byteorder="big", signed=False)
secret = _secret_bytes()
mask = hmac.new(secret, b"partner-mask-v1", hashlib.sha256).digest()[:8]
obfuscated = bytes(a ^ b for a, b in zip(raw, mask))
obfuscated = bytes(a ^ b for a, b in zip(raw, mask, strict=False))
signature = hmac.new(secret, b"partner-sign-v1:" + obfuscated, hashlib.sha256).digest()[:6]
payload = base64.urlsafe_b64encode(obfuscated + signature).decode("ascii").rstrip("=")
return f"p1_{payload}"
@@ -58,7 +58,7 @@ def decode_referral_code(value: str | None) -> int | None:
if not hmac.compare_digest(signature, expected):
return None
mask = hmac.new(secret, b"ref-mask-v1", hashlib.sha256).digest()[:8]
raw = bytes(a ^ b for a, b in zip(obfuscated, mask))
raw = bytes(a ^ b for a, b in zip(obfuscated, mask, strict=False))
parsed = int.from_bytes(raw, byteorder="big", signed=False)
return parsed if parsed > 0 else None
if token.startswith("p1_"):
@@ -88,7 +88,7 @@ def decode_partner_code(value: str | None) -> int | None:
if not hmac.compare_digest(signature, expected):
return None
mask = hmac.new(secret, b"partner-mask-v1", hashlib.sha256).digest()[:8]
raw = bytes(a ^ b for a, b in zip(obfuscated, mask))
raw = bytes(a ^ b for a, b in zip(obfuscated, mask, strict=False))
parsed = int.from_bytes(raw, byteorder="big", signed=False)
return parsed if parsed > 0 else None
if token.startswith("r1_"):