使用Python實現資料庫連接與操作

一、連接資料庫

在使用Python操作資料庫之前，我們需要先連接資料庫。Python中很多第三方庫可以支持資料庫連接，如PyMySQL、psycopg2等。

以PyMySQL為例，在使用之前需要先安裝PyMySQL庫。

    
        pip install PyMySQL
    

連接資料庫需要提供資料庫的地址、用戶名、密碼、資料庫名等信息，具體代碼如下：

    
        import pymysql

        db = pymysql.connect(
            host='localhost',
            user='root',
            password='password',
            database='database_name'
        )
    

上述代碼中，db就是所連接的資料庫

二、創建表

在連接資料庫之後，我們可以通過Python代碼創建表。創建表需要使用SQL語句，使用Python的MySQLCursor.execute()方法執行SQL語句即可。

    
        import pymysql
    
        db = pymysql.connect(
            host='localhost',
            user='root',
            password='password',
            database='database_name'
        )
    
        cursor = db.cursor()

        # SQL語句
        sql = '''
            CREATE TABLE `users` (
                `id` int(11) NOT NULL AUTO_INCREMENT,
                `name` varchar(255) COLLATE utf8mb4_unicode_ci DEFAULT NULL,
                `age` int(11) DEFAULT NULL,
                `email` varchar(255) COLLATE utf8mb4_unicode_ci DEFAULT NULL,
                PRIMARY KEY (`id`)
            ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci
        '''

        # 執行SQL語句
        cursor.execute(sql)

        # 提交到資料庫
        db.commit()

        # 關閉游標和資料庫連接
        cursor.close()
        db.close()
    

上述代碼中，我們使用了CREATE TABLE語句創建了一個名為users的表，表中包含了id、name、age、email四個欄位。

三、插入數據

在創建了表之後，我們可以使用Python代碼向表中插入數據。插入數據同樣需要使用SQL語句，Python的MySQLCursor.execute()方法同樣可以執行插入數據的SQL語句。

    
        import pymysql
    
        db = pymysql.connect(
            host='localhost',
            user='root',
            password='password',
            database='database_name'
        )
    
        cursor = db.cursor()

        # SQL語句
        sql = "INSERT INTO `users`(`name`, `age`, `email`) VALUES ('Tom', 20, 'tom@example.com')"

        # 執行SQL語句
        cursor.execute(sql)
        
        # 提交到資料庫
        db.commit()

        # 關閉游標和資料庫連接
        cursor.close()
        db.close()
    

上述代碼中，我們使用INSERT INTO語句向users表中插入了一條數據。

四、查詢數據

在向表中插入了數據之後，我們可以使用Python代碼查詢數據。查詢數據同樣需要使用SQL語句，Python的MySQLCursor.execute()方法同樣可以執行查詢數據的SQL語句。查詢數據之後需要使用MySQLCursor.fetchall()方法獲取查詢結果，並且必須在查詢完畢之後關閉游標。

    
        import pymysql

        db = pymysql.connect(
            host='localhost',
            user='root',
            password='password',
            database='database_name'
        )

        cursor = db.cursor()

        # SQL語句
        sql = "SELECT * FROM `users`"

        # 執行SQL語句
        cursor.execute(sql)

        # 獲取查詢結果
        results = cursor.fetchall()

        for row in results:
            id = row[0]
            name = row[1]
            age = row[2]
            email = row[3]

            print(f"id={id}, name={name}, age={age}, email={email}")

        # 關閉游標和資料庫連接
        cursor.close()
        db.close()
    

上述代碼中，我們使用SELECT語句查詢了users表中的所有數據，並將查詢結果列印輸出。

五、更新數據

在查詢到數據之後，我們也可以使用Python代碼更新數據。更新數據同樣需要使用SQL語句，並且需要使用Python的MySQLCursor.execute()方法執行更新數據的SQL語句。

    
        import pymysql

        db = pymysql.connect(
            host='localhost',
            user='root',
            password='password',
            database='database_name'
        )

        cursor = db.cursor()

        # SQL語句
        sql = "UPDATE `users` SET `email`='tom_new@example.com' WHERE `name`='Tom'"

        # 執行SQL語句
        cursor.execute(sql)

        # 提交到資料庫
        db.commit()

        # 關閉游標和資料庫連接
        cursor.close()
        db.close()
    

上述代碼中，我們使用UPDATE語句將users表中name為Tom的記錄的email欄位更新為tom_new@example.com。

六、刪除數據

在刪除數據之前，我們需要知道需要刪除哪些數據，並且需要使用SQL語句執行刪除數據的操作。Python的MySQLCursor.execute()方法同樣可以執行刪除數據的SQL語句。

    
        import pymysql

        db = pymysql.connect(
            host='localhost',
            user='root',
            password='password',
            database='database_name'
        )

        cursor = db.cursor()

        # SQL語句
        sql = "DELETE FROM `users` WHERE `name`='Tom'"

        # 執行SQL語句
        cursor.execute(sql)

        # 提交到資料庫
        db.commit()

        # 關閉游標和資料庫連接
        cursor.close()
        db.close()
    

上述代碼中，我們使用DELETE語句將users表中name為Tom的記錄刪除。