在看Kenneth Reitz大神的Records项目时,注意到在Readme中,读取数据有三种写法:
Values can be accessed many ways: row.user_email, row['user_email'], or row[3]
假设一个数据表如下所示:
username | active | name | user_email | timezone |
model-t | True | Henry Ford | model-t@gmail.com | 2016-02-06 22:28:23.894202 |
那么,如果你想读取user_email的值,除了可以使用 row.user_email
和 row['user_email']
以外,由于 user_email
在第3列(username是第0列),所以还可以使用 row[3]
来读取。
今天我们就来研究一下,他是如何实现这个功能的。
假设我们现在有一个类:
class People: def __init__(self, name, age): self.name = name self.age = age def walk(self): print('i am walk')
基于这个类初始化一个对象kingname:
kingname = People('kingname', 26) print(kingname.age)
运行效果如下图所示:
可以看到,我们已经实现了类似于records项目中的 row.user_email
的写法。
但是当我们想像字典一样取读取的时候,就会报错,如下图所示。
此时,为了让一个对象可以像字典一样被读取,我们需要实现它的 __getitem__
方法:
class People: def __init__(self, name, age): self.info = {'name': name, 'age': age} def walk(self): print('i am walk') def __getitem__(self, key): return self.info.get(key, '不存在')
此时就可以像字典一样去读取了,运行效果如下图所示:
但新的问题又来了,不能直接使用 kinganme.age
读取数据了,这样写会导致报错,如下图所示:
为了解决这个问题,我们再来实现这个类的 __getattr__
方法:
class People: def __init__(self, name, age): self.info = {'name': name, 'age': age} def walk(self): print('i am walk') def __getitem__(self, key): return self.info.get(key, '不存在') def __getattr__(self, key): return self.info.get(key, '不存在')
运行效果如下图所示:
那么如何实现records里面的 row[3]
这种写法呢?
这就需要先解释一下,records这个库是用来做什么的,以及它的数据是如何存放的。
K大写的records,是用于来更加方便地读写SQL数据库,避免繁琐地写各种SQL语句。
我们今天要研究的这个写法,在 Record
这个类里面。这个类用来保存MySQL中的一行数据。Record
这个类包含两个属性,分别为 _keys
和 _values
,其中 _keys
用于记录所有的字段名, _values
用于记录一行的所有值。字段名和值是按顺序一一对应的。
例如 _keys
里面下标为3表示字段名 user_email
,那么 _values
里面下标为3的数据就是 user_email
的值。
正是由于这样一个一一对应的关系,所以 row[3]
才能实现 row['user_email']
、 row.user_email
相同的效果。
回到我们的People类,为了实现相同的目的,我们再次修改代码:
class People: def __init__(self, name, age): self._keys = ['name', 'age'] self._values = [name, age] def walk(self): print('i am walk') def __getitem__(self, key): if isinstance(key, int): if 0 <= key < len(self._keys): return self._values[key] return '索引超出边界' index = self._keys.index(key) if index == -1: return 'Key不存在' return self._values[index] def __getattr__(self, key): return self.__getitem__(key)
运行效果如下图所示:
需要说明的是,无论是使用 kingname['key']
还是 kingname[1]
,他们都会进入到 __getitem__
方法中去。
我这里给出的例子相较于 records
项目的代码做了简化,不过关键的部分都已经囊括了进来。
最后,推荐有兴趣的同学通读records这个项目的源代码,这个项目只有一个文件,包括注释和空行才500多行代码,但你将会从K大的代码中学到非常多的东西。