LongArticlesJob/applications/longArticlesMysql.py

"""
@author: luojunhui
"""
import pymysql


class longArticlesMySQL(object):
    """
    PQ Mysql
    """
    connection = pymysql.connect(
        host='rm-bp14529nwwcw75yr1ko.mysql.rds.aliyuncs.com',
        port=3306,
        user='changwen_admin',
        password='changwen@123456',
        db='long_articles',
        charset='utf8mb4'
    )

    @classmethod
    def update(cls, sql, params):
        """
        更新
        :return:
        """
        cursor = cls.connection.cursor()
        cursor.execute(sql, params)
        affected_rows = cursor.rowcount
        cls.connection.commit()
        return affected_rows

    @classmethod
    def select(cls, sql):
        """
        查询
        :param sql:
        :return:
        """
        cursor = cls.connection.cursor()
        cursor.execute(sql)
        result = cursor.fetchall()
        return result

    @classmethod
    def select_json(cls, sql):
        """
        查询
        :param sql:
        :return:
        """
        cursor = cls.connection.cursor()
        cursor.execute(sql)
        result = cursor.fetchall()
        json_data = [
            dict(
                zip([column[0] for column in cursor.description], row)
            )
            for row in result
        ]
        return json_data

    @classmethod
    def insertMany(cls, sql, params_list):
        """
        :param sql:
        :param params_list:
        :return:
        """
        cursor = cls.connection.cursor()
        try:
            cursor.executemany(query=sql, args=params_list)
            affected_rows = cursor.rowcount
            cls.connection.commit()
            return affected_rows
        except Exception as e:
            print("Insert Many Defeat--{}".format(e))
            cls.connection.rollback()
            raise e