J'ai un projet Flask qui interagit avec MySQL
db à Flask-SQLAlchemy
.
Ma question est de savoir comment sélectionner une ligne dans la base de données en fonction d'une valeur OR une autre valeur.
SELECT id FROM users WHERE email=email OR name=name;
Comment y parvenir dans Flask-SQLAlchemy
?
Les éléments suivants peuvent être utiles:
# app.py
from flask import Flask
from flask_sqlalchemy import SQLAlchemy
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'url_or_path/to/database'
db = SQLAlchemy(app)
class User(db.Model):
__tablename__ = 'users'
id = db.Column(db.Integer, primary_key=True)
email = db.Column(db.String(50), unique=True)
name = db.Column(db.String(30))
def __init__(self, name=None, email=None):
if not name:
raise ValueError('\'name\' cannot be None')
if not email:
raise ValueError('\'email\' cannot be None')
self.name = name
self.email = email
class UserQuery(object):
@staticmethod
def get_user_id_by_email_or_name(email=None, name=None):
user = User.query.filter((User.email == email) | (User.name == name)).first()
return user.id if hasattr(user, 'id') else None
Le '|' peut être utilisé à l'intérieur d'un filtre au lieu de 'ou_'. Voir tilisation de OR dans SQLAlchemy .
Vous pouvez utiliser comme ceci:
>>> from app import db, User, UserQuery
>>> db.create_all()
>>> user = User(name='stan', email='[email protected]')
>>> db.session.add(user)
>>> db.session.commit()
>>> by_name_id = UserQuery.get_user_id_by_email_or_name(name='stan')
>>> by_email_id = UserQuery.get_user_id_by_email_or_name(email='[email protected]')
>>> by_name_id == by_email_id
True
J'ai aussi besoin de cette affaire aujourd'hui, j'ai trouvé cette belle réponse ici :
Donc, nous pouvons faire [~ # ~] ou [~ # ~] logique comme exemple ci-dessous:
from sqlalchemy import or_
db.session.query(User).filter(or_(User.email=='[email protected]', User.name="username")).first()