python经常复用的程序片段记载

1 有输入参数

import sys, os

if len(sys.argv) < 3:
        print 'input invalid'
        exit(1)

a = sys.argv[1]
b = sys.argv[2]
if not a.isdigit() or not b.isdigit():
        print 'input is not digit'
        exit(1)

2 打开文件错误捕捉

        try:
                f1 = open(str(i))
        except IOError, e:
                print 'cannot open file ' + str(i)
                i += 1
                continue
        line = f1.readline()
        dic = eval(line)
        f1.close()

3 用urllib和urllib2发送post请求

    body = urllib.urlencode({
        'token':token,
        'task':'{"TaskName":"testScipt"}'})
    returnedReq = urllib2.Request(
        url = 'http://www.test.com/',
        data = body)
    returnedResult = urllib2.urlopen(returnedReq).read()
    returnedResult = returnedResult.strip()
    print returnedResult

4 操作excel

import xlrd

data = xlrd.open_workbook('test.xlsx')
table = data.sheets()[0] #选择第一个工作区
cols = table.col_values(0) #选择第一列

print len(cols)

5 使用eval()或ast.literal_eval() 出现错误

NameError: name ‘total_count’ is not defined或ValueError: malformed string

比如原始字符串是 str = '{"name":"jgy","age":null}'

转为字典时会发现null不是一个字符串，就当做变量，但是没有定义（python认可的是Null True False 注意第一个字母大写），所以出错，相应的实际过程中可能还会到 ‘true’ is not defined, ‘false’ is not defined类似的等等。

既然没定义，在前面定义一下就行了，比如自己定义 null = 'null', true = 'true'..

还有个想法，可以用正则把冒号后面的不是数字也没加引号的value值给加上引号。

6 时间

获取最简洁日期格式：

>>>time.strftime('%Y-%m-%d',time.localtime(time.time()))

'2013-07-04'

获取上一秒时间：

time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time() - 1))

将一个字符串时间转为python时间格式

str1 = '2013-06-03 20:07:41'
str2 = '2013-06-03 20:08:43'
time1= time.strptime(str1,"%Y-%m-%d %H:%M:%S") #time1 is time_struct type
time1= time.strptime(str2,"%Y-%m-%d %H:%M:%S")

#now we wanna to 取得time1和time2的时间差
#需要转为datetime，然后使用datetime的timedelta方法

datetime1 = datetime.datetime(*time1[:6])
datetime2 = datetime.datetime(*time2[:6])
datedelta = datetime2 - datetime1
print datedelta
#datetime.timedelta(0, 62)  #后面是秒数
seconds = delta.seconds #获取int型秒数

7 海量数据去重

>>> a = [11,22,33,44,11,22]  
>>> b = set(a)  
>>> b  
set([33, 11, 44, 22])  
>>> c = [i for i in b]  
>>> c  
[33, 11, 44, 22]