符合Python风格的对象(再谈向量类)

再谈向量类

为了说明用于生成对象表示形式的众多方法，我们将使用一个 Vector2d 类，它与第 1 章中的类似。这一节和接下来的几节会不断实 现这个类。我们期望 Vector2d 实例具有的基本行为如示例 9-1 所示。

示例 9-1　Vector2d 实例有多种表示形式

>>> v1 = Vector2d(3, 4)
>>> print(v1.x, v1.y) ➊
3.0 4.0
>>> x, y = v1 ➋
>>> x, y
(3.0, 4.0)
>>> v1 ➌
Vector2d(3.0, 4.0)
>>> v1_clone = eval(repr(v1)) ➍
>>> v1 == v1_clone ➎
True
>>> print(v1) ➏
(3.0, 4.0)
>>> octets = bytes(v1) ➐
>>> octets
b'd\\\\x00\\\\x00\\\\x00\\\\x00\\\\x00\\\\x00\\\\x08@\\\\x00\\\\x00\\\\x00\\\\x00\\\\x00\\\\x00\\\\x10@'
>>> abs(v1) ➑
5.0
>>> bool(v1), bool(Vector2d(0, 0)) ➒

❶ Vector2d 实例的分量可以直接通过属性访问（无需调用读值方 法）。 ❷ Vector2d 实例可以拆包成变量元组。 ❸ repr 函数调用 Vector2d 实例，得到的结果类似于构建实例的源 码。 ❹ 这里使用 eval 函数，表明 repr 函数调用 Vector2d 实例得到的是 对构造方法的准确表述。

❺ Vector2d 实例支持使用 == 比较；这样便于测试。 ❻ print 函数会调用 str 函数，对 Vector2d 来说，输出的是一个有 序对。 ❼ bytes 函数会调用__bytes__ 方法，生成实例的二进制表示形式。 ❽ abs 函数会调用__abs__ 方法，返回 Vector2d 实例的模。 ❾ bool 函数会调用__bool__ 方法，如果 Vector2d 实例的模为零， 返回 False，否则返回 True。

示例 9-1 中的 Vector2d 类在 vector2d_v0.py 文件中实现（见示例 9- 2）。这段代码基于示例 1-2，除了 == 之外（在测试中用得到），其他 中缀运算符将在第 13 章实现。现在，Vector2d 用到了几个特殊方法， 这些方法提供的操作是 Python 高手期待设计良好的对象所提供的。

示例 9-2　vector2d_v0.py：目前定义的都是特殊方法

from array import array
import math
class Vector2d:
typecode = 'd' ➊
def __init__(self, x, y):
self.x = float(x) ➋
self.y = float(y)
def __iter__(self):
return (i for i in (self.x, self.y)) ➌
def __repr__(self):
class_name = type(self).__name__
return '{}({!r}, {!r})'.format(class_name, *self) ➍
def __str__(self):
return str(tuple(self)) ➎
def __bytes__(self):
return (bytes([ord(self.typecode)]) + ➏
bytes(array(self.typecode, self))) ➐
def __eq__(self, other):
return tuple(self) == tuple(other) ➑
def __abs__(self):
return math.hypot(self.x, self.y) ➒
def __bool__(self):
return bool(abs(self)) ➓

❶ typecode 是类属性，在 Vector2d 实例和字节序列之间转换时使 用。 ❷ 在__init__ 方法中把 x 和 y 转换成浮点数，尽早捕获错误，以防 调用 Vector2d 函数时传入不当参数。 ❸ 定义__iter__ 方法，把 Vector2d 实例变成可迭代的对象，这样才 能拆包（例如，x, y = my_vector）。这个方法的实现方式很简单， 直接调用生成器表达式一个接一个产出分量。 这一行也可以写成 yield self.x; yield.self.y。第 14 章会进一步讨论__iter__ 特殊方 法、生成器表达式和 yield 关键字。 ❹__repr__ 方法使用 {!r} 获取各个分量的表示形式，然后插值，构 成一个字符串；因为 Vector2d 实例是可迭代的对象，所以 *self 会把 x 和 y 分量提供给 format 函数。 ❺ 从可迭代的 Vector2d 实例中可以轻松地得到一个元组，显示为一个 有序对。 ❻ 为了生成字节序列，我们把 typecode 转换成字节序列，然后…… ❼ ……迭代 Vector2d 实例，得到一个数组，再把数组转换成字节序 列。 ❽ 为了快速比较所有分量，在操作数中构建元组。对 Vector2d 实例来 说，可以这样做，不过仍有问题。参见下面的警告。 ❾ 模是 x 和 y 分量构成的直角三角形的斜边长。 ❿__bool__ 方法使用 abs(self) 计算模，然后把结果转换成布尔 值，因此，0.0 是 False，非零值是 True。

示例 9-2 中的__eq__ 方法，在两个操作数都是 Vector2d 实例时可用，不过拿 Vector2d 实例与其他具有相同数值的可迭代 对象相比，结果也是 True（如 Vector(3, 4) == [3, 4]）。这 个行为可以视作特性，也可以视作缺陷。第 13 章讲到运算符重载 时才能进一步讨论。

我们已经定义了很多基本方法，但是显然少了一个操作：使用 bytes() 函数生成的二进制表示形式重建 Vector2d 实例。