python判断传入数据的主键在数据库中是否已存在
时间: 2023-05-16 14:06:04 浏览: 147
可以使用Python的SQLAlchemy库来判断传入数据的主键是否已存在于数据库中。具体实现方式可以参考以下代码:
```python
from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base
# 创建数据库连接
engine = create_engine('mysql+pymysql://username:password@localhost:3306/database_name')
Base = declarative_base()
# 定义数据模型
class User(Base):
__tablename__ = 'users'
id = Column(Integer, primary_key=True)
name = Column(String(50))
age = Column(Integer)
# 创建数据表
Base.metadata.create_all(engine)
# 创建Session
Session = sessionmaker(bind=engine)
session = Session()
# 判断主键是否已存在
def is_primary_key_exist(id):
user = session.query(User).filter_by(id=id).first()
if user:
return True
else:
return False
```
在上述代码中,我们首先创建了一个数据库连接,并定义了一个数据模型User,然后创建了数据表。接着,我们创建了一个Session,并定义了一个函数is_primary_key_exist来判断主键是否已存在。在函数中,我们使用session.query(User).filter_by(id=id).first()来查询数据库中是否已存在该主键对应的记录,如果存在则返回True,否则返回False。
阅读全文