又有几个Python小技巧分享

云栖号：https://yqh.aliyun.com
第一手的上云资讯，不同行业精选的上云企业案例库，基于众多成功案例萃取而成的最佳实践，助力您上云决策！

今天这篇文章为大家带来几个Python使用的小技巧：

• 快速生成依赖文件
• 文件路径处理
• 通过set将对象去重
• 单元测试unitest捕获异常

快速生成依赖文件

Python通过requirements.txt来管理项目所依赖的库，通过pip install -r requirements.txt命令可以直接安装项目所需要的依赖文件。

常见的requirements.txt生成方法由两种

• 原生pip工具pip freeze > requirements.txt，这种方法会将当前环境中所安装的库全部导出，即使你在项目中没有使用，只是之前实验时安装的依赖库。
• 第三方工具pipreqs requirements.txt --encoding=utf-8，该工具使用前需要安装pip install pipreqs，pipreqs的优点它会扫描项目的依赖，只导出使用的依赖；pipreqs还可以指定扫描目录pipreqs ./ --encoding=utf-8。

文件路径处理

Python中常见的文件路径处理方式是通过os.path进行处理。

1In[2]: import os
 2# 获取当前目录的绝对路径
 3In[3]: os.path.abspath('./')
 4Out[3]: 'D:\\Work\\CommonTools'
 5# 进行路径拼接
 6In[4]: os.path.join(os.path.abspath('./'),'new_file.py')
 7Out[4]: 'D:\\Work\\CommonTools\\new_file.py'
 8# 获取父目录
 9In[5]: os.path.abspath(os.path.dirname(os.getcwd()))
10Out[5]: 'D:\\Work'
11# 获取文件后缀名
12In[6]: os.path.splitext('new_file.py')[-1]
13Out[6]: '.py'

os.path处理文件路径时，比较复杂、难用，因此可以使用pathlib替代os.path。

1In[2]: import pathlib
 2# 获取当前目录的绝对路径
 3In[3]: path = pathlib.Path('./')
 4In[4]: path=path.absolute()
 5In[5]: path.as_posix()
 6Out[5]: 'D:/Work/CommonTools'
 7# 进行路径拼接
 8In[6]: file_path = path / 'new_file.py'
 9In[7]: file_path.as_posix()
10Out[7]: 'D:/Work/CommonTools/new_file.py'
11# 获取父目录
12In[8]: path.parent
13Out[8]: WindowsPath('D:/Work')
14# 获取文件后缀名
15In[9]: file_path.suffix
16Out[9]: '.py'
17# 列出当前目录下的py文件
18In[13]: list(path.glob('*.py'))
19Out[13]: [WindowsPath('D:/Work/CommonTools/tools.py')]

更多os.pah库和pathlib库的方法对照，请查看官方文档。

通过set将对象去重

通过set将对象去重要求对象必须实现三个方法：

• eq：判断两个对象是否相等。
• ne：判断两个对象是否不等。
• hash：生成对象的hash值。

set内部可以视作为字典，以对象的hash值作为键，通过hash来判断对象是否重复。但是hash值可能出现冲突，因此还需在hash值相同时，对比两个对象是否相等，这也是为什么需要实现这三个方法的原因。

1In[2]: class Person(object):
 2   ...:    def __init__(self, name, age):
 3   ...:        self._name = name
 4   ...:        self._age = age
 5   ...:
 6   ...:    def __eq__(self, other):
 7   ...:        if isinstance(other, Person):
 8   ...:            return (self._name == other._name) and (self._age == other._age)
 9   ...:        else:
10   ...:            return False
11   ...:
12   ...:    def __ne__(self, other):
13   ...:        return not self.__eq__(other)
14   ...:
15   ...:    def __hash__(self):
16   ...:        return hash(self._name + str(self._age))
17   ...:
18   ...:    def __repr__(self):
19   ...:        return f'{self._name}:{self._age}'
20   ...:    
21In[3]: p1 = Person('martin', 24)
22   ...:p2 = Person('martin', 24)
23In[4]: set([p1, p2])
24Out[4]: {martin:24}
25In[5]: p2 = Person('martin', 25)
26In[6]: set([p1, p2])
27Out[6]: {martin:24, martin:25}

单元测试unitest捕获异常

单元测试unitest中提供了两个方法用于捕获异常，判断异常信息是否符合预期：

• assertRaises()：只支持指定异常判断。
• assertRaisesRegex(异常类型,异常信息正则表达式,测试方法,参数)：支持匹配异常信息。

1import unittest
 2
 3def divide_exactly(a, b):
 4    return a // b
 5
 6class MyTestCase(unittest.TestCase):
 7    def test_division(self):
 8        self.assertEqual(divide_exactly(6, 6), 1)
 9
10        # 捕获指定异常
11        self.assertRaises(ZeroDivisionError, divide_exactly, 5, 0)
12        # 可以作为上下文管理器使用
13        with self.assertRaises(ZeroDivisionError):
14            divide_exactly(5, 0)
15
16        # 捕获指定异常
17        self.assertRaisesRegex(ZeroDivisionError, 'integer division or modulo by zero', divide_exactly, 5, 0)
18        # 可以作为上下文管理器使用
19        with self.assertRaisesRegex(ZeroDivisionError, 'integer division or modulo by zero'):
20            divide_exactly(5, 0)
21
22
23if __name__ == '__main__':
24    unittest.main()

云栖号：https://yqh.aliyun.com
第一手的上云资讯，不同行业精选的上云企业案例库，基于众多成功案例萃取而成的最佳实践，助力您上云决策！

原文发布时间：2020-02-11
本文作者：有只猪在飞
本文来自：“简说Python公众号”，了解相关信息可以关注“简说Python”