From e75df42bec7ee58993f97e59d224a3f3921809f6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jonathan=20K=C3=B6ssler?= Date: Wed, 25 Sep 2024 11:07:20 +0200 Subject: [PATCH] feat: skip keys in int conversion --- pyinfra/storage/proto_data_loader.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/pyinfra/storage/proto_data_loader.py b/pyinfra/storage/proto_data_loader.py index 1bb6d9c..2e9cc58 100644 --- a/pyinfra/storage/proto_data_loader.py +++ b/pyinfra/storage/proto_data_loader.py @@ -87,8 +87,13 @@ class ProtoDataLoader: def convert_int64_fields(obj): # FIXME: find a more sophisticated way to convert int64 fields (defaults to str in python) + + # we skip the following keys because the values are expected to be of type str + skip_keys = ["col", "row", "numberOfCols", "numberOfRows"] if isinstance(obj, dict): for key, value in obj.items(): + if key in skip_keys: + continue obj[key] = convert_int64_fields(value) elif isinstance(obj, list): return [convert_int64_fields(item) for item in obj]