编写同时兼容Python2.x与Python3.x版本的代码的几个示例


Posted in Python onMarch 30, 2015

编写兼容Python2.x与3.x代码

当我们正处于Python 2.x到Python 3.x的过渡期时,你可能想过是否可以在不修改任何代码的前提下能同时运行在Python 2和3中。这看起来还真是一个合理的诉求,但如何开始呢?哪些Python 2 代码在 3.x 解释器执行时容易出状况呢?
print vs print()

如果你想的和我一样,你或许会说print语句,这是个很好的着手点,先简单展示一下,print在2.x中是一条语句,而在3.x中它是一个关键字或者是保留字。换句话说,因为这个变化涉及到语言的语法,你不可以使用在if语句中,Python仍然没有#ifdef 宏。下面尝试把括号里面的参数打印出来:

>>> print('Hello World!')
Hello World!

很酷,这个在Python2和Python3中都可以运行,而且运行的效果是一样的,再来看看下面这段:
 

>>> print(10, 20) # Python 2
(10, 20)

此时,你并没有像前面那样幸运得到一样的结果,Python2中打印的是元组(tuple),而在Python3中传递多个参数到print()里面时打印的是两个值:
 

>>> print(10, 20) # Python 3
10 20

如果你思考得比较多的话,我们可以检查print是否是一个关键字,keyword模块包含一个关键字列表。print在3.x中不是关键字,可以简单验证一下:
 

>>> import keyword
>>> 'print' in keyword.kwlist
False

作为一名聪明的程序员,你可能在2.x中尝试的时候期待的结果是True,尽管这并没有错,但是为了达到Python3的效果,但你仍然会因为其他原因导致失败。
 

>>> import keyword
>>> if 'print' in keyword.kwlist:
...  from __future__ import print_function
...
File "", line 2
SyntaxError: from __future__ imports must occur at the beginning of the file

一种解决方案是使用一个函数,其功能类似于print,其中之一是sys.stdout.write(),另一个是distutils.log.warn()。不管出于什么原因,我们决定使用后者。“hello world”的例子看起来是这样的:
 

# Python 2.x
print 'Hello World!'
# Python 3.x
print('Hello World!')

下面的代码就可以在两个版本中通用:
 

# Python 2.x & 3.x compatible
from distutils.log import warn as printf
printf('Hello World!')

为什么我们不用sys.stdout.write()呢,因为我们需要添加一个NEWLINE字符在字符串的结尾来兼容这种行为(python2.x中write方法不会换行):
 

# Python 2.x & 3.x compatible
import sys
sys.stdout.write('Hello World!n')

Import your way to a solution

一般情况情况下,import时没什么烦恼,只要正确的导入就行,但在下面代码中,我们想导入urlopen()函数,在Python2中,他同时存在与urllib2和urllib2中(我们使用后者),在Python3中,他被集成到了urllib.request中,而你的方案是要既能在2.x和3.x中正常工作:

try:
 from urllib2 import urlopen
except ImportError:
 from urllib.request import urlopen

出于对内存的保护,也许你对iterator(Python3)版本的zip()更加有兴趣,在Python2中,iterator版本是itertools.izip()。这个函数在Python3中被重命名替换成了zip()。如果你使用迭代版本,导入语句也非常直白:
 

try:
 from itertools import izip as zip
except ImportError:
 pass

另一个列子是看来来并不怎么优雅的StringIO类,在Python2中,纯Python版本是StringIO模块,意味着访问的时候是通过StringIO.StringIO,同样还有一个更为快速的C语言版本,位于cStringIO.StringIO,不过这取决你的Python安装版本,你可以优先使用cStringIO然后是StringIO(如果cStringIO不能用的话)。在Python3中,Unicode是默认的string类型,但是如果你做任何和网络相关的操作,很有可能你不得不用ASCII/字节字符串来操作,所以代替StringIO,你要io.BytesIO,为了达到你想要的,这个导入看起来有点丑:
 

try:
 from io import BytesIO as StringIO
except ImportError:
 try:
  from cStringIO import StringIO
 except ImportError:
  from StringIO import StringIO

Putting it all together

如果你运气好的话,上面那些就是你要准备做的全部,剩下的代码都比开始设置的地方更简单。如果你按照上面的方式导入了distutils.log.warn()[printf()],url*urlopen(),*.StringIO和一个标准的导入:xml.etree.ElementTree(2.5及更新的),现在你就可以写一个非常简短短的解析器来展示从Google News服务中提供的头条故事(译注:当然首先得备一个梯子),只需八行代码:
 

g = urlopen('http://news.google.com/news?topic=h&output=rss')
f = StringIO(g.read())
g.close()
tree = xml.etree.ElementTree.parse(f)
f.close()
for elmt in tree.getiterator():
 if elmt.tag == 'title' and not
   elmt.text.startswith('Top Stories'):
  printf('- %s' % elmt.text)

这段脚本在2.x和3.x下面运行时,不需要做任何改动,运行效果完全一样,当然,如果你正在使用的是2.4或者更老的版本,你需要单独下载ElementTree。

但是有时候感觉这些改变把你优雅的Python代码弄得一团糟,毕竟可读性才是最重要的,如果你要优先保证代码的整洁而且在不修改任何地方的前提下运行在两个版本的Python环境中,那么你可以看一下six包。

six一个兼容库,它的主要任务是提供接口隐藏复杂的细节,你可以在这里找到它。无论你是使用像six这样的库还是用自己的方法来做,我们希望这个简短的介绍可以让你开始考虑写的代码能够在2.x和3.x下同时运行。

Python 相关文章推荐
Python使用Socket(Https)Post登录百度的实现代码
May 18 Python
Python随机生成数模块random使用实例
Apr 13 Python
Python浅拷贝与深拷贝用法实例
May 09 Python
Python基于tkinter模块实现的改名小工具示例
Jul 27 Python
Python实现解析Bit Torrent种子文件内容的方法
Aug 29 Python
使用Python读取安卓手机的屏幕分辨率方法
Mar 31 Python
PyQT实现多窗口切换
Apr 20 Python
详解Python用三种方式统计词频的方法
Jul 29 Python
Python爬虫学习之翻译小程序
Jul 30 Python
Windows下实现将Pascal VOC转化为TFRecords
Feb 17 Python
Python 实现打印单词的菱形字符图案
Apr 12 Python
python excel和yaml文件的读取封装
Jan 12 Python
以Python的Pyspider为例剖析搜索引擎的网络爬虫实现方法
Mar 30 #Python
在树莓派2或树莓派B+上安装Python和OpenCV的教程
Mar 30 #Python
Python中利用函数装饰器实现备忘功能
Mar 30 #Python
利用Python绘制MySQL数据图实现数据可视化
Mar 30 #Python
Python面向对象编程中的类和对象学习教程
Mar 30 #Python
详细介绍Python函数中的默认参数
Mar 30 #Python
在Python中利用Into包整洁地进行数据迁移的教程
Mar 30 #Python
You might like
复杂检索数据并分页显示的处理方法
2006/10/09 PHP
PHP设置图片文件上传大小的具体实现方法
2013/10/11 PHP
Yii中实现处理前后台登录的新方法
2015/12/28 PHP
php中各种定义变量的方法小结
2017/10/18 PHP
thinkPHP中钩子的使用方法实例分析
2017/11/16 PHP
PHPUnit + Laravel单元测试常用技能
2019/11/06 PHP
jquery DOM操作 基于命令改变页面
2010/05/06 Javascript
javascript与CSS复习(三)
2010/06/29 Javascript
js实现的切换面板实例代码
2013/06/17 Javascript
jquery代码实现简单的随机图片瀑布流效果
2015/04/20 Javascript
Node.js实现连接mysql数据库功能示例
2017/09/15 Javascript
Vue.js 中取得后台原生HTML字符串 原样显示问题的解决方法
2018/06/10 Javascript
vue实现pdf导出解决生成canvas模糊等问题(推荐)
2018/10/18 Javascript
python自然语言编码转换模块codecs介绍
2015/04/08 Python
介绍Python中的文档测试模块
2015/04/28 Python
浅谈python为什么不需要三目运算符和switch
2016/06/17 Python
Python使用正则表达式实现文本替换的方法
2017/04/18 Python
在python3环境下的Django中使用MySQL数据库的实例
2017/08/29 Python
python中int与str互转方法
2018/07/02 Python
Python SQL查询并生成json文件操作示例
2018/08/17 Python
python中将zip压缩包转为gz.tar的方法
2018/10/18 Python
Dlib+OpenCV深度学习人脸识别的方法示例
2019/05/14 Python
python django生成迁移文件的实例
2019/08/31 Python
解析Python3中的Import
2019/10/13 Python
python3-flask-3将信息写入日志的实操方法
2019/11/12 Python
python标识符命名规范原理解析
2020/01/10 Python
python GUI库图形界面开发之PyQt5开发环境配置与基础使用
2020/02/25 Python
python实现控制台输出彩色字体
2020/04/05 Python
Python分析最近大火的网剧《隐秘的角落》
2020/07/02 Python
解决PyCharm不在run输出运行结果而不是再Console里输出的问题
2020/09/21 Python
python绘制分布折线图的示例
2020/09/24 Python
Exoticca英国:以最优惠的价格提供豪华异国情调旅行
2018/10/18 全球购物
大四学生找工作的自荐信
2014/03/27 职场文书
2014党委书记四风问题对照检查材料思想汇报
2014/09/22 职场文书
股权转让协议范本
2014/12/07 职场文书
《分一些蚊子进来》读后感3篇
2020/01/09 职场文书