Python3 數據庫操作基本類

參考地址: https://dev.mysql.com/doc/connector-python/en/

安     裝: pip3  install  mysql-connector 

# -*- coding: utf-8 -*-

'''
Copyright (c) 2018, mersap
All rights reserved.

摘    要: base mysql api
創 建 者: mersap
創建日期: 2018-10-11
'''
import mysql.connector
class MyDB(object):
    """docstring for MyDB"""
    def __init__(self, host, user, passwd , db):
        self.host = host
        self.user = user
        self.passwd = passwd
        self.db = db

        self.connect = None
        self.cursor = None
    def db_connect(self):
        """數據庫連接
        """
        self.connect = mysql.connector.connect(host=self.host, user=self.user, passwd=self.passwd, database=self.db)
        return self
    def db_cursor(self):
        if self.connect is None:
            self.connect = self.db_connect()

        if not self.connect.is_connected():
            self.connect = self.db_connect()
        self.cursor = self.connect.cursor()
        return self
    def get_rows(self , sql):
        """ 查詢數據庫結果
        :param sql: SQL語句
        :param cursor: 數據庫遊標
        """

        self.cursor.execute(sql)
        return self.cursor.fetchall()
    def db_execute(self, sql):
        self.cursor.execute(sql)
        self.connect.commit()
    def db_close(self):
        """關閉數據庫連接和遊標
        :param connect: 數據庫連接實例
        :param cursor: 數據庫遊標
        """
        if self.connect:
            self.connect.close()
        if self.cursor:
            self.cursor.close()