編寫兼容Python2.x與3.x代碼

當(dāng)我們正處于Python 2.x到Python 3.x的過渡期時，你可能想過是否可以在不修改任何代碼的前提下能同時運行在Python 2和3中。這看起來還真是一個合理的訴求，但如何開始呢？哪些Python 2 代碼在 3.x 解釋器執(zhí)行時容易出狀況呢？
print vs print()

如果你想的和我一樣，你或許會說print語句，這是個很好的著手點，先簡單展示一下，print在2.x中是一條語句，而在3.x中它是一個關(guān)鍵字或者是保留字。換句話說，因為這個變化涉及到語言的語法，你不可以使用在if語句中，Python仍然沒有#ifdef 宏。下面嘗試把括號里面的參數(shù)打印出來：

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

很酷，這個在Python2和Python3中都可以運行，而且運行的效果是一樣的，再來看看下面這段：
 

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

此時，你并沒有像前面那樣幸運得到一樣的結(jié)果，Python2中打印的是元組(tuple)，而在Python3中傳遞多個參數(shù)到print()里面時打印的是兩個值：
 

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

如果你思考得比較多的話，我們可以檢查print是否是一個關(guān)鍵字，keyword模塊包含一個關(guān)鍵字列表。print在3.x中不是關(guān)鍵字，可以簡單驗證一下：
 

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

作為一名聰明的程序員，你可能在2.x中嘗試的時候期待的結(jié)果是True，盡管這并沒有錯，但是為了達到Python3的效果，但你仍然會因為其他原因?qū)е率　?br />  

>>> 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

一種解決方案是使用一個函數(shù)，其功能類似于print，其中之一是sys.stdout.write()，另一個是distutils.log.warn()。不管出于什么原因，我們決定使用后者?！癶ello 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字符在字符串的結(jié)尾來兼容這種行為(python2.x中write方法不會換行)：
 

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

Import your way to a solution

一般情況情況下，import時沒什么煩惱，只要正確的導(dǎo)入就行，但在下面代碼中，我們想導(dǎo)入urlopen()函數(shù)，在Python2中，他同時存在與urllib2和urllib2中（我們使用后者），在Python3中，他被集成到了urllib.request中，而你的方案是要既能在2.x和3.x中正常工作：

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

出于對內(nèi)存的保護，也許你對iterator(Python3)版本的zip()更加有興趣，在Python2中，iterator版本是itertools.izip()。這個函數(shù)在Python3中被重命名替換成了zip()。如果你使用迭代版本，導(dǎo)入語句也非常直白：
 

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

另一個列子是看來來并不怎么優(yōu)雅的StringIO類，在Python2中，純Python版本是StringIO模塊，意味著訪問的時候是通過StringIO.StringIO，同樣還有一個更為快速的C語言版本，位于cStringIO.StringIO，不過這取決你的Python安裝版本，你可以優(yōu)先使用cStringIO然后是StringIO（如果cStringIO不能用的話)。在Python3中，Unicode是默認(rèn)的string類型，但是如果你做任何和網(wǎng)絡(luò)相關(guān)的操作，很有可能你不得不用ASCII/字節(jié)字符串來操作，所以代替StringIO，你要io.BytesIO，為了達到你想要的，這個導(dǎo)入看起來有點丑：
 

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

如果你運氣好的話，上面那些就是你要準(zhǔn)備做的全部，剩下的代碼都比開始設(shè)置的地方更簡單。如果你按照上面的方式導(dǎo)入了distutils.log.warn()[printf()],url*urlopen(),*.StringIO和一個標(biāo)準(zhǔn)的導(dǎo)入：xml.etree.ElementTree(2.5及更新的)，現(xiàn)在你就可以寫一個非常簡短短的解析器來展示從Google News服務(wù)中提供的頭條故事（譯注：當(dāng)然首先得備一個梯子），只需八行代碼：
 

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下面運行時，不需要做任何改動，運行效果完全一樣，當(dāng)然，如果你正在使用的是2.4或者更老的版本，你需要單獨下載ElementTree。

但是有時候感覺這些改變把你優(yōu)雅的Python代碼弄得一團糟，畢竟可讀性才是最重要的，如果你要優(yōu)先保證代碼的整潔而且在不修改任何地方的前提下運行在兩個版本的Python環(huán)境中，那么你可以看一下six包。

six一個兼容庫，它的主要任務(wù)是提供接口隱藏復(fù)雜的細(xì)節(jié)，你可以在這里找到它。無論你是使用像six這樣的庫還是用自己的方法來做，我們希望這個簡短的介紹可以讓你開始考慮寫的代碼能夠在2.x和3.x下同時運行。

最新評論