From 5a40cf1683cabb89154ec386bcbacebc6302d7f2 Mon Sep 17 00:00:00 2001 From: OMGeeky Date: Fri, 10 Nov 2023 23:07:43 +0100 Subject: [PATCH] fix some items not having parameters crashing the script --- gen/drive3/src/api/mod.rs | 4 ++-- src/generator/lib/enum_utils.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/gen/drive3/src/api/mod.rs b/gen/drive3/src/api/mod.rs index a5b2a25052..aca6f89327 100644 --- a/gen/drive3/src/api/mod.rs +++ b/gen/drive3/src/api/mod.rs @@ -31,5 +31,5 @@ pub use method_builders::*; mod call_builders; pub use call_builders::*; -mod enums; -pub use enums::*; +pub mod enums; +pub(crate) use enums::*; diff --git a/src/generator/lib/enum_utils.py b/src/generator/lib/enum_utils.py index 98ddf573f5..c00d57d319 100644 --- a/src/generator/lib/enum_utils.py +++ b/src/generator/lib/enum_utils.py @@ -84,7 +84,7 @@ def find_enums_in_context(c: Context) -> list: for k, v in c.fqan_map.items(): # print(k) k = _parse_method_id(k, c) - if v.parameters: + if v.get('parameters'): for pk, pv in v.parameters.items(): enums = _add_enum_value(k, pk, pv, enums) return list(enums.values())