From e36f64813e6f12d541d69bdb1619dbf29b2d8c92 Mon Sep 17 00:00:00 2001
From: Cédric Bonhomme <cedric@cedricbonhomme.org>
Date: Tue, 4 Oct 2016 11:39:25 +0200
Subject: Fixed a bug when removing a user who have feeds in categories.

---
 src/web/models/category.py | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'src/web/models/category.py')

diff --git a/src/web/models/category.py b/src/web/models/category.py
index c35db52e..15b616bf 100644
--- a/src/web/models/category.py
+++ b/src/web/models/category.py
@@ -7,8 +7,13 @@ class Category(db.Model, RightMixin):
     id = db.Column(db.Integer(), primary_key=True)
     name = db.Column(db.String())
 
+    # relationships
     user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
+    feeds = db.relationship('Feed', cascade='all,delete-orphan')
+    articles = db.relationship('Article', 
+                            cascade='all,delete-orphan')
 
+    # index
     idx_category_uid = Index('user_id')
 
     # api whitelists
-- 
cgit