又有几个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”