Mysql
 sql >> Base de Dados >  >> RDS >> Mysql

Python conectar ao banco de dados MySQL com conector MySQL e exemplo de PyMySQL


Quando você deseja se conectar ao banco de dados MySQL em código Python, você pode usar o mysql-connector-python Driver MySQL e PyMySQL . O mysql-connector-python O driver MySQL é um driver interno do servidor MySQL e o PyMySQL é uma biblioteca de terceiros.

Todas as duas bibliotecas python MySQL fornecem classes para você obter conexão com o banco de dados MySQL, executar inserir, excluir, atualizar, selecionar o comando SQL. Eles também suportam o gerenciamento de transações. Este artigo mostrará um exemplo de como usar o mysql-connector-python e PyMySQL para operar na tabela do banco de dados MySQL.


1. Como usar o mysql-connector-python para operar o banco de dados MySQL.

1.1 Instale a biblioteca mysql-connector-python.

  1. O mysql-connector-python biblioteca pode ser instalada quando você instala o banco de dados MySQL.
  2. Então você pode executar o comando pip show mysql-connector-python para verificar se foi instalado ou não.
    $ pip show mysql-connector-python
WARNING: Package(s) not found: mysql-connector-python
  3. Se o mysql-connector-python biblioteca não está instalada, você pode executar o comando pip install mysql-connector-python para instalá-lo.
    $ pip install mysql-connector-python
Collecting mysql-connector-python
  Downloading mysql_connector_python-8.0.25-py2.py3-none-any.whl (319 kB)
     |████████████████████████████████| 319 kB 219 kB/s 
Collecting protobuf>=3.0.0
  Downloading protobuf-3.17.0-cp37-cp37m-macosx_10_9_x86_64.whl (959 kB)
     |████████████████████████████████| 959 kB 727 kB/s 
Requirement already satisfied: six>=1.9 in /Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages (from protobuf>=3.0.0->mysql-connector-python) (1.15.0)
Installing collected packages: protobuf, mysql-connector-python
Successfully installed mysql-connector-python-8.0.25 protobuf-3.17.0
  4. Agora, quando você executa o comando pip show mysql-connector-python novamente, você pode obter as informações de instalação.
    $ pip show mysql-connector-python
Name: mysql-connector-python
Version: 8.0.25
Summary: MySQL driver written in Python
Home-page: http://dev.mysql.com/doc/connector-python/en/index.html
Author: Oracle and/or its affiliates
Author-email: UNKNOWN
License: GNU GPLv2 (with FOSS License Exception)
Location: /Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages
Requires: protobuf
Required-by:

1.2 Use mysql-connector-python Etapas de conexão com o banco de dados MySQL.

  1. Importar o mysql.connector class.
    import mysql.connector
  2. Invoque o mysql.connector.connect() método para conectar ao servidor de banco de dados MySQL.
    # get mysql connection object.
def open_mysql_connection(user='jerry', password='jerry', host='127.0.0.1', port='3306', database='dev2qa_example', use_unicode=True):
    conn = mysql.connector.connect(user=user, password=password, host=host, port=port, database=database, use_unicode=use_unicode)
    return conn

1.3 Execute a instrução DDL para criar uma tabela.

  1. Obtenha o cursor do banco de dados MySQL.
    cursor = conn.cursor()
  2. Execute a instrução DDL. Se você deseja executar várias instruções DDL ao mesmo tempo, você precisa separar cada instrução DDL com um ponto e vírgula e adicionar o multi=True
    parâmetro para o execute() do objeto cursor , caso contrário, ele lançará o erro mysql.connector.errors.InterfaceError:Use multi=True ao executar várias instruções . 

    def execute_ddl(conn):
    cursor = conn.cursor()
    
    # The below DDL will first drop the table if exist, then create the table.
    ddl_sql= '''
    
       DROP TABLE user_account;
    
       CREATE TABLE `dev2qa_example`.`user_account` (
      `id` INT NOT NULL AUTO_INCREMENT,
      `user_name` VARCHAR(45) NULL,
      `password` VARCHAR(45) NULL,
      `email` VARCHAR(45) NULL,
      PRIMARY KEY (`id`));
       '''
    # Because the above DDL contains 2 statements ( drop, create), so need to add the multi=True parameter to avoid error mysql.connector.errors.InterfaceError: Use multi=True when executing multiple statements.
    cursor.execute(ddl_sql, multi=True)
    cursor.close()
    
    conn.close()
    
    print(ddl_sql + ' execute successfully.')

1.4 Execute a instrução DML (Inserir, Atualizar, Excluir).

  1. Obtenha o objeto cursor de conexão do banco de dados MySQL.
  2. Execute a instrução SQL insert, update, delete com o execute() do objeto cursor method.
    def execute_dml(conn):
    
    cursor = conn.cursor()
    
    dml_insert = ''' INSERT INTO `dev2qa_example`.`user_account` VALUES (null, %s, %s, %s); ''' 
    
    cursor.execute(dml_insert, ('tom', 'tom12345678', '[email protected]'))
    
    conn.commit()
    
    cursor.close()
    
    conn.close()
  3. Se você deseja inserir várias linhas na tabela MySQL, você pode executar o executemany() do objeto cursor method.
    def execute_dml_insert_many(conn):
    
    cursor = conn.cursor()
    
    dml_insert = ''' INSERT INTO `dev2qa_example`.`user_account` VALUES (null, %s, %s, %s); ''' 
    
    row_tuple = (('richard', 'richard', '[email protected]'), ('trump', 'trump', '[email protected]'),('doctor', 'doctor', '[email protected]'))
    
    cursor.executemany(dml_insert, row_tuple)
    
    conn.commit()
    
    cursor.close()
    
    conn.close()

1.5 Executar instrução SQL Select DML.

  1. Execute o execute() do objeto cursor método para executar a instrução SQL select.
    def execute_dml_select(conn):
    
    cursor = conn.cursor()
    
    dml_select = 'SELECT * FROM dev2qa_example.user_account;'
    
    cursor.execute(dml_select)
    
    # print the row header.
    for col in cursor.description:
        
        print(col[0], end='\t')
        
        
    print('\n-----------------------------') 
    
    # print each row data.
    for row in cursor:
        
        print(row)   
        
        print(row[1] + '--->' + row[2])
           
        
    cursor.close()
    
    conn.close()

2. Como usar o PtMySQL para operar o banco de dados MySQL.

2.1 Instale o PyMySQL.

  1. Antes de poder usar o PyMySQL, você deve instalá-lo primeiro. Você pode usar o pip para instalar como abaixo.
    $ pip3 install PyMySQL
  2. Você pode executar pip3 show comando para verificar a instalação do PyMySQL.
    $ pip3 show PyMySQL

2.2 Use o PyMySQL para conectar e operar o banco de dados MySQL.


Abaixo estão as etapas que você usa a biblioteca PyMySQL no código Python.
  1. Importar o PyMSQL conectar , cursores class.
    from pymysql import connect, cursors
  2. Ligar para conectar método para obter o objeto de conexão do banco de dados MySQL.
    conn = connect(host=host, user=user, password=password, db=db, charset=charset, cursorclass=cursorclass)
  3. Obtenha o objeto cursor do banco de dados pela função cursor() do objeto de conexão.
    conn.cursor() as cursor:
  4. Execute a instrução SQL usando o objeto cursor.
    cursor.execute(sql)
  5. Se a instrução sql for uma instrução select, chame o método fetchall() ou fetchone() do objeto cursor para obter o resultado da execução sql. Observe que após invocar fetchall(), o cursor se moverá para a última linha do resultado, portanto, se você chamar o método fetchall() novamente, não haverá nenhuma linha retornada.
    row = cursor.fetchall()

or

row = cursor.fetchone()
  6. Se a instrução sql for uma instrução de inserção, atualização, exclusão, chame a função commit() do objeto de conexão para confirmar as alterações no banco de dados MySQL.
    conn.commit()
  7. Não se esqueça de fechar o objeto de conexão do banco de dados MySQL no final para liberar os recursos do banco de dados.
    if conn is not None:
        conn.close()
        conn = None

2.3 Exemplo de banco de dados MySQL Operar PyMySQL.

  1. Neste exemplo, o nome do banco de dados MySQL é dev2qa_example , o nome da tabela é user_account . A tabela tem quatro colunas que são id , user_name , senha, e e-mail . A coluna id é incrementada automaticamente.
  2. O código-fonte python contém métodos para inserir, atualizar, excluir e selecionar dados de conta de usuário como abaixo.
    from pymysql import connect, cursors

global_host='127.0.0.1'
global_user='jerry'
global_password='jerry'
global_db='dev2qa_example'
global_charset='utf8'
global_cursorclass=cursors.DictCursor

# get mysql connection object.
def open_mysql_connection(host='127.0.0.1', user='jerry', password='jerry', db='dev2qa_example', charset='utf8', cursorclass=cursors.DictCursor):
    conn = connect(host=host, user=user, password=password, db=db, charset=charset, cursorclass=cursorclass)
    return conn
    
# close mysql connection.    
def close_mysql_connection(conn):
    if conn is not None:
        conn.close()
        conn = None

# execute insert, update or delete sql command.
def execute_insert_update_delete_sql(sql, host, user, password, db, charset, cursorclass):
    execute_row_count = 0
    conn = None
    print('sql = ' + sql)
    try:
        if string_is_not_empty(sql):
            conn = open_mysql_connection(host, user, password, db, charset, cursorclass)
            with conn.cursor() as cursor:
                execute_row_count = cursor.execute(sql)
    except Exception as ex:
        print(ex)
    finally:
        if conn is not None:
            conn.commit()
            close_mysql_connection(conn)
        return execute_row_count    
 
# execute select sql command.                    
def execute_select_sql(sql, host, user, password, db, charset, cursorclass):
    ret = list()
    conn = None
    print('sql = ' + sql)
    try:
        if string_is_not_empty(sql):
            conn = open_mysql_connection(host, user, password, db, charset, cursorclass)
            with conn.cursor() as cursor:
                row_count = cursor.execute(sql)
                print('select return row count = ' + str(row_count))
                row = cursor.fetchall()
                print(row)
                ret.append(row)
    except Exception as ex:
        print(ex)            
    finally:
        if conn is not None:
            close_mysql_connection(conn)
        return ret        

# insert user account to database table dev2qa_example.user_account.                
def insert_user_account(user_name, password, email):
    print('**********Begin insert_user_account.**********')
    
    if string_is_not_empty(user_name) and string_is_not_empty(password) and string_is_not_empty(email):
        # first check whether user account exist or not.
        sql = "select count(id) as count from dev2qa_example.user_account where lower(user_name) = '"+user_name.lower().strip()+"'"
        row_list = execute_select_sql(sql, global_host, global_user, global_password, global_db, global_charset, global_cursorclass)
        
        account_exist = False
        for row in row_list:
            for column in row:
                exist_count_number = column.get('count')
                if exist_count_number > 0:
                    account_exist = True
        
        if not account_exist:
            print('User ' + user_name + ' do not exist in database. Insert the user account to database table.')
            sql = "insert into user_account(user_name, password, email) values('"+user_name+"','"+password+"','"+email+"')"
            insert_row_count = execute_insert_update_delete_sql(sql, global_host, global_user, global_password, global_db, global_charset, global_cursorclass)
            print('Insert ' + str(insert_row_count) + ' row data successfully.')
        else:
            print('User account exist, can not insert.')    
    else:
        print('user_name, password, email can not be empty.')        
        
    print('**********End insert_user_account.**********')    
   
# update user account data.            
def update_user_account(user_name, password, email):
    print('**********Begin update_user_account.**********')
    
    if string_is_not_empty(user_name):
        sql = "update dev2qa_example.user_account set password = '" + password + "', email = '" + email + "' where lower(user_name) = '" + user_name.lower() + "'"
        update_row_count = execute_insert_update_delete_sql(sql, global_host, global_user, global_password, global_db, global_charset, global_cursorclass)
    
        if update_row_count == 0:
            print('User account do not exist, insert it now.')
            insert_user_account(user_name, password, email)
        else:
            print('Update ' + str(update_row_count) + ' row data successfully.')
    else:
        print('user_name can not be empty.')        
           
    print('**********End update_user_account.**********')
    
# delete user account data from database table.    
def delete_user_account(user_name):
    print('**********Begin delete_user_account.**********')
    
    if string_is_not_empty(user_name):
        sql = "delete from dev2qa_example.user_account where lower(user_name) = '"+user_name.lower()+"'"
        delete_row_count = execute_insert_update_delete_sql(sql, global_host, global_user, global_password, global_db, global_charset, global_cursorclass)
        print('Delete ' + str(delete_row_count) + ' row data successfully.')
        
    print('**********End delete_user_account.**********')
    
# execute select sql command to get user account data by user_name.    
def get_user_account_by_user_name(user_name):
    print('**********Begin get_user_account_by_user_name.**********')
    sql = "select * from dev2qa_example.user_account where lower(user_name) = '"+user_name.lower().strip()+"'"
    row_list = execute_select_sql(sql, global_host, global_user, global_password, global_db, global_charset, global_cursorclass)
    print('**********End get_user_account_by_user_name.**********')

# check whether the string is empty or not.
def string_is_not_empty(str):
    if str is None:
        return False
    elif len(str.strip()) == 0:
        return False
    else:
        return True

if __name__ == '__main__':
    # first delete user account jerry.
    delete_user_account('jerry')
    
    # then insert a user account jerry into database.
    insert_user_account('jerry', 'jerry', '[email protected]')
    # show user account data to verify the insert action.
    get_user_account_by_user_name('jerry')
    
    # update usr account information.
    update_user_account('jerry', 'jerry888', '[email protected]')
    # check the updated user account info again.
    get_user_account_by_user_name('jerry')
  3. Abaixo está o resultado da execução do código acima.
    **********Begin delete_user_account.**********
sql = delete from dev2qa_example.user_account where lower(user_name) = 'jerry'
Delete 1 row data successfully.
**********End delete_user_account.**********
**********Begin insert_user_account.**********
sql = select count(id) as count from dev2qa_example.user_account where lower(user_name) = 'jerry'
select return row count = 1
[{'count': 0}]
User jerry do not exist in database. Insert the user account to database table.
sql = insert into user_account(user_name, password, email) values('jerry','jerry','[email protected]')
Insert 1 row data successfully.
**********End insert_user_account.**********
**********Begin get_user_account_by_user_name.**********
sql = select * from dev2qa_example.user_account where lower(user_name) = 'jerry'
select return row count = 1
[{'id': 42, 'user_name': 'jerry', 'password': 'jerry', 'email': '[email protected]'}]
**********End get_user_account_by_user_name.**********
**********Begin update_user_account.**********
sql = update dev2qa_example.user_account set password = 'jerry888', email = '[email protected]' where lower(user_name) = 'jerry'
Update 1 row data successfully.
**********End update_user_account.**********
**********Begin get_user_account_by_user_name.**********
sql = select * from dev2qa_example.user_account where lower(user_name) = 'jerry'
select return row count = 1
[{'id': 42, 'user_name': 'jerry', 'password': 'jerry888', 'email': '[email protected]'}]
**********End get_user_account_by_user_name.**********

Referências
  1. Como usar o MySQL no Mac