2020.2.15

今天将通过python爬取到的数据录入数据库

学习地址https://www.bilibili.com/video/av68313310?from=search&seid=9830564145962665343

代码如下：db.py完成数据库的连接以及sql语句的调用

import pymysql

"""
常用模块：读写MySQL
"""


def get_conn():
    """
    获取MySQL的链接
    :return: mysql connection
    """
    return pymysql.connect(
        host='127.0.0.1',
        user='root',
        password='12345678',
        database='test',
        charset='utf8'
    )


def query_data(sql):
    """
    根据SQL查询数据并且返回
    :param sql: SQL语句
    :return: list[dict]
    """
    conn = get_conn()
    try:
        cursor = conn.cursor(pymysql.cursors.DictCursor)
        cursor.execute(sql)
        return cursor.fetchall()
    finally:
        # 别忘了关闭链接
        conn.close()


def insert_or_update_data(sql):
    """
    执行新增insert或者update的sql
    :param sql: insert or update sql
    :return: 不返回内容
    """
    conn = get_conn()
    try:
        cursor = conn.cursor()
        cursor.execute(sql)
        # 注意这里，只有commit才可以生效
        conn.commit()
    finally:
        # 别忘了关闭链接
        conn.close()

在通过函数调用生气了语句完成增删改查