Create a python based pymyql the client class

python code implementation

import pymysql

from qq_A_stock.configs import MYSQL_HOST, MYSQL_PORT, MYSQL_USER, MYSQL_PASSWORD, MYSQL_DB


class PyMysqBase(object):
    def __init__(self,
                 host='localhost',
                 port=3306,
                 user='user',
                 password='pwd',
                 db='db',
                 charset='utf8mb4',
                 cursorclass=pymysql.cursors.DictCursor,
                 ):
        self.connection = pymysql.connect(
            host=host,
            port=port,
            user=user,
            password=password,
            db=db,
            charset=charset,
            cursorclass=cursorclass
        )
        print("已经成功建立数据库连接")

    def _exec_sql(self, sql):
        with self.connection.cursor() as cursor:
            cursor.execute(sql)
        self.connection.commit()

    # def create_table(self, sql):
    #     self._exec_sql(sql)

    # def drop_table(self, sql):
    #     self._exec_sql(sql)

    def insert(self, sql, params):
        with self.connection.cursor() as cursor:
            cursor.execute(sql, params)
        self.connection.commit()

    def select_all(self, sql, params=None):
        with self.connection.cursor() as cursor:
            cursor.execute(sql, params)
            results = cursor.fetchall()
        return results

    def select_many(self, sql, params=None, size=1):
        with self.connection.cursor() as cursor:
            cursor.execute(sql, params)
            results = cursor.fetchmany(size)
        return results

    def select_one(self, sql, params=None):
        with self.connection.cursor() as cursor:
            cursor.execute(sql, params)
            result = cursor.fetchone()
        return result

    def __del__(self):
        self.connection.close()
        print("已经断开数据库连接")


if __name__ == "__main__":
    conf = {
        "host": MYSQL_HOST,
        "port": MYSQL_PORT,
        "user": MYSQL_USER,
        "password": MYSQL_PASSWORD,
        "db": MYSQL_DB,
    }

    demo = PyMysqBase(**conf)
    print(demo.connection)

    # # 测试建表语言
    # create_sql = """
    # CREATE TABLE `qq_Astock_news` (
    #   `id` int(11) NOT NULL AUTO_INCREMENT,
    #   `pub_date` datetime NOT NULL COMMENT '发布时间',
    #   `title` varchar(64) CHARACTER SET utf8 COLLATE utf8_bin DEFAULT NULL COMMENT '文章标题',
    #   `link` varchar(128) CHARACTER SET utf8 COLLATE utf8_bin DEFAULT NULL COMMENT '文章详情页链接',
    #   `article` text CHARACTER SET utf8 COLLATE utf8_bin COMMENT '详情页内容',
    #   `CREATETIMEJZ` datetime DEFAULT CURRENT_TIMESTAMP,
    #   `UPDATETIMEJZ` datetime DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
    #   PRIMARY KEY (`id`),
    #   UNIQUE KEY `link` (`link`),
    #   KEY `pub_date` (`pub_date`)
    # ) ENGINE=InnoDB AUTO_INCREMENT=38 DEFAULT CHARSET=utf8mb4 COMMENT='腾讯财经[A股]';
    #
    # """
    # demo._exec_sql("DROP TABLE IF EXISTS `qq_Astock_news`;")
    # demo._exec_sql(create_sql)

    # 测试删除数据库
    # drop_sql = """drop table qq_Astock_news; """
    # demo._exec_sql(drop_sql)

    # # 测试插入一条数据
    # insert_sql = """
    # insert into `qq_Astock_news` (`pub_date`, `title`, `link`, `article`) values (%s,%s,%s,%s);
    # """
    # values = ("2020-01-01", "我是标题", "我是链接3", "我是正文")
    # demo.insert(insert_sql, values)

    # # 测试查询全部数据
    # sql = """select * from qq_Astock_news; """
    # ret = demo.select_all(sql)
    # print(ret)

    # # 测试查询两条数据
    # sql = """select * from qq_Astock_news;"""
    # ret = demo.select_many(sql, size=3)
    # print(ret)

    # # 测试查询单条数据
    # sql = """select * from qq_Astock_news;"""
    # ret = demo.select_one(sql)
    # print(ret)

postscript

Updated: 2020-02-10