Basic Python Validation

[Xstrain]

Having difficulty creating error handling when inputting an integer

dvdQuant = input("Enter quantity of DVD: ")

how can I get python to return a print "Error Message" if an integer is not entered instead of
Enter quantity of DVD: gyfj
Traceback (most recent call last):
File "I:/Python/CourseworkVer3.py", line 203, in <module>
addDvd(catalogue)
File "I:/Python/CourseworkVer3.py", line 129, in addDvd
dvdQuant = input("Enter quantity of DVD: ")
File "<string>", line 1, in <module>
NameError: name 'gyfj' is not defined

I have the input put into a dictionary so I can multiply with another integer in another function, if I change input to raw_input then when I try and multiply it doesn't recognise the input as an integer

Ihope this isn't too vague and any help would be appreciated

[Laharl]

The quickest way to do this is to use raw_input to get a string, then loop over each character in said string and see if it is between '0' and '9'. If you clear the loop without any issues, you have an integer.

[elcron]

Use try except to handle errors.

try:
    n = int( raw_input("Enter a number: ") )
    print "n=%s, n squared=%s"%(n, n*n)
except ValueError:
    print "Invalid input."
 

[Subsciber123]

n=raw_input("Number: ")
while not n.isdigit(): # this is a builtin method of all str objects
    print "Error: NaN"
    n=raw_input("Number: ")
n=int(n)
 

I believe that this is the simplest possible way to do it. Looping through the string is tedious and processor intensive, even for short strings. Although this is almost guaranteed to be a short string, you should get into the habit of creating readable and efficient code.

[marcial]

Im also new with python, and what would I have to do if I want the script or the section to restart every time the user inputs wrong information??

[bvdet]

Im also new with python, and what would I have to do if I want the script or the section to restart every time the user inputs wrong information??

Following is an example using a while loop.

while True:
    s = raw_input("Enter an integer")
    try:
        i = int(s)
        break
    except ValueError:
        print 'Invalid input'
 
print 'Your input of %d is an integer.' % i

[marcial]

Following is an example using a while loop.

while True:
    s = raw_input("Enter an integer")
    try:
        i = int(s)
        break
    except ValueError:
        print 'Invalid input'
 
print 'Your input of %d is an integer.' % i

Thanks! I tried to use try-except before without the while loop, and it seems to work very weird, but now with while it works perfect! Thanks!

[fatroshi]

Works great for me 2 :)