Add ability to return package_infos with a given timestamp

Author: javihernandez

alws/crud/package_info.py

@@ -1,3 +1,4 @@
+import datetime
 from collections import namedtuple
 from typing import Optional
 
@@ -15,14 +16,15 @@ async def get_package_info(
     package_name: str,
     platform_name: str,
     arch: Optional[str] = None,
+    updated_after: Optional[str] = None,
 ):
     platform_id_query = select(Platform.id).where(
         Platform.name == platform_name
     )
     platform_id = (await bs_db.execute(platform_id_query)).scalar()
 
     if not platform_id:
-        raise PlatformNotFoundError(f'Invalid distribution: {platform_name}')
+        raise PlatformNotFoundError(f"Invalid distribution: {platform_name}")
 
     conditions = [
         Repository.platform_id == platform_id,
@@ -40,10 +42,18 @@ async def get_package_info(
 
     repo_ids = [repo.pulp_href.split('/')[-2] for repo in repositories]
 
-    subq = select(CoreRepositoryContent.content_id).where(
+    subq_conditions = [
         CoreRepositoryContent.repository_id.in_(repo_ids),
         CoreRepositoryContent.version_removed_id.is_(None),
-    )
+    ]
+    if updated_after:
+        last_updated = datetime.datetime.strptime(
+            updated_after, "%Y-%m-%d %H:%M:%S"
+        )
+        subq_conditions.append(
+            CoreRepositoryContent.pulp_last_updated >= last_updated
+        )
+    subq = select(CoreRepositoryContent.content_id).where(*subq_conditions)
 
     query = select(
         RpmPackage.name,
@@ -57,12 +67,12 @@ async def get_package_info(
     pulp_packages = (await pulp_db.execute(query)).all()
 
     PackageTuple = namedtuple(
-        'PackageTuple',
+        "PackageTuple",
         [
-            'name',
-            'version',
-            'release',
-            'changelogs',
+            "name",
+            "version",
+            "release",
+            "changelogs",
         ],
     )
     packages = []

alws/routers/package_info.py

@@ -25,6 +25,7 @@ async def get_package_info(
     name: str,
     almalinux_version: int,
     arch: Optional[str] = None,
+    updated_after: Optional[str] = None,
     bs_db: AsyncSession = Depends(
         AsyncSessionDependency(key=get_async_db_key())
     ),
@@ -41,6 +42,7 @@ async def get_package_info(
             name,
             platform_name,
             arch,
+            updated_after,
         )
         return packages
     except (PlatformNotFoundError, RepositoriesNotFoundError) as exc:

tests/fixtures/package_info.py

@@ -1,3 +1,4 @@
+import datetime
 from typing import List, Dict, Any
 
 import pytest
@@ -49,4 +50,39 @@ async def mock_func(*args, **kwargs):
         return []
     monkeypatch.setattr("alws.crud.package_info.get_package_info", mock_func)
 
+@pytest.fixture
+def mock_get_package_info_with_date_filter(monkeypatch):
+    async def mock_func(*args, **kwargs):
+        updated_after = args[-1]
+        packages = [
+            {
+                "name": "example_package",
+                "version": "1.0",
+                "release": "1.el8",
+                "changelogs": ["Initial release"],
+                "pulp_last_updated": "2024-01-01 10:00:00",
+            },
+            {
+                "name": "example_package",
+                "version": "1.1",
+                "release": "2.el9",
+                "changelogs": ["Security patch"],
+                "pulp_last_updated": "2024-06-01 12:00:00",
+            },
+        ]
+
+        if updated_after:
+            cutoff = datetime.datetime.strptime(updated_after, "%Y-%m-%d %H:%M:%S")
+            filtered = [
+                {
+                    k: v for k, v in pkg.items() if k != "pulp_last_updated"
+                }
+                for pkg in packages
+                if datetime.datetime.strptime(pkg["pulp_last_updated"], "%Y-%m-%d %H:%M:%S") >= cutoff
+            ]
+        else:
+            filtered = [ {k: v for k, v in pkg.items() if k != "pulp_last_updated"} for pkg in packages ]
 
+        return filtered
+
+    monkeypatch.setattr("alws.crud.package_info.get_package_info", mock_func)

tests/test_api/test_package_info.py

@@ -55,3 +55,19 @@ async def test_missing_required_query_params(self):
         )
 
         assert response.status_code == status.HTTP_422_UNPROCESSABLE_ENTITY
+
+    async def test_get_package_info_with_updated_after(
+        self, mock_get_package_info_with_date_filter
+    ):
+        updated_after = "updated_after=2024-05-01 00:00:00"
+        q_params = f"?name=example_package&almalinux_version=9&{updated_after}"
+        response = await self.make_request(
+            "get",
+            f"/api/v1/package_info/{q_params}",
+        )
+
+        assert response.status_code == status.HTTP_200_OK
+        data = response.json()
+        assert len(data) == 1
+        assert data[0]["version"] == "1.1"
+        assert data[0]["release"] == "2.el9"