SQLAlchemy, effacer de la base de données de contenu, mais ne laissez pas tomber le schéma

Je suis l'élaboration d'un Pylônes application qui est basée sur l'existant de la base de données, donc je suis en utilisant la réflexion. J'ai un fichier SQL avec le schéma que j'ai utilisé pour créer ma base de données de test. C'est pourquoi je ne peux pas tout simplement utiliser drop_all et create_all.

Je voudrais écrire des tests unitaires et j'ai affronté le problème de la compensation de la base de données de contenu après chaque test. Je veux juste effacer toutes les données, mais laissez les tables intacte. Est-ce possible?

L'application utilise Postgres et c'est ce qui doit être également utilisé pour les tests.

InformationsquelleAutor Juliusz Gonera | 2011-01-21
  1. 47

    J'ai demandé la même chose sur la SQLAlchemy groupe Google, et j'ai trouvé une recette qui semble bien fonctionner (tous mes tableaux sont vidées). Voir le fil de référence.

    Mon code (extrait) ressemble à ceci:

    import contextlib
from sqlalchemy import MetaData

meta = MetaData()

with contextlib.closing(engine.connect()) as con:
    trans = con.begin()
    for table in reversed(meta.sorted_tables):
        con.execute(table.delete())
    trans.commit()

    Edit: j'ai modifié le code pour supprimer des tables dans l'ordre inverse; soi-disant, cela devrait s'assurer que les enfants sont supprimés avant que les parents.

    • Il semble que cela fonctionne! Je vous remercie. Le code exact-je l'utiliser dans des Pylônes est: for table in reversed(meta.Base.metadata.sorted_tables): meta.Session.execute(table.delete()); meta.Session.commit()
    • D'où vient engine viennent? Et contextlib? Est-ce que la bibliothèque standard? Cette réponse serait mieux si les importations étaient complets.
    • Je suppose que vos edit augmente le risque d'un lecteur de confondre cette réponse, au lieu de la diminuer. Pour un lecteur qui a déjà fait de la configuration initiale d'un SQLAlchemy application, je pense que c'était déjà implicitement clair que meta était, et donc le modifier a pas d'effet sur leur compréhension. Mais pour quelqu'un qui n'a jamais vu MetaData() avant et il est juste de tâtonner dans une application de quelqu'un d'autre a créé, je crains que la réponse maintenant il fait regarder comme ils sont censés créer un nouvelles MetaData() l'instance à utiliser avec ce code, qui bien sûr ne fonctionne pas.
    • La partie de métadonnées comme il semble que la réponse n'a pas fonctionné. Tout d'abord, le méta-objet créé n'est pas utilisé n'importe où et je n'avais aucune idée de comment le faire fonctionner. Cependant, comme une solution de contournement, en remplacement de la "par" boucle avec for table in engine.table_names(): con.execute("DROP table " + table)
    InformationsquelleAutor aknuds1
  2. 12

    Pour PostgreSQL à l'aide de TRUNCATE:

    with contextlib.closing(engine.connect()) as con:
    trans = con.begin()
    con.execute('TRUNCATE {} RESTART IDENTITY;'.format(
        ','.join(table.name 
                 for table in reversed(Base.metadata.sorted_tables))))
    trans.commit()

    Remarque: RESTART IDENTITY; s'assure que toutes les séquences sont réinitialisés. Cependant, c'est plus lent que le DELETE recette par @aknuds1 de 50%.

    Une autre recette consiste à déposer tous les tableaux premier, puis les recréer. C'est plus lent par un autre 50%:

    Base.metadata.drop_all(bind=engine)
Base.metadata.create_all(bind=engine)
    • metadata.Base.metadata.drop_all(bind=engine) donne l'erreur AttributeError: 'MetaData' object has no attribute 'Base'
    InformationsquelleAutor kolypto
  3. -2

    L'utiliser truncate:

    TRONQUER [ TABLE ] nom [, ...]

    (http://www.postgresql.org/docs/8.4/static/sql-truncate.html)

    Cela va supprimer tous les enregistrements dans la table, mais laissez le schéma de tact.

    • Ce serait OK, mais comment puis-je l'utiliser dans SQLAlchemy? for table in meta.Base.metadata.tables.keys(): meta.Session.execute('truncate %s' % table); meta.Session.commit() provoque un tel message d'erreur: "InternalError: (InternalError) transaction en cours est annulée, les commandes ignoré jusqu'à la fin de l'opération en bloc" tronquer les données.sous-mots-clés' {}"
    • Je l'ai corrigé en faisant truncate %s cascade mais c'est extrêmement lent. Trop lent à exécuter après chaque test unitaire...
    • La question de sqlalchemy, pas sql.
    InformationsquelleAutor Joe L.