Python每日一练——数据存储第六关:操作MySQL数据库

Posted 孤寒者

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Python每日一练——数据存储第六关:操作MySQL数据库相关的知识,希望对你有一定的参考价值。

面试题第六关:

第一部分——考点:

操作mysql数据库:

  1. 创建MySQL数据表;
  2. 向表中插入记录;
  3. 其他数据库操作。

第二部分——面试题:

1.面试题一:如何创建MySQL数据表?

2.面试题二:如何向MySQL表中插入数据?

3.面试题三:如何查询MySQL中的数据?


第三部分——解析:

面试题一 之 创建MySQL数据表:

# coding=utf-8
# _author__ = 孤寒者

from pymysql import *

def connectDB():
    '''
    连接本地MySQL数据库,指定连接的库为test库。
    :return:
    '''
    db = connect(host='localhost', user='root', password='123456', port=3306, db='test')
    return db
db = connectDB()
print(type(db))

def createTable(db):
    c = db.cursor()
    try:
        c.execute('''create table persons
                    (id int primary key not null,
                    name text not null,
                    age int not null,
                    address char(100),
                    salary real);''')
        db.commit()
        db.commit()
        return True
    except:
        db.rollback()
    return False
if createTable(db):
    print('create table success')
else:
    print('create table failed')


使用navicat工具查看:

面试题二 之 向MySQL表中插入数据:

# coding=utf-8
# _author__ = 孤寒者

from pymysql import *

def connectDB():
    '''
    连接本地MySQL数据库,指定连接的库为test库。
    :return:
    '''
    db = connect(host='localhost', user='root', password='123456', port=3306, db='test')
    return db
db = connectDB()
print(type(db))


def insertRecords(db):
    cursor = db.cursor()
    try:
        cursor.execute("delete from persons")
        cursor.execute('''
        insert into persons(id,name,age,address,salary)
        values(1, 'GuHanZhe', 18, 'China', 9999)
        ''')
        cursor.execute('''
        insert into persons(id,name,age,address,salary)
        values(2, 'XiaoZhang', 55, 'China', 9)
        ''')
        db.commit()
        return True
    except Exception as e:
        print(e)
        db.rollback()
    return False

if insertRecords(db):
    print("成功插入记录")
else:
    print("插入记录失败")

使用navicat工具查看:

面试题三 之 查询MySQL中的数据:

# coding=utf-8
# _author__ = 孤寒者
from pymysql import *

def connectDB():
    '''
    连接本地MySQL数据库,指定连接的库为test库。
    :return:
    '''
    db = connect(host='localhost', user='root', password='123456', port=3306, db='test')
    return db
db = connectDB()


def selectRecords(db):
    cursor = db.cursor()
    sql = 'select name,age,salary from persons order by age desc'
    cursor.execute(sql)
    results = cursor.fetchall()
    print(results)
    print(type(results))    # 打印发现是元组类型

selectRecords(db)
db.close()

  • 我们发现查询数据输出类型是元组类型,如果我们想要将字段名和查询出的数据一一对应该怎么做呢?
  • 这里就需要用到两个很常用的函数dict()和zip(),如下:
# coding=utf-8
# _author__ = 孤寒者
import json

from pymysql import *

def connectDB():
    '''
    连接本地MySQL数据库,指定连接的库为test库。
    :return:
    '''
    db = connect(host='localhost', user='root', password='123456', port=3306, db='test')
    return db
db = connectDB()


def selectRecords(db):
    cursor = db.cursor()
    sql = 'select name,age,salary from persons order by age desc'
    cursor.execute(sql)
    results = cursor.fetchall()
    print(results)
    print(type(results))    # 打印发现是元组类型

    # 将字段名和查询结果整合在一起
    fields = ['name', 'age', 'salary']
    records = []
    for row in results:
        records.append(dict(zip(fields, row)))
    return json.dumps(records)      # 输出类型为JSON字符串


endresults = selectRecords(db)
print(endresults)
db.close()


第四部分——总结:

  • 注意:我们使用的是pymysql模块中的API来操作MySQL数据库,该模块需要单独安装哦~

以上是关于Python每日一练——数据存储第六关:操作MySQL数据库的主要内容,如果未能解决你的问题,请参考以下文章

Python每日一练——数据存储第五关:操作SQLite数据库

Python每日一练——数据存储第七关:ORM框架知多少

Python每日一练——数据存储第二关:XML文档和字典的互转

Python每日一练——数据存储第三关:如何将一个JSON文档映射为Python对象

Python每日一练——数据存储第一关:读取XML节点和属性值的方法

Python每日一练——数据存储第四关:如何将一个Python对象转换为对应的JSON字符串