diff --git a/models/database.py b/models/database.py index 1c770da..23640a7 100644 --- a/models/database.py +++ b/models/database.py @@ -198,6 +198,14 @@ class Database: cursor.execute('DELETE FROM articles WHERE id = ?', (article_id,)) conn.commit() return cursor.rowcount > 0 + + def get_articles_count(self): + """获取文章总数""" + with self.get_connection() as conn: + cursor = conn.cursor() + cursor.execute('SELECT COUNT(*) as count FROM articles') + row = cursor.fetchone() + return row['count'] if row else 0 # ========== 待处理产品操作 ========== def add_pending_product(self, product_name, category=None, subcategory=None, priority=0, source='manual'): diff --git a/routes/articles.py b/routes/articles.py index a135d92..6a708ee 100644 --- a/routes/articles.py +++ b/routes/articles.py @@ -14,6 +14,7 @@ def list_articles(): offset = request.args.get('offset', 0, type=int) articles = db.get_all_articles(limit=limit, offset=offset) + total_count = db.get_articles_count() # 获取总数 # 解析JSON字段 for article in articles: @@ -23,7 +24,7 @@ def list_articles(): return jsonify({ 'success': True, 'articles': articles, - 'count': len(articles) + 'count': total_count # 返回总数 }) @bp.route('/search', methods=['GET'])