import pymysql

# 第一步：连接数据库
con = pymysql.connect(
    host='localhost',
    user='root',
    password='zpvivo123..',
    database='Test',
    port=3306,
    cursorclass=pymysql.cursors.DictCursor)

# 第二步：创建游标
cur = con.cursor()

# 第三步：执行增删改查的sql语句
# sql = 'select * from books'

# result = cur.execute(sql)
# print(result)

# 获取查询结果
# res1 = cur.fetchall()
# print(res1[0])
# res2 = cur.fetchone()
# print(res2)

sql ='insert into books(name,position) value("python入门到放弃"，"A区2号5层");'
cur.execute(sql)
# 提交事务
con.commit()

# 关闭数据库
cur.close()
con.close()

# 代码封装
class DB:
    def __init__(self):
    # 第一步：连接数据库
       self.con = pymysql.connect(
           host='localhost',
           user='root',
           password='zpvivo123..',
           database='Test',
           port=3306,
           cursorclass=pymysql.cursors.DictCursor)
    # 第二步：创建游标
       self.cur.cursor()

    def query_sql(self,sql):
        """查询数据方法的封装"""
        self.cur.execute(sql)
        return self.cur.fetall()
    def updata_sql(self,sql):
       """增删改查sql的执行方法"""
       self.cur.execute(sql)
       return self.con.commit()

    def close(self):
       """断开数据库连接"""
       self.cur.colse()
       self.con.colse()