From 09da2ee75e087054afd3683af130fda332f22924 Mon Sep 17 00:00:00 2001
From: stephanchrst <stephanchrst@gmail.com>
Date: Fri, 6 Jan 2023 19:16:15 +0700
Subject: add videos total in website content api

---
 indoteknik_api/controllers/api_v1/content.py |  8 ++------
 indoteknik_api/models/__init__.py            |  3 ++-
 indoteknik_api/models/website_content.py     | 17 +++++++++++++++++
 3 files changed, 21 insertions(+), 7 deletions(-)
 create mode 100644 indoteknik_api/models/website_content.py

diff --git a/indoteknik_api/controllers/api_v1/content.py b/indoteknik_api/controllers/api_v1/content.py
index 3d18ac13..93f0dcf4 100644
--- a/indoteknik_api/controllers/api_v1/content.py
+++ b/indoteknik_api/controllers/api_v1/content.py
@@ -18,11 +18,7 @@ class WebsiteContent(controller.Controller):
         data = []
         for video in videos:
             data.append({
-                'id': video.id,
-                'channel_name': video.channel_id.name,
-                'sequence': video.sequence,
-                'slide_type': video.slide_type,
-                'name': video.name,
-                'url': video.url
+                'video_total': request.env['website.content'].search_count(query),
+                'videos': [request.env['website.content'].api_single_response(x) for x in videos]
             })
         return self.response(data)
diff --git a/indoteknik_api/models/__init__.py b/indoteknik_api/models/__init__.py
index 3e00e2f0..4447c215 100644
--- a/indoteknik_api/models/__init__.py
+++ b/indoteknik_api/models/__init__.py
@@ -3,4 +3,5 @@ from . import product_pricelist
 from . import product_product
 from . import product_template
 from . import res_users
-from . import x_manufactures
\ No newline at end of file
+from . import x_manufactures
+from . import website_content
diff --git a/indoteknik_api/models/website_content.py b/indoteknik_api/models/website_content.py
new file mode 100644
index 00000000..d06ae56f
--- /dev/null
+++ b/indoteknik_api/models/website_content.py
@@ -0,0 +1,17 @@
+from odoo import models
+
+
+class WebsiteContent(models.Model):
+    _inherit = 'website.content'
+
+    def api_single_response(self, website_content):
+        data = {
+            'id': website_content.id,
+            'channel_name': website_content.channel_id.name,
+            'sequence': website_content.sequence,
+            'slide_type': website_content.slide_type,
+            'name': website_content.name,
+            'url': website_content.url
+        }
+
+        return data
-- 
cgit v1.2.3