http://www.mediawiki.org/wiki/Special:Code/pywikipedia/10100

Revision: 10100 Author: xqt Date: 2012-04-09 12:20:35 +0000 (Mon, 09 Apr 2012) Log Message: ----------- enable single number or single str/unicode for namespace filter

Modified Paths: -------------- trunk/pywikipedia/pagegenerators.py

Modified: trunk/pywikipedia/pagegenerators.py =================================================================== --- trunk/pywikipedia/pagegenerators.py 2012-04-09 12:00:15 UTC (rev 10099) +++ trunk/pywikipedia/pagegenerators.py 2012-04-09 12:20:35 UTC (rev 10100) @@ -1026,14 +1026,17 @@ of the given namespaces.

The namespace list can contain both integers (namespace numbers) and - strings/unicode strings (namespace names). + strings/unicode strings (namespace names). Namespace may also be a single + number or a single string. """ # convert namespace names to namespace numbers if site is None: site = pywikibot.getSite() + if isinstance(namespace, (int, basestring)): + namespace = [namespace] for i in xrange(len(namespaces)): ns = namespaces[i] - if isinstance(ns, unicode) or isinstance(ns, str): + if isinstance(ns, basestring): index = site.getNamespaceIndex(ns) if index is None: raise ValueError(u'Unknown namespace: %s' % ns)