fix: allow more builtin fields

Desdaemon · May 21, 2024 · 20a554a · 20a554a
1 parent 13fe489
commit 20a554a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/src/analyze.rs b/src/analyze.rs
@@ -33,8 +33,10 @@ static MODEL_METHODS: phf::Set<&[u8]> = phf::phf_set!(
 	b"with_env",
 	b"sudo",
 	b"_for_xml_id",
+	b"exists",
 	// TODO: Limit to Forms only
 	b"new",
+	b"save",
 );
 
 /// The subset of types that may resolve to a model.

diff --git a/src/python.rs b/src/python.rs
@@ -998,10 +998,18 @@ impl Backend {
 							}
 
 							let attribute = attribute.unwrap();
-							static MODEL_BUILTINS: phf::Set<&str> =
-								phf::phf_set!("env", "id", "ids", "display_name", "create_date", "write_date");
+							static MODEL_BUILTINS: phf::Set<&str> = phf::phf_set!(
+								"env",
+								"id",
+								"ids",
+								"display_name",
+								"create_date",
+								"write_date",
+								"pool",
+								"record"
+							);
 							let prop = String::from_utf8_lossy(&contents[attribute.byte_range()]);
-							if prop.starts_with(' ') || MODEL_BUILTINS.contains(&prop) {
+							if prop.starts_with('_') || MODEL_BUILTINS.contains(&prop) {
 								return ControlFlow::Continue(entered);
 							}