def test_basic(self):
"""Test adding text."""
(text, newtext, always) = add_text(self.page,
'bar',
putText=False,
oldTextGiven='foo\n{{linkfa}}')
self.assertEqual('foo\n{{linkfa}}\nbar', newtext)
def test_with_category(self):
"""Test adding text before categories."""
(_, newtext, _) = add_text(self.page,
'bar',
putText=False,
oldTextGiven='foo\n[[Category:Foo]]')
self.assertEqual('foo\nbar\n\n[[Category:Foo]]', newtext)
def addCommonscat(self, page):
"""
Add CommonsCat template to page.
Take a page. Go to all the interwiki page looking for a commonscat
template. When all the interwiki's links are checked and a proper
category is found add it to the page.
"""
self.current_page = page
# Get the right templates for this page
primaryCommonscat, commonscatAlternatives = self.getCommonscatTemplate(
page.site.code)
commonscatLink = self.getCommonscatLink(page)
if commonscatLink:
pywikibot.output('Commonscat template is already on %s' %
page.title())
(currentCommonscatTemplate, currentCommonscatTarget, LinkText,
Note) = commonscatLink
checkedCommonscatTarget = self.checkCommonscatLink(
currentCommonscatTarget)
if (currentCommonscatTarget == checkedCommonscatTarget):
# The current commonscat link is good
pywikibot.output('Commonscat link at %s to Category:%s is ok' %
(page.title(), currentCommonscatTarget))
return True
elif checkedCommonscatTarget != '':
# We have a new Commonscat link, replace the old one
self.changeCommonscat(page, currentCommonscatTemplate,
currentCommonscatTarget,
primaryCommonscat,
checkedCommonscatTarget, LinkText, Note)
return True
else:
# Commonscat link is wrong
commonscatLink = self.findCommonscatLink(page)
if (commonscatLink != ''):
self.changeCommonscat(page, currentCommonscatTemplate,
currentCommonscatTarget,
primaryCommonscat, commonscatLink)
# TODO: if the commonsLink == '', should it be removed?
elif self.skipPage(page):
pywikibot.output('Found a template in the skip list. Skipping %s' %
page.title())
else:
commonscatLink = self.findCommonscatLink(page)
if (commonscatLink != ''):
if commonscatLink == page.title():
textToAdd = '{{%s}}' % primaryCommonscat
else:
textToAdd = '{{%s|%s}}' % (primaryCommonscat,
commonscatLink)
rv = add_text(page,
textToAdd,
self.getOption('summary'),
always=self.getOption('always'))
self.options['always'] = rv[2]
return True
return True
def addCommonscat(self, page):
"""
Add CommonsCat template to page.
Take a page. Go to all the interwiki page looking for a commonscat
template. When all the interwiki's links are checked and a proper
category is found add it to the page.
"""
self.current_page = page
# Get the right templates for this page
primaryCommonscat, commonscatAlternatives = self.getCommonscatTemplate(
page.site.code)
commonscatLink = self.getCommonscatLink(page)
if commonscatLink:
pywikibot.output(u'Commonscat template is already on %s'
% page.title())
(currentCommonscatTemplate,
currentCommonscatTarget, LinkText, Note) = commonscatLink
checkedCommonscatTarget = self.checkCommonscatLink(
currentCommonscatTarget)
if (currentCommonscatTarget == checkedCommonscatTarget):
# The current commonscat link is good
pywikibot.output(u'Commonscat link at %s to Category:%s is ok'
% (page.title(), currentCommonscatTarget))
return True
elif checkedCommonscatTarget != u'':
# We have a new Commonscat link, replace the old one
self.changeCommonscat(page, currentCommonscatTemplate,
currentCommonscatTarget,
primaryCommonscat,
checkedCommonscatTarget, LinkText, Note)
return True
else:
# Commonscat link is wrong
commonscatLink = self.findCommonscatLink(page)
if (commonscatLink != u''):
self.changeCommonscat(page, currentCommonscatTemplate,
currentCommonscatTarget,
primaryCommonscat, commonscatLink)
# TODO: if the commonsLink == u'', should it be removed?
elif self.skipPage(page):
pywikibot.output("Found a template in the skip list. Skipping %s"
% page.title())
else:
commonscatLink = self.findCommonscatLink(page)
if (commonscatLink != u''):
if commonscatLink == page.title():
textToAdd = u'{{%s}}' % primaryCommonscat
else:
textToAdd = u'{{%s|%s}}' % (primaryCommonscat,
commonscatLink)
rv = add_text(page, textToAdd,
self.getOption('summary'),
always=self.getOption('always'))
self.options['always'] = rv[2]
return True
return True
def test_basic(self):
"""Test adding text."""
page = pywikibot.Page(self.site, 'foo')
(text, newtext, always) = add_text(page,
'bar',
putText=False,
oldTextGiven='foo\n{{linkfa}}')
self.assertEqual('foo\n{{linkfa}}\nbar', newtext)
def test_with_params(self):
"""Test adding text before {{linkfa|...}}."""
page = pywikibot.Page(self.site, 'foo')
(text, newtext, always) = add_text(page,
'bar',
putText=False,
oldTextGiven='foo\n{{linkfa|...}}')
self.assertEqual('foo\nbar\n\n{{linkfa|...}}\n', newtext)
def test_with_category(self):
"""Test adding text before categories."""
page = pywikibot.Page(self.site, 'foo')
(text, newtext, always) = add_text(
page, 'bar', putText=False,
oldTextGiven='foo\n[[Category:Foo]]')
self.assertEqual(
'foo\nbar\n\n[[Category:Foo]]',
newtext)
def test_basic(self):
"""Test adding text."""
page = pywikibot.Page(self.site, 'foo')
(text, newtext, always) = add_text(
page, 'bar', putText=False,
oldTextGiven='foo\n{{linkfa}}')
self.assertEqual(
'foo\n{{linkfa}}\nbar',
newtext)
def test_with_category(self):
"""Test adding text before categories."""
page = pywikibot.Page(self.site, 'foo')
(text, newtext,
always) = add_text(page,
'bar',
putText=False,
oldTextGiven='foo\n[[Category:Foo]]')
self.assertEqual('foo\nbar\n\n[[Category:Foo]]', newtext)
def test_with_params(self):
"""Test adding text before {{linkfa|...}}."""
page = pywikibot.Page(self.site, 'foo')
(text, newtext, always) = add_text(
page, 'bar', putText=False,
oldTextGiven='foo\n{{linkfa|...}}')
self.assertEqual(
'foo\nbar\n\n{{linkfa|...}}\n',
newtext)
def treat_page(self):
"""
Add CommonsCat template to page.
Take a page. Go to all the interwiki page looking for a commonscat
template. When all the interwiki's links are checked and a proper
category is found add it to the page.
"""
page = self.current_page
# Get the right templates for this page
primaryCommonscat, commonscatAlternatives = i18n.translate(
page.site.code,
commonscatTemplates,
fallback=i18n.DEFAULT_FALLBACK)
commonscatLink = self.getCommonscatLink(page)
if commonscatLink:
pywikibot.output('Commonscat template is already on ' +
page.title())
(currentCommonscatTemplate, currentCommonscatTarget, LinkText,
Note) = commonscatLink
checkedCommonscatTarget = self.checkCommonscatLink(
currentCommonscatTarget)
if (currentCommonscatTarget == checkedCommonscatTarget):
# The current commonscat link is good
pywikibot.output(
'Commonscat link at {} to Category:{} is ok'.format(
page.title(), currentCommonscatTarget))
return
if checkedCommonscatTarget:
# We have a new Commonscat link, replace the old one
self.changeCommonscat(page, currentCommonscatTemplate,
currentCommonscatTarget,
primaryCommonscat,
checkedCommonscatTarget, LinkText, Note)
return
# Commonscat link is wrong
commonscatLink = self.find_commons_category(page)
if commonscatLink:
self.changeCommonscat(page, currentCommonscatTemplate,
currentCommonscatTarget,
primaryCommonscat, commonscatLink)
# TODO: if the commonsLink == '', should it be removed?
elif self.skipPage(page):
pywikibot.output('Found a template in the skip list. Skipping ' +
page.title())
else:
commonscatLink = self.find_commons_category(page)
if commonscatLink:
if commonscatLink == page.title():
textToAdd = '{{%s}}' % primaryCommonscat
else:
textToAdd = '{{%s|%s}}' % (primaryCommonscat,
commonscatLink)
result, _, always = add_text(page,
textToAdd,
self.opt.summary,
always=self.opt.always)
if result is True:
self._save_counter += 1
self.opt.always = always