python 常用庫整理（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)