From 40960fc7fd46deaf91bdcbff50c4576367efee66 Mon Sep 17 00:00:00 2001
From: codl <codl@codl.fr>
Date: Wed, 27 Dec 2017 22:16:13 +0100
Subject: [PATCH] take direct messages into account in estimate

---
 model.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/model.py b/model.py
index d6aa960..003dc23 100644
--- a/model.py
+++ b/model.py
@@ -177,6 +177,8 @@ class Account(TimestampMixin, RemoteIDMixin):
             query = query.filter(db.or_(Post.favourite == False, Post.is_reblog))
         if(self.policy_keep_media):
             query = query.filter(db.or_(Post.has_media == False, Post.is_reblog))
+        if(self.policy_keep_direct):
+            query = query.filter(~Post.direct)
         return query.count()
 
     def force_log_out(self):