MySQL と Python を使用してシンプルなオンライン投票システムを開発する方法

CREATE DATABASE voting_system;

USE voting_system;

CREATE TABLE candidates (
    id INT AUTO_INCREMENT PRIMARY KEY,
    name VARCHAR(255) NOT NULL,
    votes INT DEFAULT 0
);

pip install mysql-connector-python

import mysql.connector
from mysql.connector import Error

def create_connection():
    conn = None;
    try:
        conn = mysql.connector.connect(
            host='localhost',
            database='voting_system',
            user='root',
            password='your_password'
        )
        if conn.is_connected():
            print('连接到数据库成功！')
            return conn
    except Error as e:
        print(e)

    return conn

def add_candidate(conn):
    name = input('请输入要添加的候选人姓名：')
    try:
        cursor = conn.cursor()
        cursor.execute('INSERT INTO candidates(name) VALUES(%s)', (name,))
        conn.commit()
        print('候选人添加成功！')
    except Error as e:
        print(e)

def vote(conn):
    candidate_id = input('请输入要投票的候选人ID：')
    try:
        cursor = conn.cursor()
        cursor.execute('SELECT votes FROM candidates WHERE id = %s', (candidate_id,))
        result = cursor.fetchone()
        if result:
            votes = result[0]
            votes += 1
            cursor.execute('UPDATE candidates SET votes = %s WHERE id = %s', (votes, candidate_id))
            conn.commit()
            print('投票成功！')
        else:
            print('候选人不存在！')
    except Error as e:
        print(e)

def main():
    conn = create_connection()
    while True:
        print('1. 添加候选人')
        print('2. 进行投票')
        print('3. 退出')
        choice = input('请选择操作：')
        if choice == '1':
            add_candidate(conn)
        elif choice == '2':
            vote(conn)
        elif choice == '3':
            break
        else:
            print('无效的选择！')

    conn.close()