Python: How receive values from an input function and have them working as an integer?

Question

I am currently writing a program that receives input from users and processes them in separate functions. MRE:

def prompt_user():
    money1 = input("insert a value")
    money2 = input(" insert another value")
    return money1, money2

def add(money1, money2):
    cash = int(money1) + int(money2)
    return cash

two_values = prompt_user()
total = add(*two_values)
print("you have "+ str(total))

My problem with the code above is that it's really ugly. How am I able to process the data from users WITHOUT writing int() for every variable I used?

score 0 · Answer 1 · answered Dec 03 '20 at 10:31

No, you have to cast the input to integer at a point. Better approach will be to use int() on input() and .format():

def prompt_user():
    money1 = int(input("insert a value"))
    money2 = int(input(" insert another value"))
    return money1, money2

def add(money1, money2):
    cash = money1 + money2
    return cash

two_values = prompt_user()
total = add(*two_values)
print("you have {}".format(total))

Dorian Turba · Accepted Answer · 2020-12-03T10:44:09.807

You can use a map:

def prompt_user():
    money1 = input("insert a value")
    money2 = input(" insert another value")
    return map(int, (money1, money2))

Or you can create a function named input_int(text) to avoid duplication:

def input_int(text=""):
    return int(input(text))

def prompt_user():
    money1 = input_int("insert a value")
    money2 = input_int(" insert another value")
    return money1, money2

Here is a better version of input_int(text):

def input_int(text=""):
    while True:
        try:
            res = int(input(text))
            break
        except ValueError:
            print("The input must be an integer")
    return res

Python: How receive values from an input function and have them working as an integer?

2 Answers2