SQLAlchemy: Re sauver champ unique de modèle après avoir essayé de sauver la valeur non unique

Question

Dans mon application SQLAlchemy je le modèle suivant:

from sqlalchemy import Column, String 
from sqlalchemy.ext.declarative import declarative_base 
from sqlalchemy.orm import scoped_session, sessionmaker 
from zope.sqlalchemy import ZopeTransactionExtension 

DBSession = scoped_session(sessionmaker(extension=ZopeTransactionExtension())) 

class MyModel(declarative_base()): 
    # ... 
    label = Column(String(20), unique=True) 

    def save(self, force=False): 
     DBSession.add(self) 
     if force: 
      DBSession.flush() 

plus tard dans le code pour tous les nouveaux MyModel objets que je veux générer label au hasard et juste le régénérer si la valeur générée existe déjà dans DB.
Je suis en train de faire ce qui suit:

# my_model is an object of MyModel 
while True: 
    my_model.label = generate_label() 
    try: 
     my_model.save(force=True) 
    except IntegrityError: 
     # label is not unique - will do one more iteration 
     # (*) 
     pass 
    else: 
     # my_model saved successfully - exit the loop 
     break 

mais obtenir cette erreur dans le cas lors de la première généré label n'est pas unique et save() appelé au deuxième (ou version ultérieure) itération:

InvalidRequestError: This Session's transaction has been rolled back due to a previous exception during flush. To begin a new transaction with this Session, first issue Session.rollback(). Original exception was: (IntegrityError) column url_label is not unique... 

quand j'ajoute DBSession.rollback() dans la position (*) Je reçois ceci:

ResourceClosedError: The transaction is closed 

Que dois-je faire gérer cette situation correctement?
Merci

Answer 1

Si votre objet session est annulé, vous devez créer une nouvelle session et actualiser vos modèles avant de pouvoir recommencer. Et si vous utilisez zope.sqlalchemy vous devriez utiliser transaction.commit() et transaction.abort() pour contrôler les choses. Donc, votre boucle ressemblerait à quelque chose comme ceci:

# you'll also need this import after your zope.sqlalchemy import statement 
import transaction 

while True: 
    my_model.label = generate_label() 
    try: 
     transaction.commit() 
    except IntegrityError: 
     # need to use zope.sqlalchemy to clean things up 
     transaction.abort() 
     # recreate the session and re-add your object 
     session = DBSession() 
     session.add(my_model) 
    else: 
     break 

J'ai tiré l'utilisation de l'objet de la session sur la méthode save de l'objet ici. Je ne suis pas tout à fait sûr comment le ScopedSession se rafraîchit lorsqu'il est utilisé au niveau de la classe comme vous l'avez fait. Personnellement, je pense que l'intégration de SqlAlchemy choses à l'intérieur de vos modèles ne fonctionne pas vraiment bien avec l'approche unit of work de SqlAlchemy aux choses de toute façon.

Si votre objet étiquette est vraiment une valeur générée et unique, alors je serais d'accord avec TokenMacGuy et utiliser simplement une valeur uuid.

Espérons que ça aide.

Answer 2

Les bases de données ne sont pas une manière cohérente de vous dire pourquoi une transaction a échoué, sous une forme accessible à l'automatisation. Vous ne pouvez généralement pas essayer la transaction, puis réessayez car il a échoué pour une raison particulière.

Si vous connaissez une condition que vous voulez contourner (comme une contrainte unique), vous devez vérifier la contrainte vous-même. Dans sqlalchemy, cela va ressembler à quelque chose comme ceci:

# Find a unique label 
label = generate_label() 
while DBsession.query(
     sqlalchemy.exists(sqlalchemy.orm.Query(Model) 
        .filter(Model.lable == label) 
        .statement)).scalar(): 
    label = generate_label() 

# add that label to the model 
my_model.label = label 
DBSession.add(my_model) 
DBSession.flush() 

---

modifier: Une autre façon de répondre à cette question est que vous ne devriez pas réessayer automatiquement la transaction; Vous pouvez à la place renvoyer un code d'état HTTP de 307 Temporary Redirect (avec un peu de sel dans l'URL redirigée) afin que la transaction soit réellement lancée.

Answer 3

J'ai rencontré un problème similaire dans ma webapp écrite dans Pyramid. J'ai trouvé une solution un peu différente pour ce problème.

while True: 
    try: 
     my_model.label = generate_label() 
     DBSession.flush() 
     break 
    except IntegrityError: 
     # Rollback will recreate session: 
     DBSession.rollback() 
     # if my_model was in db it must be merged: 
     my_model = DBSession.merge(my_model) 

La partie de fusion est cruciale si le modèle my_model a été stocké auparavant. Sans fusionner la session serait vide donc flush ne prendrait aucune action.