Add admin for clinical specialty and make all profile relations casca…

…de delete
razzius · Oct 9, 2018 · f1e34a6 · f1e34a6
1 parent 6ee79a7
commit f1e34a6
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/server/__init__.py b/server/__init__.py
@@ -9,12 +9,13 @@
 
 from .models import (
     ActivityOption,
+    ClinicalSpecialtyOption,
+    PartsOfMeOption,
+    ProfessionalInterestOption,
     Profile,
     VerificationEmail,
     VerificationToken,
     db,
-    ProfessionalInterestOption,
-    PartsOfMeOption,
 )
 import server.views
 
@@ -80,8 +81,7 @@ def delete_view(self):
 admin.add_view(BasicAuthModelView(VerificationToken, db.session))
 admin.add_view(BasicAuthModelView(VerificationEmail, db.session))
 admin.add_view(BasicAuthModelView(Profile, db.session))
-# admin.add_view(BasicAuthModelView(HospitalAffiliationOption, db.session))
-# admin.add_view(BasicAuthModelView(ClinicalSpecialtyOption, db.session))
+admin.add_view(BasicAuthModelView(ClinicalSpecialtyOption, db.session))
 admin.add_view(BasicAuthModelView(ProfessionalInterestOption, db.session))
 admin.add_view(BasicAuthModelView(PartsOfMeOption, db.session))
 admin.add_view(BasicAuthModelView(ActivityOption, db.session))

diff --git a/server/models.py b/server/models.py
@@ -2,7 +2,7 @@
 from datetime import datetime
 
 from flask_sqlalchemy import SQLAlchemy
-from sqlalchemy.orm import relationship
+from sqlalchemy.orm import relationship, backref
 
 
 db = SQLAlchemy()
@@ -117,11 +117,11 @@ class Profile(db.Model):
 
     profile_image_url = db.Column(db.String(255))
 
-    clinical_specialties = relationship(ClinicalSpecialty)
-    affiliations = relationship(HospitalAffiliation)
-    professional_interests = relationship(ProfessionalInterest)
-    parts_of_me = relationship(PartsOfMe)
-    activities = relationship(ProfileActivity)
+    clinical_specialties = relationship(ClinicalSpecialty, cascade='all, delete')
+    affiliations = relationship(HospitalAffiliation, cascade='all, delete')
+    professional_interests = relationship(ProfessionalInterest, cascade='all, delete')
+    parts_of_me = relationship(PartsOfMe, cascade='all, delete')
+    activities = relationship(ProfileActivity, cascade='all, delete')
 
     date_created = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
     date_updated = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)