使用Python字典将用户输入从值更改为键

Question

I have made a dictionary that contains keys and values of different countries. eg

Sample variables:

property = ['Hub','Country','Division']
divisionlist = ['GE']
hublist = ['EUDIV']
countrylist = ['GER', 'GRE', 'HUN']
countrynamelist = ['Germany','Greece','Hungary']

Code to make dictionary:

countrydict ={key:value for key, value in zip(countrynamelist,countrylist)} 

Dictionary visualised:

countrydict = {'Germany': 'GER', 'Greece': 'GRE', 'Hungary': 'HUN'}

Extract from a function:

while True:
        print("Select between a 'Hub','Country' or 'Division'")
        first_property = input('Enter a property name: ').capitalize()
        if first_property in property:
            break
        else:
            continue
    if first_property == 'Hub':
        print('Available hubs: ', hublist)
        first_value = input('Enter a hub name: ').upper()
    if first_property == 'Country':
        country_value = input('Enter a country name: ').capitalize()
        first_value = countrydict[country_value]
    if first_property == 'Division':
        print('Available divisions: ', divisionlist)
        first_value = input('Enter a division name: ').upper()

I'm trying to allow users to input the name of the country instead of the acronym, as it is easier. However I'm getting this error

Traceback (most recent call last):
  File "hello_alerts.py", line 85, in <module>
    alert()
  File "hello_alerts.py", line 50, in alert
    first_value = countrydict[country_value]
KeyError: 'Germany'

Answer 1

Check if the value exists in the countrydict , and if it does not, maybe assign a NA string:

if country_value in countrydict:
    first_value = countrydict[country_value]
else:
    first_value = "NA"
print(first_value)

OUTPUT:

Enter a country name: Germany
GER