mirror of
https://github.com/rjNemo/pass-gen
synced 2026-06-11 21:06:40 +00:00
40 lines
927 B
Python
40 lines
927 B
Python
import random
|
|
import string
|
|
from typing import Protocol
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class PassGenOptions(BaseModel):
|
|
seed: int
|
|
length: int = 8
|
|
symbols: bool = False
|
|
numbers: bool = True
|
|
|
|
|
|
def generate_password(options: PassGenOptions) -> str:
|
|
characters = _build_characters(symbols=options.symbols, numbers=options.numbers)
|
|
|
|
random_generator = _new_random_generator(options.seed)
|
|
|
|
return "".join(random_generator.sample(characters, options.length))
|
|
|
|
|
|
class RandomSampler(Protocol):
|
|
def sample(self, population: str, k: int) -> list[str]:
|
|
... # pragma nocover
|
|
|
|
|
|
def _new_random_generator(seed: int) -> RandomSampler:
|
|
return random.Random(seed)
|
|
|
|
|
|
def _build_characters(symbols: bool, numbers: bool) -> str:
|
|
characters = string.ascii_letters
|
|
|
|
if symbols:
|
|
characters += string.punctuation
|
|
if numbers:
|
|
characters += string.digits
|
|
|
|
return characters
|