Reformating with Ruff

philippesaade-wmde · philippesaade-wmde · commit 45c3f7bae7fe · 2026-04-03T14:02:14.000+02:00
diff --git a/main.py b/main.py
@@ -34,45 +34,46 @@
 LABEL_CLEANUP_INTERVAL_SECONDS = int(os.environ.get("LABEL_CLEANUP_INTERVAL_SECONDS", 3600))
 _last_label_cleanup = 0.0
 
+
 @app.on_event("startup")
 async def startup():
     """Initialize database resources required by the API."""
     WikidataLabel.initialize_database()
 
+
 @app.get(
     "/",
     responses={
         200: {
             "description": "Returns a list of relevant Wikidata property PIDs with similarity scores",
             "content": {
                 "application/json": {
-                    "example": [{
-                        "Q42": "Douglas Adams (human), English writer, humorist, and dramatist...",
-                    }]
+                    "example": [
+                        {
+                            "Q42": "Douglas Adams (human), English writer, humorist, and dramatist...",
+                        }
+                    ]
                 }
             },
         },
         422: {
             "description": "Missing or invalid query parameter",
-            "content": {
-                "application/json": {
-                    "example": {"detail": "Invalid format specified"}
-                }
-            },
+            "content": {"application/json": {"example": {"detail": "Invalid format specified"}}},
         },
     },
 )
 async def get_textified_wd(
-    request: Request, background_tasks: BackgroundTasks,
+    request: Request,
+    background_tasks: BackgroundTasks,
     id: str = Query(..., examples="Q42,Q2"),
     pid: str = Query(None, examples="P31,P279"),
-    lang: str = 'en',
-    format: str = 'json',
+    lang: str = "en",
+    format: str = "json",
     external_ids: bool = True,
     references: bool = False,
     all_ranks: bool = False,
     qualifiers: bool = True,
-    fallback_lang: str = 'en'
+    fallback_lang: str = "en",
 ):
     """Retrieve Wikidata entities as structured JSON, natural text, or triplet lines.
 
@@ -95,7 +96,7 @@ async def get_textified_wd(
 
     **Returns:**
 
-    A dictionary keyed by requested entity ID (for example, `"Q42"`).  
+    A dictionary keyed by requested entity ID (for example, `"Q42"`).
     Each value depends on `format`:
 
     - **json**: Structured entity payload with label, description, aliases, and claims.
@@ -105,9 +106,9 @@ async def get_textified_wd(
     try:
         filter_pids = []
         if pid:
-            filter_pids = [p.strip() for p in pid.split(',')]
+            filter_pids = [p.strip() for p in pid.split(",")]
 
-        qids = [q.strip() for q in id.split(',')]
+        qids = [q.strip() for q in id.split(",")]
         label_factory = LazyLabelFactory(lang=lang, fallback_lang=fallback_lang)
 
         entities = {}
@@ -158,7 +159,9 @@ async def get_textified_wd(
                     fallback_lang=fallback_lang,
                     label_factory=label_factory,
                     debug=False,
-                ) if entity_data.get(qid) else None
+                )
+                if entity_data.get(qid)
+                else None
                 for qid in qids
             }
 
@@ -168,8 +171,10 @@ async def get_textified_wd(
                     all_ranks=all_ranks,
                     references=references,
                     filter_pids=filter_pids,
-                    qualifiers=qualifiers
-                ) if entity else None
+                    qualifiers=qualifiers,
+                )
+                if entity
+                else None
                 for qid, entity in entity_data.items()
             }
 
@@ -179,9 +184,9 @@ async def get_textified_wd(
                 return_data[qid] = None
                 continue
 
-            if format == 'text':
+            if format == "text":
                 results = entity.to_text(lang)
-            elif format == 'triplet':
+            elif format == "triplet":
                 results = entity.to_triplet()
             else:
                 results = entity.to_json()
diff --git a/src/Normalizer/JSONNormalizer.py b/src/Normalizer/JSONNormalizer.py
@@ -48,7 +48,6 @@ def __init__(
         self.fallback_lang = fallback_lang
         self.debug = debug
 
-
         self.label_factory = label_factory or LazyLabelFactory(lang=lang, fallback_lang=fallback_lang)
 
     # -------------------------------------------------------------------------
@@ -119,7 +118,7 @@ def normalize(
                 external_ids=external_ids,
                 include_references=references,
                 all_ranks=all_ranks,
-                qualifiers=qualifiers
+                qualifiers=qualifiers,
             )
             if claim_obj is not None and claim_obj.values:
                 claims_out.append(claim_obj)
@@ -171,7 +170,7 @@ def _build_claim(
                 statement=st,
                 datatype=datatype,
                 include_references=include_references,
-                qualifiers=qualifiers
+                qualifiers=qualifiers,
             )
             if cv is not None:
                 values.append(cv)
diff --git a/src/Normalizer/TTLNormalizer.py b/src/Normalizer/TTLNormalizer.py
@@ -94,7 +94,7 @@ def normalize(
         references: bool = False,
         all_ranks: bool = False,
         qualifiers: bool = True,
-        filter_pids: List[str] = []
+        filter_pids: List[str] = [],
     ) -> WikidataEntity:
         """Normalize the parsed graph into a ``WikidataEntity`` tree.
 
@@ -134,7 +134,7 @@ def normalize(
             include_references=references,
             all_ranks=all_ranks,
             qualifiers=qualifiers,
-            filter_pids=filter_pids
+            filter_pids=filter_pids,
         )
 
         entity = WikidataEntity(
@@ -191,7 +191,7 @@ def _claims_for_subject(
         include_references: bool,
         all_ranks: bool,
         qualifiers: bool,
-        filter_pids: List[str] = []
+        filter_pids: List[str] = [],
     ) -> Dict[str, List[Dict[str, Any]]]:
         """Return mapping: pid -> list of statement dicts."""
         out: DefaultDict[str, List[Dict[str, Any]]] = defaultdict(list)
@@ -335,7 +335,7 @@ def _build_claim_object(
 
             refs_obj: List[List[WikidataClaim]] = []
             if include_references:
-                for ref in (st.get("references") or []):
+                for ref in st.get("references") or []:
                     ref_claims = [
                         self._build_snak_claim(
                             pid=rpid,
diff --git a/src/Textifier/WikidataTextifier.py b/src/Textifier/WikidataTextifier.py
@@ -15,6 +15,7 @@
 # Atomic value types
 # ---------------------------------------------------------------------------
 
+
 @dataclass(slots=True)
 class WikidataText:
     """Object for Wikidata plain text values."""
@@ -124,6 +125,7 @@ def to_json(self) -> Any:
 # Core graph types
 # ---------------------------------------------------------------------------
 
+
 @dataclass(slots=True)
 class WikidataEntity:
     """Object for Wikidata entities."""
@@ -138,11 +140,11 @@ def __bool__(self) -> bool:
         """Return whether this entity has a usable id and label."""
         return bool(self.id) and self.label is not None and str(self.label) != ""
 
-    def to_text(self, lang='en', keep_empty: bool = False) -> str:
+    def to_text(self, lang="en", keep_empty: bool = False) -> str:
         """Render the entity into a readable text."""
-        lang_var = LANGUAGE_VARIABLES.get(lang, LANGUAGE_VARIABLES.get('en'))
+        lang_var = LANGUAGE_VARIABLES.get(lang, LANGUAGE_VARIABLES.get("en"))
 
-        label_str = str(self.label) if self.label else '<missing>'
+        label_str = str(self.label) if self.label else "<missing>"
         string = label_str
 
         if self.description:
@@ -206,12 +208,12 @@ def __bool__(self) -> bool:
             and any(bool(v) for v in self.values)
         )
 
-    def to_text(self, lang='en') -> str:
+    def to_text(self, lang="en") -> str:
         """Render the claim into a readable text."""
-        lang_var = LANGUAGE_VARIABLES.get(lang, LANGUAGE_VARIABLES.get('en'))
+        lang_var = LANGUAGE_VARIABLES.get(lang, LANGUAGE_VARIABLES.get("en"))
 
         if self.values:
-            values = lang_var[', '].join(v.to_text(lang) for v in self.values if v)
+            values = lang_var[", "].join(v.to_text(lang) for v in self.values if v)
             return f"{str(self.property.label)}: {values}"
 
         return f"{lang_var['has']} {str(self.property.label)}"
@@ -249,10 +251,9 @@ def to_triplet(self, as_qualifier: bool = False) -> str:
 @dataclass(slots=True)
 class WikidataClaimValue:
     """Object for Wikidata claim values."""
+
     claim: WikidataClaim
-    value: Optional[
-        Union[WikidataEntity, WikidataQuantity, WikidataTime, WikidataCoordinates, WikidataText]
-    ] = None
+    value: Optional[Union[WikidataEntity, WikidataQuantity, WikidataTime, WikidataCoordinates, WikidataText]] = None
     qualifiers: List[WikidataClaim] = field(default_factory=list)
     references: List[List[WikidataClaim]] = field(default_factory=list)
     rank: Optional[str] = None  # preferred|normal|deprecated
@@ -261,9 +262,9 @@ def __bool__(self) -> bool:
         """Return whether this claim value has non-empty values."""
         return self.value is not None and str(self.value) != ""
 
-    def to_text(self, lang='en') -> str:
+    def to_text(self, lang="en") -> str:
         """Render the value and qualifiers as readable text."""
-        lang_var = LANGUAGE_VARIABLES.get(lang, LANGUAGE_VARIABLES.get('en'))
+        lang_var = LANGUAGE_VARIABLES.get(lang, LANGUAGE_VARIABLES.get("en"))
 
         if not self:
             return ""
diff --git a/src/WikidataLabel.py b/src/WikidataLabel.py
diff --git a/src/utils.py b/src/utils.py
