#coding=utf-8 
import sys 
import MySQLdb 
''''' 
python操作MySQL數(shù)據(jù)庫，模擬銀行轉(zhuǎn)賬 
''' 
class Trans_for_Money(object): 
 #初始化 類 
 def __init__(self,conn): 
  self.conn = conn 
 #### 1、檢查所輸入的賬號是否存在 #### 
 def check_acct_available(self,source_acctid): 
  #使用與數(shù)據(jù)庫的鏈接并返回游標 
  cursor=self.conn.cursor() 
  try: 
   #數(shù)據(jù)庫命令 
   sql="select * from tr_money where acctid=%s" %source_acctid 
   #執(zhí)行命令 
   cursor.execute(sql) 
   #為方便觀察執(zhí)行過程 
   print "check_acct_available:" + sql 
   #講結(jié)果集放入變量result中，若result不等于1，則沒有這個賬號，輸出異常 
   result=cursor.fetchall() 
   if len(result)!=1: 
    raise Exception("賬號%s不存在" %source_acctid) 
  finally: 
   #若過程出現(xiàn)問題，仍需要關閉游標對象 
   cursor.close() 
 #### 2、檢查減款人余額是否充足，方法與上一個函數(shù)一樣，只是多加了一個money參數(shù) ### 
 def has_enough_money(self,source_acctid,money): 
  cursor=self.conn.cursor() 
  try: 
   sql="select * from tr_money where acctid=%s and money>%s" %(source_acctid,money) 
   cursor.execute(sql) 
   print "has_enough_money:" + sql 
   result=cursor.fetchall() 
   if len(result)!=1: 
    raise Exception("賬號%s余額不足" %source_acctid) 
  finally: 
   cursor.close() 
 #### 3、減款操作 ### 
 def reduce_money(self,source_acctid,money): 
  cursor=self.conn.cursor() 
  try: 
   #數(shù)據(jù)庫命令，減去對應減款人的金額數(shù) 
   sql="update tr_money set money=money-%s where acctid=%s" %(money,source_acctid) 
   cursor.execute(sql) 
   print "reduce_money:" + sql 
   #操作的execute()數(shù)據(jù)行數(shù)不等于1則減款失敗 
   if cursor.rowcount!=1: 
    raise Exception("賬號%s減款失敗" %source_acctid) 
  finally: 
   cursor.close() 
 #### 4、收款操作，與減款方法相同 ### 
 def add_money(self,target_acctid,money): 
  cursor=self.conn.cursor() 
  try: 
   sql="update tr_money set money=money+%s where acctid =%s" %(money,target_acctid) 
   cursor.execute(sql) 
   print "add_money:" + sql 
   if cursor.rowcount!=1: 
    raise Exception("賬號%s收款失敗" %target_acctid) 
  finally: 
   cursor.close() 
 #### 5、分別傳入?yún)?shù)，代入上方函數(shù)，執(zhí)行操作 ### 
 def trans_for(self,source_acctid,target_acctid,money): 
  try: 
   self.check_acct_available(source_acctid) 
   self.check_acct_available(target_acctid) 
   self.has_enough_money(source_acctid,money) 
   self.reduce_money(source_acctid,money) 
   self.add_money(target_acctid,money) 
   #提交當前事務 
   self.conn.commit() 
  except Exception as e: 
   #若出錯，回滾當前事務 
   self.conn.rollback() 
   raise e 
if __name__=="__main__": 
 # source_acctid=sys.argv[1] 
 # target_acctid=sys.argv[2] 
 # money=sys.argv[3] 
 #建立與數(shù)據(jù)庫的鏈接 
 conn = MySQLdb.Connect( 
       host='127.0.0.1', 
       port=3306, 
       user='root', 
       passwd='12345678', 
       db='tt', 
       charset='utf8' 
       ) 
 #手動輸入減款人、收款人、轉(zhuǎn)款數(shù) 
 source_acctid=raw_input("請輸入減款人： ") 
 target_acctid=raw_input("請輸入收款人： ") 
 money=raw_input("請輸入轉(zhuǎn)款數(shù)： ") 
 #將參數(shù)傳入類中 
 tr_money=Trans_for_Money(conn) 
 try: 
  tr_money.trans_for(source_acctid,target_acctid,money) 
 except Exception as e: 
  print"出現(xiàn)問題："+str(e) 
 finally: 
  conn.close() 
  #關閉鏈接