如何使用Python进行药品库存的CRUD操作?
时间: 2024-09-26 16:08:25 浏览: 23
在Python中,我们可以使用像SQLite、MySQL这样的数据库系统来实现药品库存的CRUD操作(Create, Read, Update, Delete)。以下是一个简单的例子,我们将使用sqlite3库作为演示:
```python
import sqlite3
# 创建连接和创建数据库(如果不存在)
conn = sqlite3.connect('pharmacy.db')
cursor = conn.cursor()
# Create (插入)
def create_stock(item_name, quantity):
cursor.execute("INSERT INTO inventory (item_name, quantity) VALUES (?, ?)", (item_name, quantity))
conn.commit()
# Read (查询)
def get_stock(item_name):
cursor.execute("SELECT * FROM inventory WHERE item_name=?", (item_name,))
return cursor.fetchone()
# Update (修改)
def update_stock(item_name, new_quantity):
cursor.execute("UPDATE inventory SET quantity=? WHERE item_name=?", (new_quantity, item_name))
conn.commit()
# Delete (删除)
def delete_stock(item_name):
cursor.execute("DELETE FROM inventory WHERE item_name=?", (item_name,))
conn.commit()
# 关闭连接
conn.close()
```
上述代码中,我们首先建立到数据库的连接,然后定义了创建库存、获取库存、更新库存和删除库存的方法。每个方法对应SQL语句的不同部分。最后别忘了关闭连接以释放资源。