python - 设计我的票务 API

Question

我有一个名为 getTicket 的函数，它接受两个参数 id，即数字和格式(字符串)

def getTicket(id, format):
    if format == "pdf":
        getTicketPDF(id) #some specialized pdf method gets called
    elif format == "json":
        getTicketJSON(id) #specialized json method

现在，如果我必须支持一些新格式，例如“html”，那么我可以为 html 创建另一个 elif。

但是我想概括这个代码，这样如果将来添加新方法，我就不必更改我的代码

如何设计我的 getTicket api？

Answer 1

您可以创建一个字典来存储 format 到函数映射，例如 "pdf" 映射到函数 getTicketPDF 等。然后在您的 getTicket() 函数中，您调用字典的 format 值，并通过向其传递 id 参数来调用它。示例-

funcdict = {"pdf":getTicketPDF
           "json":getTicketJSON}
def getTicket(id, format):
    try:
        funcdict[format](id)
    except KeyError:
        #Handle case where format is not found in dictionary

如果稍后您决定为新格式添加新函数，则只需向字典添加新映射即可。