File size: 1,106 Bytes
89419ea | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | """Helpers for Temp-Mail local-part generation and domain choice."""
from __future__ import annotations
import random
import string
from typing import Optional, Sequence
def _rng(rng: Optional[random.Random] = None) -> random.Random:
return rng or random.SystemRandom()
def generate_local_part(*, enable_prefix: bool = False, rng: Optional[random.Random] = None) -> str:
"""Generate a variable-length alphanumeric local-part.
Keeps the previous style (letters+digits+letters), but defaults to no tmp prefix.
"""
r = _rng(rng)
letters = ''.join(r.choices(string.ascii_lowercase, k=5))
digits = ''.join(r.choices(string.digits, k=r.randint(1, 3)))
suffix = ''.join(r.choices(string.ascii_lowercase, k=r.randint(1, 3)))
base = f"{letters}{digits}{suffix}"
return f"tmp{base}" if enable_prefix else base
def choose_domain(domains: Sequence[str], *, rng: Optional[random.Random] = None) -> str:
"""Choose one domain from candidate pool."""
if not domains:
raise ValueError("domains is empty")
r = _rng(rng)
return r.choice(list(domains))
|