如何使用'def'获取输入值

Question

If I use global variables. I can write like this way.

while True:
    name_str = input('Please give me the file name:')
    if name_str == 'table.csv':
        break
    else:
        print('Bad file name, please try again.')

How to def a function and get input from this function? Get users' input in the beginning function and call main function to get users' input.

def get_name(n):
    x = input('Please give me the file name:')
    ...
    ...

def main():
    ...
    ...

Answer 1

You want to have this in a file (test.py):

def get_input():
    return raw_input('get file: ')

if __name__ == '__main__':
    print get_input()

And then call it:

$ python test.py

Note two things:

1. input() vs raw_input() -- input() is not safe
2. using idiom 'if name ...' for script environment instead of having main() function.

Answer 2

Kind of depends on what you're trying to accomplish.

If your method is gathering the information from the user, you can write it this way:

def get_name():
    return input('Please give me the file name. ')

If your method is parsing information from the user, it would accept the file name as an argument and be written something like this:

def check_file_name(fname):
    if fname == 'table.csv':
        return True
    return False