79 lines
2.8 KiB
Python
79 lines
2.8 KiB
Python
import mysql.connector
|
|
from mysql.connector import errorcode
|
|
|
|
# Define database connection details
|
|
db_config = {
|
|
'user': 'neuer', # Replace with your MariaDB username
|
|
'password': 'S3raph1n!', # Replace with your MariaDB password
|
|
'host': 'mariadb', # The service name of your MariaDB container
|
|
}
|
|
|
|
# Initialize connection and cursor variables
|
|
conn = None
|
|
cursor = None
|
|
|
|
try:
|
|
# Connect to MariaDB server
|
|
conn = mysql.connector.connect(**db_config)
|
|
cursor = conn.cursor()
|
|
print("Connected to MariaDB server")
|
|
|
|
# Force the session to use utf8mb4_general_ci collation
|
|
cursor.execute("SET SESSION collation_connection = 'utf8mb4_general_ci'")
|
|
cursor.execute("SET NAMES 'utf8mb4' COLLATE 'utf8mb4_general_ci'")
|
|
print("Session collation set to utf8mb4_general_ci")
|
|
|
|
# Drop the existing database if it exists
|
|
cursor.execute("DROP DATABASE IF EXISTS test_db")
|
|
print("Existing 'test_db' database dropped.")
|
|
|
|
# Create a new database with utf8mb4_general_ci collation
|
|
cursor.execute("CREATE DATABASE test_db CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci")
|
|
print("Database 'test_db' created with utf8mb4_general_ci collation.")
|
|
|
|
# Select the newly created database
|
|
conn.database = 'test_db'
|
|
|
|
# Ensure all tables in this database use utf8mb4_general_ci collation
|
|
create_table_query = (
|
|
"CREATE TABLE employees ("
|
|
" id INT AUTO_INCREMENT PRIMARY KEY,"
|
|
" first_name VARCHAR(50) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci,"
|
|
" last_name VARCHAR(50) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci,"
|
|
" hire_date DATE"
|
|
") ENGINE=InnoDB CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci"
|
|
)
|
|
cursor.execute(create_table_query)
|
|
print("Table 'employees' created with utf8mb4_general_ci collation.")
|
|
|
|
# Insert a sample record
|
|
add_employee_query = (
|
|
"INSERT INTO employees (first_name, last_name, hire_date) "
|
|
"VALUES (%s, %s, %s)"
|
|
)
|
|
employee_data = ('John', 'Doe', '2024-08-14')
|
|
cursor.execute(add_employee_query, employee_data)
|
|
conn.commit()
|
|
print("Sample record inserted into 'employees' table.")
|
|
|
|
# Retrieve and display the inserted record
|
|
cursor.execute("SELECT * FROM employees")
|
|
rows = cursor.fetchall()
|
|
for row in rows:
|
|
print(f"ID: {row[0]}, First Name: {row[1]}, Last Name: {row[2]}, Hire Date: {row[3]}")
|
|
|
|
except mysql.connector.Error as err:
|
|
if err.errno == errorcode.ER_ACCESS_DENIED_ERROR:
|
|
print("Something is wrong with your user name or password")
|
|
elif err.errno == errorcode.ER_BAD_DB_ERROR:
|
|
print("Database does not exist")
|
|
else:
|
|
print(err)
|
|
finally:
|
|
# Close cursor and connection only if they were successfully created
|
|
if cursor:
|
|
cursor.close()
|
|
if conn:
|
|
conn.close()
|
|
print("Connection closed.")
|