From b61cb49c1a26f572fb9837b6e7340b71d2404cab Mon Sep 17 00:00:00 2001
From: HibiKier <45528451+HibiKier@users.noreply.github.com>
Date: Thu, 9 Sep 2021 16:01:35 +0800
Subject: [PATCH] Update setu.py

---
 models/setu.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/models/setu.py b/models/setu.py
index 1decac7f..a6a7c0cd 100644
--- a/models/setu.py
+++ b/models/setu.py
@@ -91,7 +91,10 @@ class Setu(db.Model):
         说明：
             查询图片数量
         """
-        return len(await cls.query_image(r18=r18, limit=999999))
+        flag = False if r18 == 0 else True
+        setattr(Setu, 'count', db.func.count(cls.local_id).label('count'))
+        count = await cls.select('count').where(cls.is_r18 == flag).gino.first()
+        return count
 
     @classmethod
     async def get_image_in_hash(cls, img_hash: str) -> "Setu":