24 python mysql

python操作数据

    安装mysql模块儿

                                                                       
1 import pymysql
2 connection==pymysql.connect("127.0.0.1","root", "123456", "test")
  3 cursor = connection.cursor()
  4 try:
  5 
  6     sql = "create table test(id int(4))"
  7     
  8     cursor.execute(sql)
  9 except Exception as ex:
 10     print(ex)
 11 finally:
 12     cursor.close()
 13     connection.close()

mysql封装

import pymysql

class MysqlHelper(object):
        config = {
            "host":"127.0.0.1",
            "user":"root",
            "password":"123456",
            "db":"test",
            "charset":"utf8"
        }
        def __init__(self):
            self.connection = None
            self.cursor = None

        #从数据库中查询一行数据
        def getOne(self, sql, *args):
            try:
                self.connection = pymysql.connect(**MysqlHelper.config)
                self.cursor = self.connection.cursor()
                self.cursor.execute(sql,args)
                return self.cursor.fetchone()
            except Exception as ex:
                print(ex)
            finally:
                self.close()

        # 从数据库中查询多行数据
        def getList(self, sql, *args):
            try:
                self.connection = pymysql.connect(**MysqlHelper.config)
                self.cursor = self.connection.cursor()
                self.cursor.execute(sql, args)
                return self.cursor.fetchall()
            except Exception as ex:
                print(ex)
            finally:
                self.close()

        # 对数据库进行增删改 返回影响的行数
        def executeDML(self, sql, *args):
            try:
                self.connection = pymysql.connect(**MysqlHelper.config)
                self.cursor = self.connection.cursor()
                num = self.cursor.execute(sql, args)
                self.connection.commit()
                return num
            except Exception as ex:
                self.connection.rollback()
                print(ex)
            finally:
                self.close()

        def close(self):
            if self.cursor:
                self.cursor.close()
            if self.connection:
                self.connection.close()

if __name__ == "__main__":
    mysql = MysqlHelper()
    sql = "create table test(id int(4))"

    print( mysql.executeDML(sql))