80486DX2-66/gists
1
0
Fork 0
mirror of https://gitlab.com/80486DX2-66/gists synced 2024-11-09 21:32:02 +05:30
Code

Python: generate_C_lang_token.py: support STDIN as an argument

Browse Source
This commit is contained in:
Intel A80486DX2-66 2024-02-24 21:26:12 +03:00
parent b965674095
commit 19a84731d8
Signed by: 80486DX2-66
GPG Key ID: 83631EF27054609B
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
python-programming/generate_C_lang_token.py
View File

@ -7,6 +7,7 @@
from random import choice
from string import ascii_lowercase, ascii_uppercase, digits
from sys import argv, stdin
first_char_alphabet = ascii_uppercase + ascii_lowercase + "_"
alphabet = first_char_alphabet + digits
@ -14,6 +15,10 @@ alphabet = first_char_alphabet + digits
random_token = lambda n: \
"".join( \
[choice(first_char_alphabet)] + \
[choice(alphabet) for i in range(n - 1)]) if n > 0 else ""
[choice(alphabet) for i in range(n - 1)]) if (n := int(n)) > 0 else ""
if "-" in argv[1:] and stdin:
print(random_token(list(stdin)[0]))
else:
print("Random token:", random_token(input("Token length? ")))
print("Random token:", random_token(int(input("Token length? "))))