diff --git a/IzzyLib/misc.py b/IzzyLib/misc.py index c966e99..28dbcf3 100644 --- a/IzzyLib/misc.py +++ b/IzzyLib/misc.py @@ -38,14 +38,20 @@ def Boolean(v, return_value=False): return True -def RandomGen(length=20, chars=None): +def RandomGen(length=20, letters=True, digits=True, extra=None): if not isinstance(length, int): raise TypeError(f'Character length must be an integer, not {type(length)}') - characters = string.ascii_letters + string.digits + characters = '' - if chars: - characters += chars + if letters: + characters += string.ascii_letters + + if digits: + characters += string.digits + + if extra: + characters += extra return ''.join(random.choices(characters, k=length))