Skip to content

Commit 0461193

Browse files
committed
Include data.versions in platforms endpoint return payload
Resolves: AlmaLinux/build-system#484
1 parent b7a9871 commit 0461193

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

alws/schemas/platform_schema.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import typing
22

3-
from pydantic import BaseModel
3+
from pydantic import BaseModel, root_validator
44

55
from alws.schemas.repository_schema import RepositoryCreate
66

@@ -54,6 +54,16 @@ class Platform(BaseModel):
5454
pgp_key: typing.Optional[str] = None
5555
arch_list: typing.List[str]
5656
modularity: typing.Optional[typing.Dict] = None
57+
# We're only going to take versions from 'data' column
58+
data: typing.Optional[typing.Dict[str, typing.Any]] = None
59+
60+
@root_validator(pre=True)
61+
def filter_data_to_versions_only(cls, values):
62+
raw_data = values.data if hasattr(values, 'data') else {}
63+
versions = raw_data.get("versions")
64+
65+
values.data = {"versions": versions} if versions else {}
66+
return values
5767

5868
class Config:
5969
from_attributes = True

0 commit comments

Comments
 (0)