使用Python操作MySQL数据库

Posted 2023-01-01 randysun

使用Python操作mysql数据库

一、安装PyMySQL的安装

pip install pymysql

二、连接数据库步骤

• pymysql连接数据库的必要参数：主机、端口、用户名、密码、数据库
• 注：pymysql不能提供创建数据库的服务，数据库要提前创建

import pymysql
# 1）建立数据库连接对象 conn
conn = pymysql.connect(user='root', passwd='root', database='db')
# conn = pymysql.connect(user='root', passwd='root', database='db', autocommit=True)

# 2）通过 conn 创建操作sql的 游标对象
# 注：游标不设置参数，查询的结果就是数据元组，数据没有标识性
# 设置pymysql.cursors.DictCursor，查询的结果是字典，key是表的字段

cursor = conn.cursor(pymysql.cursors.DictCursor)

# 3）编写sql交给 cursor 执行
sql='create, select, insert, update, delete.....'

# 4）如果是查询，通过 cursor对象 获取结果
cursor.execute(sql)

# 5）操作完毕，端口操作与连接断开连接

cursor.close()
conn.close()

三、创建表

import pymysql

# 1.建立数据连接对象 conn

conn = pymysql.connect(user='root', password='root', database='db1')

# 2 通过conn创建sql的游标对象
cursor = conn.cursor(pymysql.cursors.DictCursor)

# 3.编写sql语句

sql = 'create table test(id int, x int, y int)'

# 4.执行sql语句

cursor.execute(sql)

# 5.关闭
cursor.close()
conn.close()

四、插入数据

import pymysql

# 1.建立数据连接对象 conn

conn = pymysql.connect(user='root', password='root', database='db1')

# 2 通过conn创建sql的游标对象
cursor = conn.cursor(pymysql.cursors.DictCursor)

# 3.编写sql语句

# 增加一条数据
# sql= 'insert into test1 values(1, 11,22)'

# 通过参数插入
sql = 'insert into test1 values(%s,%s,%s)'

# 4.执行sql语句

# 增加一条数据
# row = cursor.execute(sql, (1, 2, 100))

# 增加多条数据
row = cursor.executemany(sql, [(2, 22, 22), (3, 33, 22)])

# 5.提交sql

## 在创建conn对象时，不设置autocommit，默认开启事务，增删改操作不会直接映射到数据库中，
# 需要执行 conn.commit() 动作
conn.commit()
if row:
    print("插入成功")
# 5.关闭
cursor.close()
conn.close()

五、删除数据

import pymysql
"""
@author RansySun
@create 2019-09-27-11:45
"""

conn = pymysql.connect(user='root', password='root', database='db1')

curson = conn.cursor(pymysql.cursors.DictCursor)

sql = 'delete from test1 where id=5'

row = curson.execute(sql)

conn.commit()

if row:
    print("删除成功")

   
cursor.close()
conn.close()

六、修改数据

import pymysql
"""
@author RansySun
@create 2019-09-27-11:48
"""

# 创建连接对象
conn = pymysql.connect(user='root', password='root', database='db1')

# 创建游标
cursor = conn.cursor(pymysql.cursors.DictCursor)

# 编写sql语句
sql = 'update test1 set x=%s where id=2'

# 执行sql语句
row = cursor.execute(sql, 10000)

# 提交sql语句
conn.commit()

if row:
    print('修改成功')

cursor.close()
conn.close()

七、查询数据

Python查询Mysql使用 fetchone() 方法获取单条数据，使用

• fetchall()：方法获取多条数据。
• fetchone(): 该方法获取下一个查询结果集。结果集是一个对象
• fetchall(): 接收全部的返回结果行.
• rowcount(): 这是一个只读属性，并返回执行execute()方法后影响的行数。

import pymysql

"""
@author RansySun
@create 2019-09-27-11:48
"""
# 1.建立数据连接对象 conn

conn = pymysql.connect(user='root', password='root', database='db1')

# 2 通过conn创建sql的游标对象
cursor = conn.cursor(pymysql.cursors.DictCursor)

# 3.编写sql语句 查询语句
sql = 'select * from test1'

# 4.执行sql语句
try:
    row = cursor.execute(sql)
    print('row', row)

    # 1获取全部记录
    # res = cursor.fetchall()
    # for row in res:
    #     print(row)

    # 2获取一条记录
    res = cursor.fetchone()
    print(res)

    # 3.获取指定条数
    res = cursor.fetchmany(3)
    print(res)

except:
    conn.rollback()

# 5.关闭
cursor.close()
conn.close()

八、游标操作

import pymysql

"""
@author RansySun
@create 2019-09-27-12:07
"""
# 1.建立数据连接对象 conn

conn = pymysql.connect(user='root', password='root', database='db1')

# 2 通过conn创建sql的游标对象
cursor = conn.cursor(pymysql.cursors.DictCursor)

# 3.编写sql语句

sql = 'select * from test1'

# 4.如果是查询，通过 cursor对象 获取结果
try:
    cursor.execute(sql)

    # absolute绝对偏移，游标重置，从头(0)开始偏移
    # cursor.scroll(3, 'absolute')
    # r2 = cursor.fetchone(5)
    # print(r2)

    # relative相对偏移，游标在当前位置进行左右偏移
    cursor.scroll(2, 'relative')
    res = cursor.fetchmany(1)
    print(res)

except:
    conn.rollback()

cursor.close()
conn.close()

九、pymysql事务

import pymysql

"""
@author RansySun
@create 2019-09-27-12:15
"""

# 1.建立数据连接对象 conn

conn = pymysql.connect(user='root', password='root', database='db1')

# 2 通过conn创建sql的游标对象
cursor = conn.cursor(pymysql.cursors.DictCursor)

# 3.编写sql语句
crete_sql = 'create table t3(id int, name char(4), money int)'
try:

    row = cursor.execute(crete_sql)
    print(row)
except:
    print('表已创建')
    pass

# 空表才插入

# 空表才插入
row = cursor.execute('select * from t3')
if not row:
    inert_sql = 'insert into t3 values(%s,%s,%s)'
    row = cursor.executemany(inert_sql, [(1, 'tom', 10), (2, 'Bob', 10)])
    conn.commit()

"""
try:
    sql1 = 'update t3 set money=money-1 where name="tom"'
    cursor.execute(sql1)
    sql2 = 'update t3 set money=money+1 where name="ob"'
    cursor.execute(sql2)
except:
    print('转账执行异常')
    conn.rollback()
else:
    print('转账成功')
    conn.commit()
"""

try:
    sql1 = 'update t3 set money=money-1 where name="tom"'
    r1 = cursor.execute(sql1)
    sql2 = 'update t3 set money=money+1 where name="laowang"'  # 转入的人不存在
    r2 = cursor.execute(sql2)
except:
    print('转账执行异常')
    conn.rollback()
else:

    if r1 and r2:
        print('转账成功')
        conn.commit()
    else:
        print("转账失败")
        conn.rollback()

十、sql防注入

import pymysql
from pymysql.cursors import DictCursor
conn = pymysql.connect(user='root', passwd='root', db='oldboy')
cursor = conn.cursor(DictCursor)

try:
    sql = 'create table user(id int, name char(4), password char(6))'
    row = cursor.execute(sql)
    print(row)
except:
    print('表已创建')
    pass

# 空表才插入
row = cursor.execute('select * from user')
if not row:
    sql = 'insert into user values(%s,%s,%s)'
    row = cursor.executemany(sql, [(1, 'tom', '123'), (2, 'bob', 'abc')])
    conn.commit()



# 用户登录
usr = input('usr: ')
pwd = input('pwd: ')

# 自己拼接参数一定有sql注入，将数据的占位填充交给pymysql

"""
sql = 'select * from user where name="%s" and password="%s"' % (usr, pwd)
row = cursor.execute(sql)
if row:
    print('登录成功')
else:
    print('登录失败')
"""
sql = 'select * from user where name=%s and password=%s'
row = cursor.execute(sql, (usr, pwd))
if row:
    print('登录成功')
else:
    print('登录失败')


# 知道用户名时
# 输入用户时：
#   tom => select * from user where name="tom" and password="%s"
#   tom" # => select * from user where name="tom" #" and password="%s"

# 不自定义用户名时
#   " or 1=1 # => select * from user where name="" or 1=1 #" and password="%s"

十一、索引

索引就是 键 - key

1）键 是添加给数据库表的 字段
2）给表创建 键 后，该表不仅会形参 表结构、表数据，还有 键的B+结构图
3）键的结构图是需要维护的，在数据完成增、删、改操作时，只要影响到有键的字段，结构图都要维护一次, 所以创建键后一定会降低 增、删、改 的效率
4）键可以极大的加快查询速度(开发需求中，几乎业务都和查有关系)
5）建立键的方式：主键、外键、唯一键、index

import pymysql
from pymysql.cursors import DictCursor

conn = pymysql.connect(user='root', passwd='root', db='db1')
cursor = conn.cursor(DictCursor)
# 创建两张表

sql1 = """create table a1(
    id int primary key auto_increment,
    x int,
    y int
)"""

cursor.execute(sql1)

sql2 = """create table a2(
    id int primary key auto_increment,
    x int,
    y int,
    index(x)
)"""

cursor.execute(sql2)

# 每个表插入5000条数据
import random

for i in range(1, 5001):
    x = i
    y = random.randint(1, 5000)
    cursor.execute('insert into a1(x, y) values(%s, %s)', (x, y))
    cursor.execute('insert into a2(x, y) values(%s, %s)', (x, y))
    conn.commit()
    
    
import time

# a1的x、a1的id、a2的x

# a1的id时间
b_time = time.time()
sql = 'select * from a1 where id=4975'
cursor.execute(sql)
e_time = time.time()
print(e_time - b_time)


# a1 的x时间
b_time = time.time()
sql = 'select * from a1 where x=4975'
cursor.execute(sql)
e_time = time.time()
print(e_time - b_time)


# a2时间
b_time = time.time()
sql = 'select * from a2 where x=4975'
cursor.execute(sql)
e_time = time.time()
print(e_time - b_time)