本文實例為大家分享了Python3連接MySQL模擬轉賬的具體實現代碼，供大家參考，具體內容如下
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
# coding:utf8import sysimport pymysql class TransferMoney(object): def __init__(self,conn): self.conn=conn def check_acct_available(self,acctid): cursor = self.conn.cursor() try: sql="select * from account where acctid=%s" % acctid cursor.execute(sql) print ("check_acct_available:" + sql) rs = cursor.fetchall() if len(rs) ! = 1: raise Exception("賬號%s不存在"% acctid) finally: cursor.close() def has_enough_money(self,acctid,money): cursor = self.conn.cursor() try: sql="select * from account where acctid=%s and money>%s" % (acctid,money) cursor.execute(sql) print ("has_enough_money:"+sql) rs = cursor.fetchall() if len(rs) ! = 1: raise Exception("賬號%s余額不足"% acctid) finally: cursor.close() def reduce_money(self,acctid,money): cursor = self.conn.cursor() try: 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() def add_money(self,acctid,money): cursor = self.conn.cursor() try: sql="update account set money=money+%s where acctid=%s" % (money,acctid) cursor.execute(sql) print ("add_money:"+sql) if cursor.rowcount ! = 1: raise Exception("賬號%s加款失敗" % acctid) finally: cursor.close() def transfer(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] conn = pymysql.Connect( host = 'localhost', unix_socket = "..mysql/mysql.sock", port = 3306, user = 'root', passwd = '', db = 'python_db', ) tr_money = TransferMoney(conn) try: tr_money.transfer(source_acctid,target_acctid,money) except Exception as e: print ("出現問題" + str(e)) finally: conn.close() |
以上就是本文的全部內容，希望對大家學習python程序設計有所幫助。
