Merge "Add title function to MediaInfo."

wikimedia · Sep 8, 2024 · 7c36718 · 7c36718
2 parents fb6351b + 161ad71
commit 7c36718
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/pywikibot/page/_wikibase.py b/pywikibot/page/_wikibase.py
@@ -521,6 +521,14 @@ def getID(self, numeric: bool = False):
         self._assert_has_id()
         return super().getID(numeric=numeric)
 
+    def title(self):
+        """
+        Return ID as title of the MediaInfo.
+
+        :return: str: the entity identifier
+        """
+        return self.getID()
+
     def editLabels(self, labels: LANGUAGE_TYPE, **kwargs) -> None:
         """Edit MediaInfo labels (eg. captions).
 

diff --git a/tests/file_tests.py b/tests/file_tests.py
@@ -362,6 +362,7 @@ def test_data_item(self):
         self.assertEqual('-1', item.id)
         item.get()
         self.assertEqual('M14634781', item.id)
+        self.assertEqual('M14634781', item.title())
         self.assertIsInstance(
             item.labels, pywikibot.page._collections.LanguageDict)
         self.assertIsInstance(
@@ -400,6 +401,8 @@ def test_data_item_when_no_file_or_data_item(self):
 
         with self.assertRaises(NoWikibaseEntityError):
             item.get()
+        with self.assertRaises(NoWikibaseEntityError):
+            item.title()
         with self.assertRaises(NoWikibaseEntityError):
             item.labels