Python 函数回调和 if 问题

Question

I just ended to write my Python code for an assignment but I'm a bit stuck on a last point, could you help me to figure out why ?

Here is my code :

import random

def get_rand_int(arg1, arg2):

    rand = random.randint(arg1, arg2)
    if float(rand) == float(arg1):
        return None
    elif float(rand) == float(arg2):
        return None
    else:
        return rand 

def print_random():
    try :
        prompt = int(input("Please enter an integer greater than 0:"))
        assert(prompt > 0)

    except:
        print("Invalid input: You must input an integer greater than 0 for your input.")

    rand = get_rand_int(0, prompt)

    if rand > 0: 
        print( "Your random number is ",rand)
    elif rand is None:
        print( " Please Launch the program again")

The second function call the first one to generate a random number according to the user prompt. Everything work except in the case that the generate random number is the same than one of the argument, my program should print " Please launch the program again". However its not working and nothing happen in this situation ( nothing is printed ) Do you have any idea to how make properly work this last if loop?

Thanks

Answer 1

Try this solution:

import random
def get_rand_int(arg1, arg2):

    rand = random.randint(arg1, arg2)
    #print("Test", rand)
    if rand == arg1:
        return None
    elif rand == arg2:
        return None
    else:
        return rand 

def print_random():
    try :
        prompt = int(input("Please enter an integer greater than 0:"))
        assert(prompt > 0)

    except:
        print("Invalid input: You must input an integer greater than 0 for your input.")

    rand = get_rand_int(0, prompt)

    if rand != None: 
        print( "Your random number is ",rand)
    else:
        print( " Please Launch the program again")

print_random()

More Optimal Solution:

import random
def get_rand_int(arg1, arg2):
    rand = random.randint(arg1, arg2)
    if rand != arg1 or rand != arg2:
        return rand
    return None

def print_random():
    try :
        prompt = int(input("Please enter an integer greater than 0:"))
        assert(prompt > 0)

    except:
        print("Invalid input: You must input an integer greater than 0 for your input.")

    rand = get_rand_int(0, prompt)

    if rand != None: 
        print( "Your random number is ",rand)
    else:
        print( " Please Launch the program again")

print_random()

Output:

Please enter an integer greater than 0:7                                                                                                                                                                                            
Your random number is  2 

Another Output:

Please enter an integer greater than 0:0                                                                              
Invalid input: You must input an integer greater than 0 for your input.                                               
Please Launch the program again