Получение аргумента от пользователя и поиск правильного идентификатора

#python

#python

Вопрос:

arg1 = любая из стран в списке, который я собираюсь. Пользователь выбирает их.

arg2 = любая из стран в списке, который я собираюсь. Пользователь выбирает их.

 ukraine = '1'
russia = '0'
finland = '2'
sweden = '3'
usa = '5'
api_key = "xxxxx"
    
resp = requests.get(f"https://hidden.ee/i.php?api_key={api_key}amp;action=naamp;country={arg1}country2={arg2}")

print(resp.url)
 

Единственное, что я получаю правильно, это ключ API, а не arg1 или arg2. Есть ли способ посмотреть и сравнить, соответствует ли arg1 / 2 в моем списке, пожалуйста?

Ответ №1:

Вы можете использовать словарь для поиска идентификатора страны на основе имени, переданного в аргументы

 arg1 = 'ukraine'
arg2 = 'sweden'
# use a dictionary to look up the country
countries = {
    'ukraine': '1',
    'russia': '0',
    'finland': '2',
    'sweden': '3',
    'usa': '5',
}
# assign the country to its id based on the dictionary
country1 = countries[arg1]
country2 = countries[arg2]

api_key = 'xxxxx'

url = f'https://hidden.ee/i.php?api_key={api_key}amp;action=naamp;country={country1}country2={country2}'

# check out the url
print(url)
 

Вывод:

 https://hidden.ee/i.php?api_key=xxxxxamp;action=naamp;country=1country2=3
 

Комментарии:

1. Спасибо, работает как шарм. Ценю, что вы прояснили мою логику.