Python實作一個簡單的銀行轉帳操作方法

前言

在進行一個應用系統的開發過程中，從上到下一般需要四個構件：客戶端-業務邏輯層-資料存取層-資料庫，其中資料存取層是一個底層、核心的技術。而且在實際開發中，資料庫的操作也就是說資料存取層都是嵌套在其他語言中的，其是程式設計的核心。本文面向的是python語言，也就是透過python操作資料庫來實現簡單的銀行轉帳作業。

工具

python提供了python DB API用來統一操作資料庫，使存取資料庫的介面規範化，在沒有python DB API之前，介面程式十分混亂，不同的資料庫需要不同的操作接口，所以這個接口提供了極大的方便。在具體操作的時候，我們需要操作資料庫以及其他邏輯的python程式碼，資料庫連接物件connection來建立連接，資料庫互動物件cursor來「運送」數據，一個健壯的系統必不可少的便是資料庫異常類Exceptions。整個存取資料庫流程如下圖：

接下來分別介紹下兩個主要對象：

connection：資料庫連接對象，建立python客戶端與資料庫的網路連線。
建立方法：MySQLdb.connect()，包含的主要成員方法：
cursor():使用該連線建立並傳回遊標
commit():提交目前交易
rollback():回滾動當前交易
close()關閉連接
cursor:遊標對象，用於執行查詢與獲取結果，cursor對象支援的主要方法如下：
execute():執行SQL語句，將結果從資料庫取得到客戶端
fetchone():取得結果集的下一行
fetchmany(size):取得結果集的下size行
fetchall():取得結果集中剩下的所有行
rowcount:最近一次execute回傳資料的行數
close():關閉遊標物件
在上面的方法中提到了一個關鍵名詞：事務，什麼是事務呢？他是存取和更新資料的一個程式執行單元，很多操作的一個集合，有四個特點：

原子性：事物中包括的諸操作要么都做，要么都不做
一致性：交易必須使資料庫從一致性狀態變成另一個一致性狀態
隔離型：一個交易的執行不被其他交易幹擾
持久性：交易一旦提交，它對資料庫的改變就是持久性的
事務的上述特點正是我們完成銀行轉帳作業的關鍵。

具體實作

在開發中我們怎麼樣使用事務呢？

關閉自動commit()
正常結束交易：conn.commit()，
異常結束交易：conn.rollback()
在銀行轉帳系統中，需要考慮以下需求：例如A給B轉賬，當A帳戶上減少了M錢時，必須在B帳戶上多了M錢，不能A減了B沒加，也不能B加了A還沒減，當然帳戶必須是有效的，M錢的金額一定大於A帳戶上的金額。所以在具體設計的時候，需要將A帳戶的金錢減少和B帳戶的金錢增加作為一個事務，要么同時成功，要么一起失敗。依照這個需求，書寫程式碼，詳細程式碼見github，程式碼複製和資料庫如下，有兩個帳戶，分別擁有金錢110和10，在運行程式碼的時候在參數欄輸入1,2,100（source_acctid, target_acctid, tranfer_money）。

整個程式碼的邏輯如下：首先連接資料庫，之後執行邏輯，然後斷開資料庫連接，執行的邏輯包括檢查轉帳雙方的帳戶是否有效，轉帳金額是否多於轉帳人的帳戶餘額，分別給予轉帳雙方的帳號金額變更。如果正常結束事務，提交修改資料庫，否則回滾。

#coding:utf-8
import sys
import MySQLdb
class TransferMoney():
def __init__(self, conn):
self.conn = conn
def transfer(self, src, target, money):
try:
self.check_acct_available(src)
self.check_acct_available(target)
self.has_enough_money(src, money)
self.reduce_money(src, money)
self.add_money(target, money)
self.conn.commit()
except Exception as e:
print e
self.conn.rollback()
def reduce_money(self, src, money):
cursor = self.conn.cursor()
try:
sql = "update account set money = money - %s where acctid = %s" %(money, src)
cursor.execute(sql)
print "reduce_money: " + sql
#rs = cursor.fetchall()
if cursor.rowcount != 1:
raise Exception("the account reduce money fail")
finally:
cursor.close()
def add_money(self, target, money):
cursor = self.conn.cursor()
try:
sql = "update account set money = money + %s where acctid = %s" %(money, target)
cursor.execute(sql)
print "add_money: " + sql
#rs = cursor.fetchall()
if cursor.rowcount != 1:
raise Exception("the account add money fail")
finally:
cursor.close()
def check_acct_available(self, accit):
cursor = self.conn.cursor()
try:
sql = "select * from account where acctid = %s" %accit
cursor.execute(sql)
print "check_acct_available: " + sql
rs = cursor.fetchall()
if len(rs) != 1:
raise Exception("the account %s is not exist" %accit)
finally:
cursor.close()
def has_enough_money(self, src, money):
cursor = self.conn.cursor()
try:
sql = "select * from account where acctid = %s and money >= %s " %(src, money)
cursor.execute(sql)
print "has_enough_money: " + sql
rs = cursor.fetchall()
if len(rs) != 1:
raise Exception("the account does not have enough money")
finally:
cursor.close()
if __name__ == "__main__":
source_acctid = sys.argv[1]
target_acctid = sys.argv[2]
money = sys.argv[3]
conn = MySQLdb.connect(
host = "127.0.0.1", user = '******', passwd = '******', port = 3306, db = '******'
)
tr_money = TransferMoney(conn)
try:
tr_money.transfer(source_acctid, target_acctid, money)
except Exception as e:
print e
finally:
conn.close()

總結

#透過對資料庫的操作就可以實現一個簡單的銀行轉帳系統，所以在系統開發的時候，我們應該盡最大的可能，讓整個系統不只是多個元件的拼接，應該實現1+1>2。

更多Python實現一個簡單的銀行轉帳操作方法相關文章請關注PHP中文網！