企业HR助手搭建

import sqlite3

# 连接到SQLite数据库
conn = sqlite3.connect('hr_database.db')
cursor = conn.cursor()

# 创建员工表
cursor.execute('''
CREATE TABLE IF NOT EXISTS employees (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    name TEXT NOT NULL,
    position TEXT NOT NULL,
    department TEXT NOT NULL
)
''')

# 添加员工信息
def add_employee(name, position, department):
    cursor.execute('''
    INSERT INTO employees (name, position, department) VALUES (?, ?, ?)
    ''', (name, position, department))
    conn.commit()

# 查询所有员工信息
def get_all_employees():
    cursor.execute('SELECT * FROM employees')
    return cursor.fetchall()

# 更新员工信息
def update_employee(employee_id, name, position, department):
    cursor.execute('''
    UPDATE employees SET name = ?, position = ?, department = ? WHERE id = ?
    ''', (name, position, department, employee_id))
    conn.commit()

# 删除员工信息
def delete_employee(employee_id):
    cursor.execute('DELETE FROM employees WHERE id = ?', (employee_id,))
    conn.commit()

# 示例调用
add_employee('张三', '工程师', '研发部')
print(get_all_employees())
update_employee(1, '李四', '高级工程师', '研发部')
print(get_all_employees())
delete_employee(1)
print(get_all_employees())

# 关闭数据库连接
conn.close()