usercontrol.py

python + wxpython + twisted 服务器和客户端demo代码

import MySQLdb

dataBase=None
userDataTable=None

class userControl():

    def __init__(self,host,databaseUser,databasePassword):
        self.host=host
        self.user=databaseUser
        self.password=databasePassword
        self.dataBase=dataBase
        try:
            self.conn=MySQLdb.connection(self.host,self.user,self.password,self.dataBase)
            self.cursor =self.conn.cursor()
        except:
            del self

    def findUser(self,user,password):
        sql="""select * from %s where name='%s' and password='%s'""" %(userDataTable,user,password)
        try:
            if(self.cursor.execute(sql)!=1L):
                print "没有找到用户"
                return -1
            else:
                print "找到用户"
                return 0
        except:
            print "sql执行异常"
            return -2

    def getUserFriends(self,user):
        sql="""select * from %sfriends """ %(user)
        try:
            self.cursor.execute(sql)
            print "获取了好友列表"
            return self.cursor.fetchall()
        except:
            print "sql执行异常"
            return -2

    def findUserFriend(self,user,userFriend):
        userFriends=self.getUserFriends(user)
        if userFriends== -2 :
            return -1
        if (userFriend,) in userFriends :
            print "存在此好友"
            return 0
        else:
            print "无此好友"
            return -1

    def checkUser(self,user):
        sql="""select * from %s where name='%s' """ %(userDataTable,user)
        try:
            if(self.cursor.execute(sql)==0L):
                print "没有找到用户"
                return 0
            else:
                print "找到用户"
                return -1
        except:
            print "sql执行异常"
            return -2

    def checkFriend(self,user,friend):
        sql="""select * from %sfriends where name='%s' """ %(user,friend)
        try:
            if(self.cursor.execute(sql)==0L):
                print "没有找到好友"
                return 0
            else:
                print "找到好友"
                return -1
        except:
            print "sql执行异常"
            return -2


    def addUserFriend(self,user,friend):
        if self.checkFriend(user,friend)!=0:
            return -1

        sql="""insert into %sfriends valuse('$s')"""%(user,friend)
        try:
            if(self.cursor.execute(sql)==1L):
                self.conn.commit()
                print "添加用户好友成功"
                return 0
            else:
                print "添加用户好友失败"
                return -1
        except:
            print "sql执行异常"
            return -2

    def deleteUserFriend(self,user,friend):
        if self.checkFriend(user,friend)!=1:
            return 1

        sql="""delete from %sfriends where name ='%s'"""%(user,friend)
        try:
            if(self.cursor.execute(sql)==1L):
                self.conn.commit()
                print "删除用户好友成功"
                return 0
            else:
                print "2 删除用户好友成功"
                return -1
        except:
            print "sql执行异常"
            return -2

    def addUserFriendsTable(self,user):
        sql="""create table %sfriends (friend varchar(45) NOT NULL)"""%(user)
        try:
            if(self.cursor.execute(sql)==1L):
                print "添加用户好友数据库表成功"
                return 0
            else:
                print "添加用户好友数据库表失败"
                return -1
        except:
            print "sql执行异常"
            return -2

    def deleteUserFriendsTable(self,user):
        sql="""drop table %sfriends """%(user)
        try:
            if(self.cursor.execute(sql)==1L):
                print "删除用户好友数据库表成功"
                return 0
            else:
                print "删除用户好友数据库表失败"
                return -1
        except:
            print "sql执行异常"
            return -2

    def addUser(self,user,password):
        if self.checkUser(user)!=0:
            return -1

        sql="""insert into %s valuse('$s','%s','%sfriends')"""%(userDataTable,user,password,user)
        try:
            if(self.cursor.execute(sql)==1L):
                if(self.addUserFriendsTable(user)==0):
                    self.conn.commit()
                    print "添加用户成功"
                    return 0
                else:
                    self.conn.rollback()
                    print "2 添加用户失败"
                    return -1
            else:
                print "1 添加用户失败"
                return -1
        except:
            print "sql执行异常"
            return -2

    def deleteUser(self,user):
        if self.checkUser(user)!=1:
            return 1

        sql="""delete from %s where name ='%s'"""%(userDataTable,user)
        try:
            if(self.cursor.execute(sql)==1L):
                if(self.deleteUserFriendsTable(user)==0):
                    self.conn.commit()
                    print "删除用户成功"
                    return 0
                else:
                    self.conn.rollback()
                    print "1 删除用户失败"
                    return -1
            else:
                print "2 删除用户成功"
                return -1
        except:
            print "sql执行异常"
            return -2