01、与mysql建立连接，进行一些操作(数据的增删改查)

注意：
python连接mysql，每完成一个cursor最好就要关闭一次cursor。连接可以不用关闭，直到你不再使用当前连接，再关闭连接。python是解释执行，建立数据库连接后，会占用解释器线程，当你不关闭连接时，那么这个线程一直占用，会使执行其他程序变慢。如果不关闭虽然也会过期，但是会较长时间占用mysql宝贵的连接资源。

import pymysql


def main():
    # 1.建立连接
    con_obj = pymysql.connect(
        host='localhost',
        user='root',
        password='zhuming',
        database='localdata',
        port=3306,
        charset='utf8',
        autocommit=True)

    # 2.获取游标对象 - 提供数据库操作的上下文
    # 注意：游标提供的上下文是事务环境
    with con_obj.cursor() as cursor_obj:
    # 在这个里面操作数据库
    # 操作数据库，执行sql语句：游标对象.execute(sql语句)
    # 返回执行结果，这个sql操作影响的行

        # 1)创建数据库
        # base_name = input('需要创建的数据库的名字(必须是英文)：')
        # sqlstr = "create database if not exists %s default charset utf8;" % base_name
        # result = cursor_obj.execute(sqlstr)
        # print(result)

        # 2)删除数据库
        # base_name = input('需要创建的数据库的名字(必须是英文)：')
        # cursor_obj.execute('drop database if exists %s;' % base_name)

        # 3)使用数据库
        cursor_obj.execute('use python1;')

        # 4)创建表
        # cursor_obj.execute("""create table if not exists tb_person
        # (
        #     pid int not null,
        #     pname varchar(20) not null,
        #     page int default 0,
        #     gender bit default 0,
        #     primary key(pid)
        # );
        # """)

        # 5)删除表
        # cursor_obj.execute("drop table if exists tb_person;")
        #
        # # 6)增删改
        # cursor_obj.execute("""
        # insert into tb_person
        # (pid,pname,page,gender)
        # values
        # (1, 'zhangsan', 18, 1),
        # (2, 'lisi', 19, 0),
        # (3, '小明',20, 1);
        # """)
        # cursor_obj.execute('delete from tb_person where pid=1;')
        # cursor_obj.execute('update tb_person set pname="hhh" where pname="ttt";')

        # 7)查：通过游标对象执行查询语句后，查询结果是保存在游标对象中
        result = cursor_obj.execute('select pname,pid,page from tb_person')
        print(result)

        # 游标对象.fetchall()  -  获取查询结果，
        # 结果是个元祖，元祖中的元素是一个小元祖代表的是每一条记录；小元祖中的元素是每一条记录中每个字段对应的值
        # print(cursor_obj.fetchall())
        # 通过游标获取结果的时候，取一个就少一个
        all_person = cursor_obj.fetchall()
        for p in all_person:
            print('姓名：', p[0])
            print('id：', p[1])
            print('age:', p[2])

        print(cursor_obj.fetchone())  # none
        print(cursor_obj.fetchall())  # ()

        # 事务提交
        # 注意：如果在建立连接的时候给参数 autocommit赋值为True，后面就不用手动提交了
        # con_obj.commit()
    # 关闭连接
    con_obj.close()

    connect_obj = pymysql.connect(
        host='localhost',
        user='root',
        password='zhuming',
        port=3306,
        charset='utf8',
        autocommit=True
    )

    with connect_obj.cursor(pymysql.cursors.DictCursor) as cursor_obj:
        cursor_obj.execute('use python1;')
        cursor_obj.execute('select * from tb_person;')
        re = cursor_obj.fetchall()
        print(re)

    connect_obj.close()


if __name__ == '__main__':
    main()

02、对象存储(对象数据与数据库数据的交互)

import pymysql


class Mountain:
    def __init__(self, name, altitude, location):
        self.name = name
        self.altitude = altitude
        self.location = location


mountains = [
    Mountain('雀儿山', 6168, '四川'),
    Mountain('半脊峰', 5430, '四川'),
    Mountain('玉珠峰', 6178, '青海'),
    Mountain('慕士塔格峰', 7546, '新疆'),
    Mountain('贡嘎山', 7556, '四川')
]


# obj, 通过类创建对象，类名是唯一的，根据类名创建表名
def insert_obj_database(obj: object, cursor_obj):
    class_name = obj.__class__.__name__
    table_name = 'tb_' + class_name
    # 插入表的时候表的字段根据对象属性来确定
    keys = ",".join(obj.__dict__.keys())

    # 插入表的时候表的字段的值是根据对象属性的值来确定
    values = ""
    for value in obj.__dict__.values():
        if isinstance(value, str):
            values += "'%s'" % value
        else:
            values += '%s' % value
        values += ','
    values = values[:-1]

    insert_sql = "insert into %s(%s) values (%s);" % (table_name, keys, values)
    # 如果表已经存在，那就直接插入，如果没有表，就先创建再插入
    try:
        cursor_obj.execute(insert_sql)
    except pymysql.err.ProgrammingError:
        # 创建表。表的字段根据对象的属性来确定，并且要加上类型和约束（约束不一定）
        filed = ""
        for key in obj.__dict__:
            value = obj.__dict__[key]
            if isinstance(value, str):
                filed += '%s text' % key
            else:
                filed += '%s float' % key
            filed += ','
        filed = filed[:-1]

        create_sql = "create table %s (%sid int not null auto_increment, %s, primary key(%sid));" % (table_name, table_name[3:6], filed, table_name[3:6])
        cursor_obj.execute(create_sql)
        cursor_obj.execute(insert_sql)


def main():
    # 建立连接
    connection_obj = pymysql.connect(
        host='localhost',
        user='root',
        password='zhuming',
        port=3306,
        charset='utf8',
        autocommit=True
    )

    # 获取游标
    # with connection_obj.cursor() as cursor_obj:
    with connection_obj.cursor(pymysql.cursors.DictCursor) as cursor_obj:
        cursor_obj.execute('use python1;')
        
        # 将对象数据插入数据库中对应的表
        # for moun in mountains:
        #     insert_obj_database(moun, cursor_obj)

        # insert_obj_database(Mountain('雪宝顶', 5588, '四川'), cursor_obj)
        
        # 从数据库中取出表的数据并转化成对应的对象
        select_sql = "select * from tb_mountain;"
        cursor_obj.execute(select_sql)
        re = cursor_obj
        for item in re:
            print(item)
            del item['Mouid']
            moun = Mountain(**item)
            print(moun)

    # 关闭连接
    connection_obj.close()


if __name__ == '__main__':
    main()