#引入开发包
import sys
import pymysql
#创建TransferMoney类
class TransferMoney(object):
#创建TransferMoney类构造函数并定义属性connection接收外部connection
def __init__(self, connection):
self.connection = connection
#创建transfer方法
def transfer(self,source_acctid, target_acctid, money):
try:
self.check_acct_avaliable(source_acctid)
self.check_acct_avaliable(target_acctid)
self.has_enough_money(source_acctid, money)
self.reduce_money(source_acctid, money)
self.add_money(target_acctid, money)
self.connection.commit()
except Exception as e:
self.connection.rollback()
raise e
#创建check_acct_avaliable函数，检查账户可用性
def check_acct_avaliable(self, acctid):
cursor = self.connection.cursor()
try:
sql = "select * from account where acctid = %s" % acctid
cursor.execute(sql)
print("check_acct_avaliable"+sql)
rs = cursor.fetchall()
if len(rs) != 1:
raise Exception("账号不存在"%acctid)
finally:
cursor.close()
#创建has_enough_money函数，检查账户的钱是否足够转账
def has_enough_money(self, acctid, money):
try:
cursor = self.connection.cursor()
sql = "select * from account where acctid = %s and money >=%s" % (acctid, money)
cursor.execute(sql)
rs = cursor.fetchall()
if len(rs) != 1:
raise Exception("账号%s没有足够的钱" % acctid)
finally:
cursor.close()
# 创建reduce_money函数，转出账户钱减少
def reduce_money(self, acctid, money):
try:
cursor = self.connection.cursor()
sql = "update account set money = money-%s where acctid = %s " % (money,acctid )
cursor.execute(sql)
print("reduce_money:"+sql)
if cursor.rowcount != 1:
raise Exception("账号%s减款失败" % acctid)
finally:
cursor.close()
# 创建add_money函数，转入账户钱增加
def add_money(self, acctid, money):
try:
cursor = self.connection.cursor()
sql = "update account set money = money+%s where acctid = %s " % (money, acctid, )
cursor.execute(sql)
print("reduce_money:"+sql)
if cursor.rowcount != 1:
raise Exception("账号%s加款失败" % acctid)
finally:
cursor.close()
#程序主入口
if __name__== "__main__":
#通过argv从程序外部获取实参
source_acctid = sys.argv[1]
target_acctid = sys.argv[2]
money = sys.argv[3]
connection = pymysql.connect(host='localhost',
user='root',
password='123456',
db='wikiurl',
charset='utf8')
#创建TransferMoney类的对象tr_money
tr_money = TransferMoney(connection)
try:
tr_money.transfer(source_acctid, target_acctid, money)
except Exception as e:
print("出现问题"+str(e))
finally:
connection.close()
