论文爬取（四）

关键字存储查询

#!/usr/bin/python
# -*- coding: UTF-8 -*-
import traceback

import pymysql

import pymysql.cursors

import time
import Keyword


# 获得conn cursor
def get_conn():

    # 打开数据库连接
    conn = pymysql.connect(host='localhost',
                           user='root',
                           passwd='123',
                           db='text3',
                           port=3306,
                           charset='utf8')

    # 创建游标，默认是元组型
    cursor = conn.cursor()
    return conn, cursor


# 关闭conn cursor
def close_conn(conn, cursor):
    if cursor:
        cursor.close()
    if conn:
        conn.close()


def query(sql, *args):
    conn, cursor = get_conn()
    cursor.execute(sql, args)
    res = cursor.fetchall()
    close_conn(conn, cursor)
    return res


def select_lunwenj(str, value):
        args = value
        sql = "select * from xiangxi where "+str+" = "+args
        # print(sql)
        res = query(sql)
        return res


def select_lunwenm(str):
    # args = '%%' + value + '%%'
    # sql = "select * from xiangxi where " + str + " like '%s'" % args
    sql = "select * from xiangxi where " + str
    print(sql)
    res = query(sql)
    return res


def select_ab():
    sql = "select abstract from xiangxi "
    res = query(sql)
    return res


def insert_key(word, count):
    cursor = None
    conn = None
    try:
        print(f"{time.asctime()}开始插入历史数据")
        conn, cursor = get_conn()
        sql = "insert into keyword values(%s,%s)"
        cursor.execute(sql, [word, count])  # %Y-%m-%d %H:%M:%S

        conn.commit()
        print(f"{time.asctime()}插入历史数据完毕")
    except:
        traceback.print_exc()
    finally:
        close_conn(conn, cursor)

def select_key():
    sql = "select * from keyword "
    res = query(sql)
    return res