Как использовать 'def' для получения входного значения

Question

Если я использую глобальные переменные. Я могу писать так.

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.') 

Как определить функцию и получить ввод от этой функции? Получить вход пользователя в начале функции и вызвать главную функцию, чтобы получить вход пользователя.

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

def main(): 
    ... 
    ... 

Answer 1

Вы хотите иметь это в файле (test.py):

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

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

А потом называют это:

$ питон test.py

Примечание две вещи:

1. вход() против raw_input() - вход() является not safe
2. используя idiom 'если имя ...' для script environment вместо основной() функция.

Answer 2

Вид зависит от того, что вы пытаетесь выполнить.

Если ваш метод сбор информации от пользователя, вы можете написать так:

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

Если ваш метод разбора информации от пользователя, он будет принять файл имя в качестве аргумента и быть написано что-то вроде этого:

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