✨ Support dict(entry) by adding Entry.keys() and __iter__

bibtexparser/model.py

@@ -1,4 +1,5 @@
 import abc
+from collections.abc import Iterator
 from typing import Any
 
 _ALLOWED_ENCLOSINGS = (None, "{", '"', "no-enclosing")
@@ -454,6 +455,16 @@ def items(self) -> list[tuple[str, Any]]:
             ("ID", self.key),
         ] + [(f.key, f.value) for f in self.fields]
 
+    def keys(self) -> list[str]:
+        """Dict-mimicking, for partial v1.x backwards compatibility.
+
+        Same keys and order as in `items()`, e.g. allowing ``dict(entry)``."""
+        return [key for key, _ in self.items()]
+
+    def __iter__(self) -> Iterator[str]:
+        """Dict-mimicking iteration over the keys, as in `keys()`."""
+        return iter(self.keys())
+
     def __hash__(self) -> int:
         return hash((type(self), self._start_line_in_file, self._raw, self._entry_type, self._key))
 

tests/test_model.py

@@ -103,6 +103,14 @@ def test_entry_contains():
     assert "ID" in entry
 
 
+def test_entry_dict_conversion():
+    entry = Entry("article", "key", [Field("field", "value", 1), Field("foo", "bar", 2)], 1, "raw")
+    assert list(entry) == ["ENTRYTYPE", "ID", "field", "foo"]
+    expected = {"ENTRYTYPE": "article", "ID": "key", "field": "value", "foo": "bar"}
+    assert dict(entry) == dict(entry.items()) == expected
+    assert {**entry} == expected
+
+
 def test_entry_setitem_field():
     entry = Entry("article", "key", [Field("field", "value", 1)], 1, "raw")
     entry["field"] = "new_value"