jenkins-bot has submitted this change. ( https://gerrit.wikimedia.org/r/c/pywikibot/core/+/786171 )
Change subject: [IMPR] Enable short creation of a site if family name is equal to site code ......................................................................
[IMPR] Enable short creation of a site if family name is equal to site code
Remove assert statement in Family.load because it is tested lated and a meaningful warning is printed in such case.
Change-Id: Ie8d36c367715ca45c5bf3572b8f88160f5249622 --- M pywikibot/__init__.py M pywikibot/data/sparql.py M pywikibot/family.py M pywikibot/page/_wikibase.py M pywikibot/site/_apisite.py M pywikibot/specialbots/_upload.py M scripts/checkimages.py M scripts/data_ingestion.py M scripts/harvest_template.py M scripts/illustrate_wikidata.py M scripts/maintenance/sorting_order.py M tests/i18n_tests.py M tests/page_tests.py M tests/site_tests.py M tests/wikibase_edit_tests.py M tests/wikibase_tests.py 16 files changed, 25 insertions(+), 22 deletions(-)
Approvals: Xqt: Looks good to me, approved jenkins-bot: Verified
diff --git a/pywikibot/__init__.py b/pywikibot/__init__.py index b94b610..aead197 100644 --- a/pywikibot/__init__.py +++ b/pywikibot/__init__.py @@ -1113,7 +1113,7 @@
Override default family::
- site = pywikibot.Site(family='wikisource') + site = pywikibot.Site(fam='wikisource')
Setting a specific site::
@@ -1140,6 +1140,10 @@ **Never create a site object via interface class directly.** Always use this factory method.
+ .. versionchanged:: 7.3 + Short creation if site code is equal to family name like + `Site('commons')`, `Site('meta')` or `Site('wikidata')`. + :param code: language code (override config.mylang) code may also be a sitename like 'wikipedia:test' :param fam: family name or object (override config.family) @@ -1167,6 +1171,9 @@ 'should be provided') fam, _, code = code.partition(':') else: + if not fam: # try code as family + with suppress(exceptions.UnknownFamilyError): + fam = Family.load(code) # Fallback to config defaults code = code or _config.mylang fam = fam or _config.family diff --git a/pywikibot/data/sparql.py b/pywikibot/data/sparql.py index c3a44ca..d08d323 100644 --- a/pywikibot/data/sparql.py +++ b/pywikibot/data/sparql.py @@ -49,7 +49,7 @@ """ # default to Wikidata if not repo and not endpoint: - repo = Site('wikidata', 'wikidata') + repo = Site('wikidata')
if repo: try: diff --git a/pywikibot/family.py b/pywikibot/family.py index 08b02cb..871a36c 100644 --- a/pywikibot/family.py +++ b/pywikibot/family.py @@ -573,10 +573,6 @@ if fam is None: fam = config.family
- assert all(x in NAME_CHARACTERS for x in fam), \ - 'Name of family "{}" must be ASCII letters and digits ' \ - '[a-zA-Z0-9]'.format(fam) - if fam in Family._families: return Family._families[fam]
diff --git a/pywikibot/page/_wikibase.py b/pywikibot/page/_wikibase.py index 3042b44..e6fda16 100644 --- a/pywikibot/page/_wikibase.py +++ b/pywikibot/page/_wikibase.py @@ -1304,7 +1304,7 @@ 'wikibase-form': lambda value, site: LexemeForm(site, value['id']), 'wikibase-sense': lambda value, site: LexemeSense(site, value['id']), 'commonsMedia': lambda value, site: - FilePage(pywikibot.Site('commons', 'commons'), value), # T90492 + FilePage(pywikibot.Site('commons'), value), # T90492 'globe-coordinate': pywikibot.Coordinate.fromWikibase, 'geo-shape': pywikibot.WbGeoShape.fromWikibase, 'tabular-data': pywikibot.WbTabularData.fromWikibase, @@ -1882,7 +1882,7 @@ Basic usage sample:
>>> import pywikibot - >>> repo = pywikibot.Site('wikidata:wikidata') + >>> repo = pywikibot.Site('wikidata') >>> L2 = pywikibot.LexemePage(repo, 'L2') # create a Lexeme page >>> list(L2.claims.keys()) # access the claims ['P5831', 'P5402'] diff --git a/pywikibot/site/_apisite.py b/pywikibot/site/_apisite.py index ce2c893..98c2c81 100644 --- a/pywikibot/site/_apisite.py +++ b/pywikibot/site/_apisite.py @@ -195,7 +195,7 @@ """ # TODO this only works for some WMF sites if not site: - site = pywikibot.Site('meta', 'meta') + site = pywikibot.Site('meta') req = site._request(expiry=datetime.timedelta(days=10), parameters={'action': 'sitematrix'}) data = req.submit() diff --git a/pywikibot/specialbots/_upload.py b/pywikibot/specialbots/_upload.py index 66a81bb..ddffd20 100644 --- a/pywikibot/specialbots/_upload.py +++ b/pywikibot/specialbots/_upload.py @@ -113,7 +113,7 @@ self.force_if_shared = force_if_shared
if config.upload_to_commons: - default_site = pywikibot.Site('commons:commons') + default_site = pywikibot.Site('commons') else: default_site = pywikibot.Site() self.target_site = target_site or default_site diff --git a/scripts/checkimages.py b/scripts/checkimages.py index 0c9430e..6bea64c 100755 --- a/scripts/checkimages.py +++ b/scripts/checkimages.py @@ -831,7 +831,7 @@ except NoPageError: return False # Image deleted, no hash found. Skip the image.
- site = pywikibot.Site('commons', 'commons') + site = pywikibot.Site('commons') commons_image_with_this_hash = next( iter(site.allimages(sha1=hash_found, total=1)), None) if commons_image_with_this_hash: diff --git a/scripts/data_ingestion.py b/scripts/data_ingestion.py index 28c75ae..c567c05 100755 --- a/scripts/data_ingestion.py +++ b/scripts/data_ingestion.py @@ -139,7 +139,7 @@ self.contents = None
if not site: - site = pywikibot.Site('commons:commons') + site = pywikibot.Site('commons')
# default title super().__init__(site, self.get_title('%(_filename)s.%(_ext)s')) diff --git a/scripts/harvest_template.py b/scripts/harvest_template.py index e647160..f6b778b 100755 --- a/scripts/harvest_template.py +++ b/scripts/harvest_template.py @@ -335,7 +335,7 @@ continue claim.setTarget(match.group('url')) elif claim.type == 'commonsMedia': - commonssite = pywikibot.Site('commons', 'commons') + commonssite = pywikibot.Site('commons') imagelink = pywikibot.Link( value, source=commonssite, default_namespace=6) image = pywikibot.FilePage(imagelink) diff --git a/scripts/illustrate_wikidata.py b/scripts/illustrate_wikidata.py index 6c8b587..35245ef 100755 --- a/scripts/illustrate_wikidata.py +++ b/scripts/illustrate_wikidata.py @@ -57,7 +57,7 @@ return
newclaim = pywikibot.Claim(self.repo, self.wdproperty) - commonssite = pywikibot.Site('commons', 'commons') + commonssite = pywikibot.Site('commons') imagelink = pywikibot.Link(imagename, source=commonssite, default_namespace=6) image = pywikibot.FilePage(imagelink) diff --git a/scripts/maintenance/sorting_order.py b/scripts/maintenance/sorting_order.py index 1d80e2e..06c373f 100755 --- a/scripts/maintenance/sorting_order.py +++ b/scripts/maintenance/sorting_order.py @@ -68,7 +68,7 @@
def main(): """Main entry function.""" - site = pywikibot.Site('meta', 'meta') + site = pywikibot.Site('meta') for list_name, page_name in pages.items(): page = pywikibot.Page(site, page_name, ns=site.namespaces.MEDIAWIKI) update_family(list_name, page) diff --git a/tests/i18n_tests.py b/tests/i18n_tests.py index 9fa1e94..4a17748 100755 --- a/tests/i18n_tests.py +++ b/tests/i18n_tests.py @@ -60,7 +60,7 @@ Test whether the localzation is found either with the Site object or with the site code. """ - site = Site('commons', 'commons') + site = Site('commons') for code in (site, 'commons'): with self.subTest(code=code): self.assertEqual(i18n.translate(code, self.xdict), diff --git a/tests/page_tests.py b/tests/page_tests.py index 5176308..c58d6b8 100755 --- a/tests/page_tests.py +++ b/tests/page_tests.py @@ -1238,7 +1238,7 @@ def test_create_short_link(self): """Test create_short_link function.""" # Make sure test user is logged in on meta:meta (T244062) - meta = pywikibot.Site('meta:meta') + meta = pywikibot.Site('meta') if not meta.logged_in(): meta.login() if not meta.user(): diff --git a/tests/site_tests.py b/tests/site_tests.py index 2d5642e..2c0f0e3 100755 --- a/tests/site_tests.py +++ b/tests/site_tests.py @@ -3346,7 +3346,7 @@ with self.assertRaises(KeyError): site.family.hostname('wowwiki') with self.assertRaises(UnknownSiteError): - pywikibot.Site('wowwiki', 'wowwiki') + pywikibot.Site('wowwiki') with self.assertRaises(UnknownSiteError): pywikibot.Site('ceb', 'wowwiki')
diff --git a/tests/wikibase_edit_tests.py b/tests/wikibase_edit_tests.py index 40cd4d3..589d12d 100755 --- a/tests/wikibase_edit_tests.py +++ b/tests/wikibase_edit_tests.py @@ -349,7 +349,7 @@
# set new claim claim = pywikibot.page.Claim(testsite, 'P27199', datatype='geo-shape') - commons_site = pywikibot.Site('commons', 'commons') + commons_site = pywikibot.Site('commons') page = pywikibot.Page(commons_site, 'Data:Lyngby Hovedgade.map') target = pywikibot.WbGeoShape(page) claim.setTarget(target) @@ -369,7 +369,7 @@ # set new claim claim = pywikibot.page.Claim( testsite, 'P30175', datatype='tabular-data') - commons_site = pywikibot.Site('commons', 'commons') + commons_site = pywikibot.Site('commons') page = pywikibot.Page(commons_site, 'Data:Bea.gov/GDP by state.tab') target = pywikibot.WbGeoShape(page) claim.setTarget(target) diff --git a/tests/wikibase_tests.py b/tests/wikibase_tests.py index 4ced0de..96b7ad6 100755 --- a/tests/wikibase_tests.py +++ b/tests/wikibase_tests.py @@ -640,7 +640,7 @@
def setUp(self): """Setup tests.""" - self.commons = pywikibot.Site('commons', 'commons') + self.commons = pywikibot.Site('commons') self.page = Page(self.commons, 'Data:Lyngby Hovedgade.map') super().setUp()
@@ -716,7 +716,7 @@
def setUp(self): """Setup tests.""" - self.commons = pywikibot.Site('commons', 'commons') + self.commons = pywikibot.Site('commons') self.page = Page(self.commons, 'Data:Bea.gov/GDP by state.tab') super().setUp()