I want to make it like this:
>>> myfunc("strawberry")
ok
# myfunc only works with strawberry
I know that most people will answer with:
def myfunc(something):
if something == "strawberry":
print("ok")
But I want to do all this in the parameter setting.
Kind of like this:
def myfunc(something: OnlyThese["strawberry", "cake"]):
print("ok")
Although the code above is incorrect, I want to see if Python already has a feature like this.
Sound like you are looking for enum. If you are using Python 3.11 or later, there's the StrEnum class:
from enum import StrEnum
class Something(StrEnum):
strawberry = "strawberry"
cake = "cake"
def myfunc(something: Something):
something = Something(something)
print(something)
# OK
myfunc(Something.strawberry)
myfunc("cake")
# ValueError
myfunc("blackberry")
Inside your function something is a subclass of str so it can be treated as a string almost every where.
StrEnum the best practice? Thks
Don't believe there is a way to do what you are wanting to do without writing code in the function body.
I found answers to a similar question at
typing.Literalis the standard annotation for what you tried to do withOnlyThese.