python def demo_Python操作数据库demo

-- coding: utf-8 --

import sys

import MySQLdb

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

res = cursor.fetchall()

if len(res)!=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

if cursor.rowcount != 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 = MySQLdb.connect(user='root', db='demo', passwd='123', host='127.0.0.1', port=3306)

tr_money = transfermoney(conn)

try:

tr_money.transfer(source_acctid,target_acctid,money)

except Exception as e:

print "出现问题"+str(e)

finally:

conn.close()