第七章 重载

开发的过程中,我们很有可能会碰到这样的场景,需要修改官方定义的某个模块中的某个方法。虽然我们可以直接修改源码达到目的,但那样会污染官方源码,不方便于维护和移植。

重载的概念就是通过继承,重写原有模块的某些方法,拦截原有的执行路径,达到修改或是改造的目的。

简单的重载

接下来,我们先看一下,如何对一个既有模块的方法进行重载的操作,还是以我们的book_store模块为例,我们先给book_store.book对象添加要给新的类方法:log_book_name

@api.one
def log_book_name(self):
    _logger.info(f"图书名称:{self.name}")

非常简单,就是打印一下本书的名字,然后为了方便测试,我们在使用odoorpc远程调用一下这个方法:

odoo = odoorpc.ODOO(host=host, port=port)
odoo.login(db, user, password)

# 调用log_book_name方法

book_obj = odoo.env['book_store.book']
book = book_obj.browse(1)

book.log_book_name()

运行测试,可以看到,当我们修改book的内容时,后台日志会输出本图书的名字。

log

重载

现在,我们希望能够把作者的名字也添加到日志中,但是不修改Book类的代码。这时候我们需要新建一个类,继承自Book(这里为了好区别,新建要给代码文件,名为SubModel.py):

class SubBook(models.Model):

    _inherit = "book_store.book"

    @api.one
    def log_book_name(self):
        """重载父类方法"""
        _logger.info(f"图书名称{self.name},作者:{self.author.name}")

重启升级模块儿后,通过rpc调用log_book_name方法可以看到,输出的日志已经带有作者信息了:

log2

复用父类调用的结果

如果希望在父类方法返回的结果上进行修改,可以通过super方法进行调用,这个例子中由于父类方法并不返回值,因此只是调用了一下父类,并没有真正的复用:

@api.one
def log_book_name(self):
    """重载父类方法"""
    res = super(SubBook,self).log_book_name()
    _logger.info(f"图书名称{self.name},作者:{self.author.name}")

可以看到,父类方法也被调用:

log3

私有方法不会被重载

有个例外,就是如果你的方法是私有方法(双下划线开头),那么重载将不会起作用。

class Book(models.Model):
    ...
    @api.one
    def log_book_name(self):
        self.__log_book_name()

    def __log_book_name(self):
        print('Book私有方法')
        _logger.info(f"图书名称:{self.name}")

class SubBook(models.Model):
    ...
    def __log_book_name(self):
        print('SubBook私有方法')
        _logger.info(f"图书名称{self.name},作者:{self.author.name}")

将只有Book的方法会被调用,这点跟Python本身的继承表现是一致的。

多层重载

细心的朋友可能会有疑惑,如果多个子类都重载了父类的方法,那么会执行哪一个呢?下面我们就来揭晓这个问题的答案。

上面我们创建的一个子类SubBook,这里我们再次创建一个子类SubBook2,方法跟SubBook类似:


class SubBook2(models.Model):

    _inherit = "book_store.book"

    @api.one
    def log_book_name(self):
        """重载父类方法"""
        # res = super(SubBook,self).log_book_name()
        _logger.info(f"SubBook2的方法:图书名称{self.name},作者:{self.author.name}")

运行rpc脚本,输出结果如下:

log5

由此可以看出,当有多个子类重载了同一个父类方法时,只有最后加载的那个子模块的方法会被调用。

指定重载

如果,在子类SubBook中调用了父类的方法,同时子类SubBoo2也调用父类方法,那么会出现什么结果呢?

class SubBook(models.Model):

    _inherit = "book_store.book"

    @api.one
    def log_book_name(self):
        """重载父类方法"""
        # res = super(SubBook,self).log_book_name()
        _logger.info(f"SubBook的方法:图书名称{self.name},作者:{self.author.name}")
        super(SubBook,self).log_book_name()

class SubBook2(models.Model):

    _inherit = "book_store.book"

    @api.one
    def log_book_name(self):
        """重载父类方法"""
        # res = super(SubBook,self).log_book_name()
        _logger.info(f"SubBook2的方法:图书名称{self.name},作者:{self.author.name}")
        super(SubBook2,self).log_book_name()

输出:

可以看到,三个方法都被调用了。因此,在子类中使用Super调用父类方法,会调用继承链上所有的重载方法。

那么,我们如何只通过子类SubBook2调用父类方法,而不触发SubBook中的同名方法呢?

答案是SubBook2中直接条用SubBook的父类方法:

super(SubBook,self).log_book_name()

results matching ""

    No results matching ""