building static badge url to use in documentation using python

badge.py

#getting static url for badge using shields.io
#build string
#eg: https://shields.io/static/v1?label=flask&message=2.2.2&color=green

#inputs: label , message, color

print('Create shields.io static label\n')
label = input("Label: ")
message = input("Message: ")

colors = ['brightgreen','green','yellowgreen','yellow','orange','red','blue','lightgrey',
'success','important','critical', 'informational', 'inactive', 'blueviolet', 'ff69b4', '9cf']
print('Select color: \n')
for i in range(0, len(colors)):
   print(i, '=>', colors[i])

color = input("Your Color: ")
color = colors[int(color)]
 
build_url = f"https://shields.io/static/v1?label={label}&message={message}&color={color}"
print('Your static badge url: ', build_url)