python交易記錄鏈的實現(xiàn)過程詳解

更新時間：2019年07月03日 15:13:03 作者：charles_lun

這篇文章主要介紹了python交易記錄鏈的實現(xiàn)過程詳解,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下

接著上篇的內(nèi)容，這里實現(xiàn)一個交易記錄鏈，廢話不多說，先看圖：

跟之前的邏輯類似，但也有少許不同，這里多了一個payloadhash,以及對payloadhash和prehash的hash,下面看代碼實現(xiàn)：

import hashlib
import datetime
# 交易記錄類
class DaDaMessage:

  def __init__(self, data):
    self.hash = None # 自身hash
    self.preHash = None # 上一個區(qū)塊hash
    self.data = data # 交易信息
    self.timestamp = datetime.datetime.now() # 交易時間
    self.payloadHash = self.payload_hash() # 交易后的哈希，防止篡改

  #對交易信息進行hash
  def payload_hash(self):
    sha = hashlib.md5()
    datastr = (str(self.data) + str(self.timestamp)).encode("utf-8")
    sha.update(datastr)
    return sha.hexdigest()
  #密封，鎖定交易信息
  def seal(self):
    self.hash = self.hash_message()
  #將交易信息和上一個區(qū)塊密封
  def hash_message(self):
    sha = hashlib.md5()
    datastr = (str(self.preHash) + str(self.payloadHash)).encode("utf-8")
    sha.update(datastr)
    return sha.hexdigest()
  #校驗區(qū)塊信息
  def validate(self):
    if self.hash != self.hash_message():
      raise InvalidateMessage("交易哈希鏈接唄修改")
    if self.payloadHash != self.payload_hash():
      raise InvalidateMessage("交易時間與數(shù)據(jù)被修改")
  #鏈接區(qū)塊
  def link(self,Message):
    self.preHash = Message.hash
  def __repr__(self):
    return "hash:{},preHash:{},data:{}".format(self.hash,self.preHash,self.data)

自定義異常類：

class InvalidateMessage(Exception):
  def __init__(self,*args,**kwargs):
    super(Exception,self).__init__(*args,**kwargs)

編寫測試模塊：

if __name__=="__main__":
  m1 = DaDaMessage("I love coins 12")
  m2 = DaDaMessage("I love coins 13")
  m3 = DaDaMessage("I love coins 14")
  try:
    m1.seal()
    m2.link(m1)
    m2.seal()
    m3.link(m2)
    m3.seal()
    m1.hash = "0xaaaajjjjj"
    m1.data = "I　don't love "
    m1.validate()
    m2.validate()
    m3.validate()
    print(m1)
    print(m2)
    print(m3)
  except InvalidateMessage as e:
    print(e)

查看打印結(jié)果：