Compare a7d6e5e ... +5 ... d8acd41

from . import contrib  # noqa:F401

from typing import Optional

import hashlib

from ..totp import TOTP

STEAM_CHARS = "23456789BCDFGHJKMNPQRTVWXY"  # steam's custom alphabet

STEAM_DEFAULT_DIGITS = 5  # Steam TOTP code length

class Steam(TOTP):

    def __init__(self, s: str, name: Optional[str] = None,

        self.interval = interval

        super().__init__(s=s, digits=10, digest=hashlib.sha1, name=name, issuer=issuer)

    def generate_otp(self, input: int) -> str:

        str_code = super().generate_otp(input)

        int_code = int(str_code)

        steam_code = ""

        total_chars = len(STEAM_CHARS)

        for _ in range(STEAM_DEFAULT_DIGITS):

            pos = int_code % total_chars

            char = STEAM_CHARS[int(pos)]

            steam_code += char

            int_code //= total_chars

        return steam_code

from .steam import Steam  # noqa:F401