[youtube-data-api] feat: add support for sorts

Author: AndreyMarkinPPC

libs/community/google/youtube/youtube-data-api/garf_youtube_data_api/report_fetcher.py

@@ -20,6 +20,7 @@
 from collections.abc import Iterable, MutableSequence
 from typing import Any, Final
 
+import pandas as pd
 from garf_core import parsers, report, report_fetcher
 from typing_extensions import override
 
@@ -96,4 +97,16 @@ def fetch(
           query_specification, args, **{name: element}, **kwargs
         )
         results.append(res)
-    return functools.reduce(operator.add, results)
+    res = functools.reduce(operator.add, results)
+    if sorts := (
+      query_editor.YouTubeDataApiQuery(text=query_specification)
+      .generate()
+      .sorts
+    ):
+      sorts = sorts[0]
+      key, *desc = sorts.split(' ')
+      asc_order = not desc or 'ASC' in desc[0]
+      return report.GarfReport.from_pandas(
+        res.to_pandas().sort_values(by=key, ascending=asc_order)
+      )
+    return res

libs/community/google/youtube/youtube-data-api/pyproject.toml

@@ -5,11 +5,12 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "garf-youtube-data-api"
 dependencies = [
-  "garf-core",
+  "garf-core[pandas]",
   "garf-io",
   "google-api-python-client",
 ]
 authors = [
+  {name = "Andrei Markin", email = "amarkin@google.com"},
   {name = "Google Inc. (gTech gPS CSE team)", email = "no-reply@google.com"},
 ]
 license = {text = "Apache 2.0"}

libs/community/google/youtube/youtube-data-api/tests/unit/test_report_fetcher.py

@@ -22,7 +22,7 @@ class TestYouTubeDataApiReportFetcher:
   def fetcher(self):
     return YouTubeDataApiReportFetcher()
 
-  def test_fetch(self, mocker, fetcher):
+  def test_fetch_returns_filtered_data(self, mocker, fetcher):
     query = """
       SELECT
         id,
@@ -61,3 +61,43 @@ def test_fetch(self, mocker, fetcher):
     )
 
     assert result == expected_report
+
+  def test_fetch_returns_sorted_data(self, mocker, fetcher):
+    query = """
+      SELECT
+        id,
+        statistics.viewCount AS views,
+        statistics.likeCount AS likes,
+        snippet.publishedAt AS published_at,
+      FROM videos
+      ORDER BY likes DESC
+    """
+
+    mocker.patch(
+      'garf_youtube_data_api.api_clients.YouTubeDataApiClient._list',
+      return_value={
+        'items': [
+          {
+            'id': 1,
+            'statistics': {'viewCount': 10, 'likeCount': 1},
+            'snippet': {'publishedAt': '2024-07-10T22:15:44Z'},
+          },
+          {
+            'id': 2,
+            'statistics': {'viewCount': 11, 'likeCount': 2},
+            'snippet': {'publishedAt': '2025-07-10T22:15:44Z'},
+          },
+        ],
+      },
+    )
+
+    result = fetcher.fetch(query, id=['1', '2'])
+    expected_report = garf_core.GarfReport(
+      results=[
+        [2, 11, 2, '2025-07-10T22:15:44Z'],
+        [1, 10, 1, '2024-07-10T22:15:44Z'],
+      ],
+      column_names=['id', 'views', 'likes', 'published_at'],
+    )
+
+    assert result == expected_report