some minor simplifications

StaatsbibliothekBerlin · Jul 20, 2023 · 86f2e6e · 86f2e6e
1 parent 8200511
commit 86f2e6e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/oai_status/list_records.py b/oai_status/list_records.py
@@ -1,4 +1,5 @@
 import sys
+from types import SimpleNamespace
 from typing import Iterable
 
 from delb import Document, TagNode
@@ -26,8 +27,7 @@ def get_resumption_token(doc: Document) -> str:
     'metadataPrefix%3Doai_dc%26set%3Dillustrierte.liedflugschriften%26cursor%3D50%26batch_size%3D51'
 
     '''
-    token_node = doc.xpath('//resumptionToken').first
-    if token_node:
+    if token_node := doc.xpath('//resumptionToken').first:
         return token_node.full_text
     return None
 
@@ -67,7 +67,7 @@ def list_records(
     retrieve records of a certain set from OAI endpoint, i.e. make requests with the `ListRecords`
     verb.
     '''
-    counter = type('counter', (), dict(left=limit))()
+    counter = SimpleNamespace(left=limit)
 
     def yield_records(doc: Document) -> Iterable[TagNode]:
         for record in get_records(doc):