title_with_video/applications/db/__init__.py

"""
@author: luojunhui
self.app.mysql_pool = await aiomysql.create_pool(
            host='rm-bp1159bu17li9hi94.mysql.rds.aliyuncs.com',
            port=3306,
            user='crawler',
            password='crawler123456@',
            db='piaoquan-crawler',
            charset='utf8mb4',
            connect_timeout=120,
        )
"""
import aiomysql


class AsyncMySQLClient(object):
    """
    Async MySQL
    """

    def __init__(self, app):
        self.app = app

    async def initPool(self):
        """
        初始化连接
        host='',
        port=3306,
        user='changwen_admin',
        password='changwen@123456',
        db='long_articles',
        charset='utf8mb4'
        :return:
        """
        self.app.mysql_pool = await aiomysql.create_pool(
            host='rm-bp14529nwwcw75yr1ko.mysql.rds.aliyuncs.com',
            port=3306,
            user='changwen_admin',
            password='changwen@123456',
            db='long_articles',
            charset='utf8mb4',
            connect_timeout=120,
        )
        print("mysql init successfully")

    async def closePool(self):
        """
        关闭 mysql 连接
        :return:
        """
        self.app.mysql_pool.close()
        await self.app.mysql_pool.wait_closed()

    async def asyncSelect(self, sql):
        """
        select method
        :param sql:
        :return:
        """
        async with self.app.mysql_pool.acquire() as conn:
            async with conn.cursor() as cursor:
                await cursor.execute(sql)
                result = await cursor.fetchall()
                return result

    async def asyncInsert(self, sql, params):
        """
        insert and update method
        :param params:
        :param sql:
        :return:
        """
        async with self.app.mysql_pool.acquire() as coon:
            async with coon.cursor() as cursor:
                try:
                    await cursor.execute(sql, params)
                    await coon.commit()
                except Exception as e:
                    await coon.rollback()
                    raise


class TaskMySQLClient(object):
    """
    Async MySQL
    """

    def __init__(self):
        self.mysql_pool = None

    async def init_pool(self):
        """
        初始化连接
        :return:
        """
        self.mysql_pool = await aiomysql.create_pool(
            host='rm-bp14529nwwcw75yr1ko.mysql.rds.aliyuncs.com',
            port=3306,
            user='changwen_admin',
            password='changwen@123456',
            db='long_articles',
            charset='utf8mb4',
            connect_timeout=120
        )
        print("mysql init successfully")

    async def close_pool(self):
        """
        关闭 mysql 连接
        :return:
        """
        self.mysql_pool.close()
        await self.mysql_pool.wait_closed()

    async def asyncSelect(self, sql):
        """
        select method
        :param sql:
        :return:
        """
        async with self.mysql_pool.acquire() as conn:
            async with conn.cursor() as cursor:
                await cursor.execute(sql)
                result = await cursor.fetchall()
                return result

    async def asyncInsert(self, sql, params):
        """
        insert and update method
        :param params:
        :param sql:
        :return:
        """
        async with self.mysql_pool.acquire() as coon:
            async with coon.cursor() as cursor:
                await cursor.execute(sql, params)
                await coon.commit()