feat: Parse numerical strings in yaml files as strings

src/docs/generate_oci_doc_yaml.py

@@ -273,9 +273,9 @@ def read_documentation_yaml(doc_file: str) -> Dict:
         logging.info("Opening file %s", doc_file)
         with open(doc_file, "r", encoding="UTF-8") as file:
             try:
-                base_doc_data = DocSchema(**yaml.safe_load(file) or {}).dict(
-                    exclude_none=True
-                )
+                base_doc_data = DocSchema(
+                    **yaml.load(file, Loader=yaml.BaseLoader) or {}
+                ).dict(exclude_none=True)
             except (yaml.YAMLError, pydantic.ValidationError) as exc:
                 msg = f"Error loading the {doc_file} file"
                 raise Exception(msg) from exc

src/image/merge_release_info.py

@@ -57,7 +57,7 @@ def backfill_higher_risks(channels: dict) -> None:
 
     print(f"Getting existing image trigger from {args.image_trigger}")
     with open(args.image_trigger, encoding="UTF-8") as trigger:
-        image_trigger = yaml.safe_load(trigger)
+        image_trigger = yaml.load(trigger, Loader=yaml.BaseLoader)
 
     _ = ImageSchema(**image_trigger)
 

src/image/prepare_single_image_build_matrix.py

@@ -44,7 +44,7 @@ def validate_image_trigger(data: dict) -> None:
 
     print(f"Generating build matrix for {image_trigger_file}")
     with open(image_trigger_file, encoding="UTF-8") as bf:
-        image_trigger = yaml.safe_load(bf)
+        image_trigger = yaml.load(bf, Loader=yaml.BaseLoader)
         try:
             validate_image_trigger(image_trigger)
         except pydantic.error_wrappers.ValidationError as err:

src/image/release.py

@@ -67,7 +67,7 @@
 
 print(f"Parsing image trigger {args.image_trigger}")
 with open(args.image_trigger, encoding="UTF-8") as trigger:
-    image_trigger = ImageSchema(**yaml.safe_load(trigger))
+    image_trigger = ImageSchema(**yaml.load(trigger, Loader=yaml.BaseLoader))
 
 tag_mapping_from_trigger = {}
 for track, risks in image_trigger.release.items():

src/tests/requirements.txt

@@ -1,2 +1,3 @@
 docker
-pydantic==1.9.0
+pydantic==1.9.0
+requests==2.31

src/uploads/infer_image_track.py

@@ -6,6 +6,7 @@
 import subprocess
 import yaml
 
+
 logging.basicConfig()
 
 
@@ -31,7 +32,7 @@ def get_release_from_codename(codename: str) -> str:
 with open(
     f"{args.recipe_dirname.rstrip('/')}/rockcraft.yaml", encoding="UTF-8"
 ) as rockcraft_file:
-    rockcraft_yaml = yaml.safe_load(rockcraft_file)
+    rockcraft_yaml = yaml.load(rockcraft_file, Loader=yaml.BaseLoader)
 
 rock_base = (
     rockcraft_yaml["base"]
@@ -43,7 +44,7 @@ def get_release_from_codename(codename: str) -> str:
     base_release = float(rock_base.replace(":", "@").split("@")[-1])
 except ValueError:
     logging.warning(
-        f"Could not infer rock's base release from {rock_base}. Trying with codename."
+        f"Could not infer ROCK's base release from {rock_base}. Trying with codename."
     )
     base_release = float(
         get_release_from_codename(rock_base.replace(":", "@").split("@")[-1])