Add debug representations for UserMembership and Organisation

betagouv · Apr 21, 2020 · 3ba6651 · 3ba6651
1 parent c3562f6
commit 3ba6651
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/repondeur/zam_repondeur/visam/models/membership.py b/repondeur/zam_repondeur/visam/models/membership.py
@@ -21,6 +21,8 @@ class UserMembership(Base):
 
     __tablename__ = "users_chambres"
 
+    __repr_keys__ = ("user", "chambre", "organisation")
+
     user_pk = Column(
         Integer,
         ForeignKey("users.pk", onupdate="CASCADE", ondelete="CASCADE"),

diff --git a/repondeur/zam_repondeur/visam/models/organisation.py b/repondeur/zam_repondeur/visam/models/organisation.py
@@ -8,6 +8,8 @@
 class Organisation(Base):
     __tablename__ = "organisations"
 
+    __repr_keys__ = ("name",)
+
     pk = Column(Integer, primary_key=True)
     name: str = Column(Text, nullable=False, unique=True)