Python MySQLdb - 類中的連接

Question

我正在制作一個 Python 項目，我必須在其中從數據庫中查找和檢索數據。
我嘗試創建一個類，在其中聲明連接並進行查詢，這是迄今為止我所擁有的。

import MySQLdb
dbc =("localhost","root","1234","users")
class sql:
    db = MySQLdb.connect(dbc[0],dbc[1],dbc[2],dbc[3])
    cursor = db.cursor()

    def query(self,sql):
        sql.cursor.execute(sql)
        return sql.cursor.fetchone()

    def rows(self):
        return sql.cursor.rowcount

sqlI = sql()
print(sqlI.query("SELECT `current_points` FROM `users` WHERE `nick` = 'username';"))

所以，主要問題是變量db和cursor不能從同一個類的其他定義/函數調用。 我想得到的是一個完善的查詢，我可以在其中進行查詢並檢索其內容。 這將總結我的代碼，因此我應該這樣做。

Answer 1

我通常使用psycopg2/postgres，但這是我經常使用的基本DB類，以Python的SQLite為例：

import sqlite3

class Database:
    def __init__(self, name):
        self._conn = sqlite3.connect(name)
        self._cursor = self._conn.cursor()

    def __enter__(self):
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.close()

    @property
    def connection(self):
        return self._conn

    @property
    def cursor(self):
        return self._cursor

    def commit(self):
        self.connection.commit()

    def close(self, commit=True):
        if commit:
            self.commit()
        self.connection.close()

    def execute(self, sql, params=None):
        self.cursor.execute(sql, params or ())

    def fetchall(self):
        return self.cursor.fetchall()

    def fetchone(self):
        return self.cursor.fetchone()

    def query(self, sql, params=None):
        self.cursor.execute(sql, params or ())
        return self.fetchall()

這將允許您使用Database類，通常像db = Database('db_file.sqlite)或在with語句中：

with Database('db_file.sqlite') as db:
    # do stuff

當with語句退出時，連接將自動提交和關閉。

然后，您可以將您經常使用的特定查詢封裝在方法中並使其易於訪問。 例如，如果您正在處理交易記錄，您可以使用一種方法來按日期獲取它們：

def transactions_by_date(self, date):
    sql = "SELECT * FROM transactions WHERE transaction_date = ?"
    return self.query(sql, (date,))

下面是一些示例代碼，我們在其中創建一個表，添加一些數據，然后將其讀回：

with Database('my_db.sqlite') as db:
    db.execute('CREATE TABLE comments(pkey INTEGER PRIMARY KEY AUTOINCREMENT, username VARCHAR, comment_body VARCHAR, date_posted TIMESTAMP)')
    db.execute('INSERT INTO comments (username, comment_body, date_posted) VALUES (?, ?, current_date)', ('tom', 'this is a comment'))
    comments = db.query('SELECT * FROM comments')
    print(comments)

我希望這會有所幫助！

Answer 2

這不是你在 Python 中編寫類的方式。 您需要在__init__方法中定義連接和游標，並通過self引用它們。

class sql:

    dbc = ("localhost","root","1234","users")

    def __init__(self):
        db = MySQLdb.connect(*self.dbc)
        self.cursor = db.cursor()

    def query(self,sql):
        self.cursor.execute(sql)
        return self.cursor.fetchone()

    def rows(self):
        return self.cursor.rowcount

Answer 3

您可以使用構造函數進行連接。 當類的對象將被創建時，構造函數將自動調用。

import MySQLdb

class Connection:
    def __init__(self):
        self.con=MySQLdb.connect("127.0.0.1","root","","db_name",3306)
        self.cmd=self.con.cursor()
obj=Connection()

Answer 4

from config import Config
import MySQLdb

class Connection:
    def __init__(self):
        self.db=MySQLdb.connect(
            Config.DATABASE_CONFIG['server'],
            Config.DATABASE_CONFIG['user'],
            Config.DATABASE_CONFIG['password'],
            Config.DATABASE_CONFIG['name']
            )
        self.db.autocommit(True)
        self.db.set_character_set('utf8mb4')
        self.cur=self.db.cursor()

示例配置類（config.py）：

class Config(object):

  DATABASE_CONFIG = {
          'server': 'localhost',
          'user': 'dbuser',
          'password': 'password',
          'name': 'dbname',
          }