pb连接mysql

import mysql.connector
import protobuf_module_pb2  # 假设protobuf_module为生成的PB模块

# 连接MySQL数据库
db = mysql.connector.connect(
    host="localhost",
    user="username",
    password="password",
    database="database_name"
)

cursor = db.cursor()

# 创建表（如果表不存在）
cursor.execute("""
CREATE TABLE IF NOT EXISTS users (
    id INT AUTO_INCREMENT PRIMARY KEY,
    name VARCHAR(255),
    email VARCHAR(255)
)
""")

# 创建PB消息对象
user = protobuf_module_pb2.User()
user.name = "John Doe"
user.email = "john.doe@example.com"

# 序列化PB消息对象
serialized_user = user.SerializeToString()

# 插入数据到MySQL表
cursor.execute("INSERT INTO users (name, email) VALUES (%s, %s)", (user.name, user.email))
db.commit()

# 查询数据并反序列化
cursor.execute("SELECT * FROM users")
result = cursor.fetchone()
if result:
    fetched_user = protobuf_module_pb2.User()
    fetched_user.ParseFromString(serialized_user)
    print(fetched_user.name, fetched_user.email)

# 关闭连接
cursor.close()
db.close()