【Python学习笔记】-利用MySQLdb操作mysql数据库

超、凢脫俗 2022-08-19 13:22 175阅读 0赞

Python要连接mysql数据库，必须要用到MySQLdb。安装好mysql数据库和MySQLdb之后，就可以用Python来操作mysql数据库了。Python操作mysql数据库的主要步骤如下：

（1）建立数据库连接

（2）获取操作游标

（3）选择数据库

（4）进行一系列数据库操作

（5）提交事务，这一步很重要，只有提交事务后，数据才能真正写入数据库

（6）关闭操作游标

（7）关闭数据库连接

以下是将生成的激活码存储到mysql数据库中的实例。

#!/usr/bin/python                                                               
    # -*- coding:utf-8 -*- 
    
    import MySQLdb
    import random
    import string
    
    def gene_active_code(number,length):
        '''
        @number:生成激活码的个数
        @length:生成激活码的长度
        '''
        result = {}
        source = list(string.ascii_uppercase)
        for index in range(0,10):
            source.append(str(index))
        while len(result) < number:
            key= ''
            for index in range(length):
                key += random.choice(source)
            if key in result:
                pass
    	     else:
                result[key] = 1
        return result
    
    def save_mysql(host,user,passwd,port,number,length):
    	  '''
        @user:数据库用户名
        @passwd:密码
        @port:端口
        @number:生成激活码的个数
        @length:生成激活码的长度
        '''
        result = gene_active_code(number,length)
        values = []
        index = 0;
        for key in result:
            values.append((index,key))
            index += 1
        conn = MySQLdb.connect(host=host,user=user,passwd=passwd,port=port)	#连接数据库
        conn.select_db('test')	#选择数据库
        cur = conn.cursor()		#获取操作游标
        cur.execute('create table activecode(id int,name varchar(20))')  		#创建数据表
        cur.executemany('insert into activecode values(%s,%s)',values)		#向数据表中插入数据
        conn.commit()	#提交事务
        cur.close()		#关闭操作游标
        conn.close()		#关闭数据库连接
    
    if __name__ == "__main__":
        host = 'localhost'
        user = 'root'
        passwd = '612412'
        port = 3306
        number = 10
        length = 16
        save_mysql(host,user,passwd,port,number,length)

执行完代码后，查看数据库中的数据，如下所示：

![Center][]

[Center]: /images/20220731/e9bb8c7aa0cc4b709c7299aa73c60e37.png