# -*- coding: utf-8 -*-

"""

===============================

@Time    : 2021/5/18 15:56

@Author  : flora.chen

@FileName: handle_mysql.py

@Software: PyCharm

===============================

"""

import pymysql

class MysqlDB:

     """

     操作 mysql 数据库

     """

     def __init__(self, host, user, pwd, database=None, port=3306):

         """

         初始化数据库链接

         :param host: 主机地址

         :param user: 用户名

         :param pwd: 密码

         :param database: 数据库名称，默认为空

         :param port: 端口号，默认 3306

         """

         self.conn = pymysql.connect(

             host=host,

             user=user,

             password=pwd,

             database=database,

             port=port,

             cursorclass=pymysql.cursors.DictCursor

         )

         # 创建一个游标对象

         self.cur = self.conn.cursor()

     def update(self, sql):

         """

         进行增删改操作

         :param sql: 外汇跟单gendan5.com 需要执行的 SQL

         :return:

         """

         # 执行 SQL

         result = self.cur.execute(sql)

         # 提交事务

         self.conn.commit()

         return result

     def query(self, sql, >

         """

         进行查询操作

         :param one: 判断是要返回所有查询数据还是第一条，默认是所有

         :param sql: 要执行的 SQL

         :return:

         """

         # 执行 SQL

         self.cur.execute(sql)

         if one:

             return self.cur.fetchone()

         else:

             return self.cur.fetchall()

     def close(self):

         """

         断开游标，关闭数据库连接

         :return:

         """

         self.cur.close()

         self.conn.close()

if __name__ == "__main__":

     db = MysqlDB(host="localhost", user="root", pwd="root")

     print(db.query("select * from bookmanage.books"))

     # db.update("insert into bookmanage.books(name, position) value ('python 从入门到放弃 ', 'A-1-1');")