Python Jupyter 中函数输入的提示

Question

是否可以添加某种装饰器或别名来获得有关变量可以在函数中获得的可能输入的提示？

在下面的示例函数中 input_var 可以取任何值，但我可以提示使用值 'year'、'month' 或 'day' 吗？

def some_function(input_var):
    """ This function does something cool, use input_var as bla, blah or blahh """
    if input_var=='year':
        print("Selected 1")
    if input_var=='month':
        print("Selected 2")
    if input_var=='day':
        print("Selected 3")

到目前为止，我将“提示”添加到文档字符串中，但我想要这样的东西：

它可以在哪里显示 input_var='year/month/day' 或类似的内容。

使用 Jupyter 笔记本，python 3

谢谢

Answer 1

• 看函数注解
• 示例：

def foo(x: 'year/month/day'):
  return "abc"

• 现在，如果您键入 x 并按 Tab 键获取建议，它将显示 x : 'year/month/day'。