Skip to content

Commit

Permalink
Fix differences to iib without Pydantic
Browse files Browse the repository at this point in the history
  • Loading branch information
xDaile committed Jan 14, 2024
1 parent 4f7cfc7 commit 819b658
Showing 1 changed file with 12 additions and 11 deletions.
23 changes: 12 additions & 11 deletions iib/common/pydantic_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -123,6 +123,7 @@ def bundles_needed_with_check_related_images(self) -> 'AddPydanticModel':
def get_json_for_request(self):
"""Return json with the parameters we store in the db."""
return self.model_dump(
# include=["deprecation_list"],
exclude=[
"add_arches",
"build_tags",
Expand All @@ -131,7 +132,7 @@ def get_json_for_request(self):
"overwrite_from_index",
"overwrite_from_index_token",
],
exclude_defaults=True,
exclude_none=True,
)


Expand Down Expand Up @@ -167,7 +168,7 @@ def get_json_for_request(self):
"""Return json with the parameters we store in the db."""
return self.model_dump(
exclude=["add_arches", "build_tags", "overwrite_from_index", "overwrite_from_index_token"],
exclude_defaults=True,
exclude_none=True,
)

def _get_all_keys_to_check_in_db(self):
Expand All @@ -191,7 +192,7 @@ class RegenerateBundlePydanticModel(PydanticModel):
"""Datastructure of the request to /builds/regenerate-bundle API point."""

# BUNDLE_IMAGE, from_bundle_image_resolved, build_tags?
bundle_replacements: Optional[Dict[str, str]] = {}
bundle_replacements: Optional[Dict[str, str]] = None
from_bundle_image: Annotated[str, AfterValidator(image_format_check)]
organization: Optional[str] = None
registry_auths: Optional[RegistryAuths] = None # not in db
Expand All @@ -200,7 +201,7 @@ def get_json_for_request(self):
"""Return json with the parameters we store in the db."""
return self.model_dump(
exclude=["registry_auths"],
exclude_defaults=True,
exclude_none=True,
)

def _get_all_keys_to_check_in_db(self):
Expand Down Expand Up @@ -255,7 +256,7 @@ def get_json_for_request(self):
"""Return json with the parameters we store in the db."""
return self.model_dump(
exclude=["build_tags", "overwrite_target_index", "overwrite_target_index_token"],
exclude_defaults=True,
exclude_none=True,
)

def _get_all_keys_to_check_in_db(self):
Expand All @@ -275,13 +276,13 @@ class CreateEmptyIndexPydanticModel(PydanticModel):
AfterValidator(image_format_check),
AfterValidator(length_validator),
]
labels: Optional[Dict[str, str]] = {}
output_fbc: Optional[bool] = False
labels: Optional[Dict[str, str]] = None # old request without this parameter will not have empty labels
output_fbc: Optional[bool] = None # old request without this parameter will not have empty output_fbc

def get_json_for_request(self):
"""Return json with the parameters we store in the db."""
return self.model_dump(
exclude_defaults=True,
exclude_none=True,
)

def _get_all_keys_to_check_in_db(self):
Expand All @@ -301,7 +302,7 @@ def get_json_for_request(self):
"""Return json with the parameters we store in the db."""
return self.model_dump(
exclude=["registry_auths"],
exclude_defaults=True,
exclude_none=True,
)


Expand All @@ -321,7 +322,7 @@ class FbcOperationsPydanticModel(PydanticModel):
AfterValidator(length_validator),
AfterValidator(get_unique_bundles),
AfterValidator(images_format_check),
] = []
] = None # old request without this parameter will not have empty list but None
build_tags: Optional[List[str]] = []
distribution_scope: Annotated[
Optional[DISTRIBUTION_SCOPE_LITERAL], BeforeValidator(distribution_scope_lower),
Expand Down Expand Up @@ -349,7 +350,7 @@ def get_json_for_request(self):
"""Return json with the parameters we store in the db."""
return self.model_dump(
exclude=["add_arches", "build_tags", "overwrite_from_index", "overwrite_from_index_token"],
exclude_defaults=True,
exclude_none=True,
)

def _get_all_keys_to_check_in_db(self):
Expand Down

0 comments on commit 819b658

Please sign in to comment.