Add Claim.reset_id() method (#570)

fix #472
LeMyst · Jun 15, 2023 · 9e326de · 9e326de
1 parent c42e1ae
commit 9e326de
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/test/test_wbi_core.py b/test/test_wbi_core.py
@@ -291,3 +291,10 @@ def test_count_references(self):
 
     def test_get_qualifier_properties(self):
         self.assertTrue(len(self.common_item.claims.get(property='P2067')))
+
+    def test_claim_reset_id(self):
+        item = wbi.item.get('Q582')
+        claim = item.claims.get('P31')[0]
+        assert claim.id is not None
+        claim.reset_id()
+        assert claim.id is None
diff --git a/wikibaseintegrator/models/claims.py b/wikibaseintegrator/models/claims.py
@@ -131,7 +131,8 @@ def __len__(self):
 class Claim(BaseModel):
     DTYPE = 'claim'
 
-    def __init__(self, qualifiers: Optional[Qualifiers] = None, rank: Optional[WikibaseRank] = None, references: Optional[Union[References, List[Union[Claim, List[Claim]]]]] = None, snaktype: WikibaseSnakType = WikibaseSnakType.KNOWN_VALUE) -> None:
+    def __init__(self, qualifiers: Optional[Qualifiers] = None, rank: Optional[WikibaseRank] = None,
+                 references: Optional[Union[References, List[Union[Claim, List[Claim]]]]] = None, snaktype: WikibaseSnakType = WikibaseSnakType.KNOWN_VALUE) -> None:
         """
 
         :param qualifiers:
@@ -309,6 +310,12 @@ def has_equal_qualifiers(self, other: Claim) -> bool:
 
         return True
 
+    def reset_id(self):
+        """
+        Reset the ID of the current claim
+        """
+        self.id = None
+
     # TODO: rewrite this?
     def __contains__(self, item):
         if isinstance(item, Claim):