太好用!教你几招Python魔法方法的妙用

liftword13小时前技术文章3

专注Python、AI、大数据,请关注公众号七步编程!

Python是一种简单的编程语言,满足一个需求,可以有各种各样的实现方法。正是因为它可以通过各种串联满足很多复杂的逻辑,因此,对代码可读性关注度不高的开发者而言,会编写出很多复杂而令人难以理解的代码。

Python语法本身虽然简单,但是要想实现可读性高,让其他开发者能够轻松理解的代码却是一项需要长时间学习和加强的技能。

而在这众多技能中,魔法方法(Magic Methods)便是其中一个很棒的选择,它允许用户在定义类的时候能够使用内置运算符和关键字,让使用类的方法更加直观明了。

魔法方法

如果你已经使用了一段时间Python,那么一定了解或者接触过魔法方法。

魔法方法是一种以双下划线开头和结尾的一种特殊方法,在使用类的时候非常常见,例如,经常会用到的__init__。它的功能是作为构造函数,能够在类初始化时调用,你可以在初始化方法中定义一些初始化变量、初始化操作,当执行到类内部时,它会首先执行这些方法。

当然,魔法方法远不至于__init__,为了帮助大家理解Python魔法方法的价值,本文就以一个示例来开始本文的讲解。

下面通过实现一个名为TimePeriod的类,来看一下如何使用Python魔法方法使得代码更加清晰、可读性更高。

基础的TimePeriod类

下面TimePeriod主要实现2个方法:

  • 时间的增加
  • 时间的比较

基础的实现方法大多会是下面这样:

class TimePeriod:

    def __init__(self, hours=0, minutes=0):
        self.hours = hours
        self.minutes = minutes

    def add(self, other):
        minutes = self.minutes + other.minutes
        hours = self.hours + other.hours

        if minutes >= 60:
            minutes -= 60
            hours += 1

        return TimePeriod(hours, minutes)

    def greater_than(self, other):
        if self.hours > other.hours:
            returnTrue
        elif self.hours < other.hours:
            returnFalse
        elif self.minutes > other.minutes:
            returnTrue
        else:
            returnFalse

实现addgreater_than两个方法,分别用于增加和对比时间大小。

或许,从这段代码中看不出有任何问题。接下来,我们使用这个类来看一下。

time_i_sleep = TimePeriod(9, 0)
time_i_work = TimePeriod(0, 30)
print(time_i_sleep.greater_than(time_i_work))
# True

这段代码在执行的时候没有任何问题,也不会报错。但是,如果你想要执行更为复杂的操作,例如,2个时间段加和再和另外2个时间段的加和进行对比,类似于A+BC+D进行比较,如果使用上述这个类,对比会是这样的:

time_i_sleep.add(time_i_watch_netflix).greater_than(time_i_work.add(time_i_do_chores))

这样看上去是不是非常复杂?

即便是作为一名聪明的开发人员,把这段代码拆开执行,也会像下面这样复杂:

time_spent_unproductively = time_i_sleep.add(time_i_watch_netflix)
time_spent_productively = time_i_work.add(time_i_do_chores)
time_spent_unproductively.greater_than(time_spent_productively)

魔法方法实现方式

其实在Python中,很容易就可以实现上述功能,Python内置的有2个魔方方法__add____gt__分别对应于+>运算。

现在,通过魔法方法来修改一下上面的类:

class TimePeriod:

    def __init__(self, hours=0, minutes=0):
        self.hours = hours
        self.minutes = minutes

    def __add__(self, other):
        minutes = self.minutes + other.minutes
        hours = self.hours + other.hours

        if minutes >= 60:
            minutes -= 60
            hours += 1

        return TimePeriod(hours, minutes)

    def __gt__(self, other):
        if self.hours > other.hours:
            returnTrue
        elif self.hours < other.hours:
            returnFalse
        elif self.minutes > other.minutes:
            returnTrue
        else:
            returnFalse

现在,再来进行一下对比运算:

(time_i_sleep + time_i_watch_netflix) > (time_i_work + time_i_do_chores)

这样看起来是不是更加容易理解和阅读了?

除此之外,你还可以加入更多魔法方法,实现更为丰富的功能。

例如,你想比较2个时间是否相等,会用到==运算,这时候你可以使用魔法方法__eq__,具体实现如下:

def __eq__(self, other):
    return self.hours == other.hours and self.minutes == other.minutes

Python魔法方法不止有算术运算和和比较运算,还有其他很多丰富的功能。例如,__str__,可以创建易于理解类的字符串。

def __str__(self):
    returnf"{self.hours} hours, {self.minutes} minutes"

如果你需要,还可以通过__getitem__把类转化成字典:

def __getitem__(self, item):
    if item == 'hours':
        return self.hours
    elif item == 'minutes':
        return self.minutes
    else:
        raise KeyError()

这样,可以把很多字典的优质特性加入到类中,可以像访问字典一样去访问类的属性。

其他

除了上述提到的一些魔法方法,Python还有很多魔法方法值得使用。本文不再逐一举例,下面介绍一下一些常用方法的功能,需要的可以在以后编码中用一下。

  • __new__:初始化类的实例时会调用__init__方法,而在实际创建实例时会更早地调用__new__方法。
  • __call____call__方法允许我们的实例像方法或函数一样可调用。
  • __len__:这允许你自定义Python内置len()函数。
  • __repr__:这类似于__str__ 魔法方法,它允许你定义类的字符串表示形式。但是,区别在于__str__是针对最终用户的,它提供了一个更加用户友好的非正式字符串,而__repr__是针对开发人员的,并且可能包含有关类的内部状态的更复杂的信息。
  • __setitem__:前面示例中我们已经看过__getitem__方法,它主要用于获取键值,而__setitem__则用于设置键值。
  • __enter____exit__:这两种方法通常一起使用,可以将你的类用作上下文管理器,实现类似Python中with语句的功能。

总结

Python虽然语法方面简单,但是要成为一名优秀的Python开发人员,需要持续学习和积累,不断学习这些书籍和教程之外的知识,来提升你代码的规范性、可读性,这些小技能都是需要留意、必不可少的。

相关文章

11 每个程序员都应该知道的 Python 魔法方法

在 Python 中,魔法方法帮助你模拟 Python 类中内置函数的行为。这些方法有前后双下划线(__),因此也被称为魔法方法 。这些魔法方法也帮助你实现 Python 中的运算符重载。你很可能见...

Python中关于魔法方法、单例模式的知识

目录:init,del,add,str和 repr,call,单例模式,class,dict,doc,bases,mro魔法方法:定义:在特定条件下,触发方法在python里面很多以双下划线开头且结尾...

掌握Python的&quot;魔法&quot;:特殊方法与属性完全指南

在Python的世界里,以双下划线开头和结尾的"魔法成员"(如__init__、__str__)是面向对象编程的核心。它们赋予开发者定制类行为的超能力,让自定义对象像内置类型一样优雅工...

Python 魔法方法的工作原理:实用指南(三)

以前的文章我们讲了python中对象表示、运算符等魔法方法的用法,下面我们继续深入探索容器、资源管理、性能方面的魔法方法。容器方法(类似列表、字典对象)当你需要自定义存储和检索数据的对象时,此方式可以...

Python进阶——如何正确使用魔法方法?(上)

微信搜索关注「水滴与银弹」公众号,第一时间获取优质技术干货。7年资深后端研发,用简单的方式把技术讲清楚。在做 Python 开发时,我们经常会遇到以双下划线开头和结尾的方法,例如 __init__、_...