在Python中,帮我用Python数据库操作写代码完成公司管理系统
时间: 2024-09-23 17:14:22 浏览: 59
在Python中,我们可以使用多种库来创建简单的公司管理系统,比如SQLite、MySQLdb(用于MySQL)或者psycopg2(用于PostgreSQL)。这里我会给出一个基本的例子,展示如何使用sqlite3库来管理公司的员工信息:
```python
import sqlite3
# 连接到SQLite数据库(如果不存在则会自动创建)
def create_connection(db_file):
conn = None;
try:
conn = sqlite3.connect(db_file)
print(sqlite3.version)
except Error as e:
print(e)
return conn
# 创建员工表
def create_table(conn):
cursor = conn.cursor()
cursor.execute('''CREATE TABLE IF NOT EXISTS employees
(id INTEGER PRIMARY KEY AUTOINCREMENT,
name TEXT NOT NULL,
department TEXT NOT NULL,
position TEXT)''')
conn.commit()
# 插入新员工
def insert_employee(conn, name, department, position):
cursor = conn.cursor()
cursor.execute("INSERT INTO employees VALUES (NULL,?,?,?)", (name, department, position))
conn.commit()
# 查询所有员工
def select_allEmployees(conn):
cursor = conn.cursor()
cursor.execute("SELECT * FROM employees")
rows = cursor.fetchall()
for row in rows:
print(row)
# 示例用法
def main():
database = r"company.db"
conn = create_connection(database)
if conn is not None:
create_table(conn)
insert_employee(conn, "张三", "人事部", "经理")
insert_employee(conn, "李四", "技术部", "工程师")
select_allEmployees(conn)
else:
print("无法连接到数据库")
if __name__ == '__main__':
main()
```
这个脚本创建了一个名为`employees`的表,并实现了添加员工和查询所有员工的功能。注意这只是一个基础示例,实际的公司管理系统可能需要处理更复杂的数据结构和业务逻辑。
阅读全文