python 常用库整理（yml、ujson、hashlib、base64、cgi、markdown篇）

配置文件：yml

import yaml

with open("test.yml", 'r') as f:
    obj = yaml.load(f.read())
    print obj

使用 ujson: 更加快

import ujson as json

print json.dumps({"a":1})

加密：hashlib

md5摘要：

import hashlib

md5 = hashlib.md5()
md5.update("a")
print md5.hexdigest()  # 长度32

sha1 / sha256

import hashlib

sha256 = hashlib.sha256()
sha256.update("a")
print sha256.hexdigest()  # 长度64

非对称加密, 签名/验证

from Crypto.PublicKey import RSA
from Crypto.Cipher import PKCS1_OAEP
from Crypto.Signature import PKCS1_PSS
# 建议使用上面的两个方法进行加解密

pri_key = RSA.importKey("private.pem")
pub_key = RSA.importKey("public.pem")

plain_text = "hello, world!"
K = 1

# 公钥加密，私钥解密
encrypted_text = pub_key.encrypt(plain_text, K)
plain = pri_key.decrypt(encrypted_text)
print plain

# 私钥签名，公钥验证
sign = pri_key.sign(plain_text, K)
is_true = pub_key.verify(plain_text, sign)
print is_true

注意分析K的作用
留空

秘钥生成

from Crypto import Random
from Crypto.PublicKey import RSA

rsa_gen = RSA.generate(1024, Random.new().read)

# 生成私钥
private_key = rsa_gen.exportKey()
public_key = rsa_gen.publickey().exportKey()
# 写入文件即可

编码：对一些文本数据，二进制数据进行编码，如图片数字签名等

base64 编码，
将二进制数据 3 个字节编码为 4 个字符
3 个字节 24 bit , 化为4组每组6bit 总共可表示64个字符，这就是64的由来
对于转换后的每组四个字符都是通过查询 字符表而来

import base64

tmp = base64.b64encode("hello,world")
# tmp = base64.urlsafe_b64encode("hello,world")  # 与上面的区别就是将 + / 换为了-_
print base64.b64decode(tmp)

注意，一些会将后面的==号去掉，在解码的时候注意长度补齐4的倍数，也就是补充=号

defaultdict 有默认值的字段

from collections import defaultdict

dd = defaultdict(lambda :"N/A")
print dd["a"]  # 输出 "N/A"

dd = defaultdict(int)
print dd["a"]  # 输出 0

HTML 标签与实体进行转换

主要用到cgi和HTMLParser

import cgi
import HTMLParser

text = "<p>hello</p>"
tmp = cgi.escape(text)
print tmp

parser = HTMLParser.HTMLParser()
print parser.unescape(tmp)