跳转至

Python 进阶教程系列 7：工厂模式

本文是 Python 进阶教程系列 7，主要介绍了 Python 工厂模式。

工厂模式是指调用方可以通过调用一个简单函数就可以创建不同的对象。工厂模式一般包含工厂方法和抽象工厂两种模式。

本文转载了 Python 设计模式：工厂模式（factory pattern）。

工厂方法（factory method）

工厂方法模式是指定义一个可以根据输入参数的不同返回不同对象的函数。

Python
class JSONParser:
    def parse(self, raw_data):
        return json.loads(raw_data)


class XMLParser:
    def parse(self, raw_data):
        return xml2dict(raw_data)


def new_parser(type, **kwargs):
    if type == "json":
        return JSONParser()
    elif type == "xml":
        return XMLParser()


parser = new_parser("json")
with open("hello.json") as fp:
    data = parser.parse(fp.read())
print(data)

抽象工厂（abstract factory）

一系列的工厂方法组合在一起实现了一个抽象工厂。

还是上面那个例子，我们再加一个工厂方法：

Python
class DBSaver:
    def save(self, obj, **kwargs):
        model = Model(**obj)
        model.save()


class FileSaver:
    def __init__(self, save_dir):
        self.save_dir = save_dir

    def save(self, obj, name):
        path = os.path.join(self.save_dir, name)
        with open(path, "wb") as fp:
            data = json.dumps(obj)
            fp.write(data)


def new_saver(type, **kwargs):
    if type == "db":
        return DBSaver()
    elif type == "file":
        save_dir = kwargs["save_dir"]
        return FileSaver(save_dir)


class FileHandler:
    def __init__(self, parse_type, save_type, **kwargs):
        self.parser = new_parser(parse_type, **kwargs)
        self.saver = new_saver(save_type, **kwargs)

    def do(self, data, filename):
        obj = self.parser.parse(data)
        self.saver.save(obj, filename)


handler = FileHandler("json", "file", save_dir="save")
with open("hello.json") as fp:
    data = fp.read()
handler.do(data, "data.json")

工厂方法和抽象工厂的选择：先使用工厂方法，当发现需要使用一系列工厂方法来创建多个对象的时候，可以考虑把这些创建对象的过程合并到一个抽象工厂。

Python 进阶教程系列文章

评论