@@ -0,0 +1,6 @@ | |||||
*.pyc | |||||
*.egg | |||||
*.egg-info | |||||
.DS_Store | |||||
build | |||||
docs/_build |
@@ -0,0 +1,19 @@ | |||||
Copyright (C) 2012 by Ben Kurtovic <ben.kurtovic@verizon.net> | |||||
Permission is hereby granted, free of charge, to any person obtaining a copy | |||||
of this software and associated documentation files (the "Software"), to deal | |||||
in the Software without restriction, including without limitation the rights to | |||||
use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies | |||||
of the Software, and to permit persons to whom the Software is furnished to do | |||||
so, subject to the following conditions: | |||||
The above copyright notice and this permission notice shall be included in all | |||||
copies or substantial portions of the Software. | |||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | |||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | |||||
SOFTWARE. |
@@ -0,0 +1,105 @@ | |||||
mwtemplateparserfromhell | |||||
======================== | |||||
**mwtemplateparserfromhell** (the *MediaWiki Template Parser from Hell*) is a | |||||
Python package that provides an easy-to-use and outrageously powerful template | |||||
parser for MediaWiki_ wikicode. | |||||
Coded by Earwig_ and named by `Σ`_. | |||||
Installation | |||||
------------ | |||||
The easiest way to install the parser is through the `Python Package Index`_, | |||||
so you can install the latest release with ``pip install | |||||
mwtemplateparserfromhell`` (`get pip`_). Alternatively, get the latest | |||||
development version:: | |||||
git clone git://github.com/earwig/mwtemplateparserfromhell.git mwtemplateparserfromhell | |||||
cd mwtemplateparserfromhell | |||||
python setup.py install | |||||
You can run the comprehensive unit testing suite with ``python setup.py test``. | |||||
Usage | |||||
----- | |||||
Normal usage is rather straightforward (where ``text`` is page text):: | |||||
>>> import mwtemplateparserfromhell | |||||
>>> parser = mwtemplateparserfromhell.Parser() | |||||
>>> templates = parser.parse(text) | |||||
``templates`` is a list of ``mwtemplateparserfromhell.Template`` objects, which | |||||
contain a ``name`` attribute, a ``params`` attribute, and a ``get()`` method. | |||||
For example:: | |||||
>>> templates = parser.parse("{{foo|bar|baz|eggs=spam}}") | |||||
>>> print templates | |||||
[Template(name="foo", params={"1": "bar", "2": "baz", "eggs": "spam"})] | |||||
>>> print templates[0].name | |||||
foo | |||||
>>> print templates[0].params | |||||
['bar', 'baz'] | |||||
>>> print templates[0].get(0) | |||||
bar | |||||
>>> print templates[0].get("eggs") | |||||
spam | |||||
If ``get``\ 's argument is a number *n*, it'll return the *n*\ th parameter, | |||||
otherwise it will return the parameter with the given name. Unnamed parameters | |||||
are given numerical names starting with 1, so ``{{foo|bar}}`` is the same as | |||||
``{{foo|1=bar}}``, and ``templates[0].get(0) is templates[0].get("1")``. | |||||
By default, nested templates are supported like so:: | |||||
>>> templates = parser.parse("{{foo|this {{includes a|template}}}}") | |||||
>>> print templates | |||||
[Template(name="foo", params={"1": "this {{includes a|template}}"})] | |||||
>>> print templates[0].get(0) | |||||
this {{includes a|template}} | |||||
>>> print templates[0].get(0).templates | |||||
[Template(name="includes a", params={"1": "template"})] | |||||
>>> print templates[0].get(0).templates[0].params[0] | |||||
template | |||||
Integration | |||||
----------- | |||||
``mwtemplateparserfromhell`` is used by and originally developed for | |||||
EarwigBot_; ``Page`` objects have a ``parse_templates`` method that essentially | |||||
calls ``Parser().parse()`` on ``page.get()``. | |||||
If you're using PyWikipedia_, your code might look like this:: | |||||
import mwtemplateparserfromhell | |||||
import wikipedia as pywikibot | |||||
def parse_templates(title): | |||||
site = pywikibot.get_site() | |||||
page = pywikibot.Page(site, title) | |||||
text = page.get() | |||||
parser = mwtemplateparserfromhell.Parser() | |||||
return parser.parse(text) | |||||
If you're not using a library, you can parse templates in any page using the | |||||
following code (via the API_):: | |||||
import json | |||||
import urllib | |||||
import mwtemplateparserfromhell | |||||
API_URL = "http://en.wikipedia.org/w/api.php" | |||||
def parse_templates(title): | |||||
raw = urllib.urlopen(API_URL, data).read() | |||||
res = json.loads(raw) | |||||
text = res["query"]["pages"].values()[0]["revisions"][0]["*"] | |||||
parser = mwtemplateparserfromhell.Parser() | |||||
return parser.parse(text) | |||||
.. _MediaWiki: http://mediawiki.org | |||||
.. _Earwig: http://en.wikipedia.org/wiki/User:The_Earwig | |||||
.. _Σ: http://en.wikipedia.org/wiki/User:Σ | |||||
.. _Python Package Index: http://pypi.python.org | |||||
.. _get pip: http://pypi.python.org/pypi/pip | |||||
.. _EarwigBot: https://github.com/earwig/earwigbot | |||||
.. _PyWikipedia: http://pywikipediabot.sourceforge.net/ | |||||
.. _API: http://mediawiki.org/wiki/API |
@@ -0,0 +1,38 @@ | |||||
# -*- coding: utf-8 -*- | |||||
# | |||||
# Copyright (C) 2012 by Ben Kurtovic <ben.kurtovic@verizon.net> | |||||
# | |||||
# Permission is hereby granted, free of charge, to any person obtaining a copy | |||||
# of this software and associated documentation files (the "Software"), to deal | |||||
# in the Software without restriction, including without limitation the rights | |||||
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |||||
# copies of the Software, and to permit persons to whom the Software is | |||||
# furnished to do so, subject to the following conditions: | |||||
# | |||||
# The above copyright notice and this permission notice shall be included in | |||||
# all copies or substantial portions of the Software. | |||||
# | |||||
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |||||
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |||||
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | |||||
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |||||
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |||||
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | |||||
# SOFTWARE. | |||||
""" | |||||
`mwtemplateparserfromhell | |||||
<https://github.com/earwig/mwtemplateparserfromhell>`_ (the MediaWiki Template | |||||
Parser from Hell) is a Python package that provides an easy-to-use and | |||||
outrageously powerful template parser for `MediaWiki <http://mediawiki.org>`_ | |||||
wikicode. | |||||
""" | |||||
__author__ = "Ben Kurtovic" | |||||
__copyright__ = "Copyright (C) 2012 by Ben Kurtovic" | |||||
__license__ = "MIT License" | |||||
__version__ = "0.1.dev" | |||||
__email__ = "ben.kurtovic@verizon.net" | |||||
from mwtemplateparserfromhell import parameter, parser, template | |||||
from mwtemplateparserfromhell.parser import Parser |
@@ -0,0 +1,95 @@ | |||||
# -*- coding: utf-8 -*- | |||||
# | |||||
# Copyright (C) 2012 by Ben Kurtovic <ben.kurtovic@verizon.net> | |||||
# | |||||
# Permission is hereby granted, free of charge, to any person obtaining a copy | |||||
# of this software and associated documentation files (the "Software"), to deal | |||||
# in the Software without restriction, including without limitation the rights | |||||
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |||||
# copies of the Software, and to permit persons to whom the Software is | |||||
# furnished to do so, subject to the following conditions: | |||||
# | |||||
# The above copyright notice and this permission notice shall be included in | |||||
# all copies or substantial portions of the Software. | |||||
# | |||||
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |||||
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |||||
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | |||||
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |||||
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |||||
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | |||||
# SOFTWARE. | |||||
__all__ = ["Parameter"] | |||||
class Parameter(object): | |||||
def __init__(self, name, value, templates=None): | |||||
self._name = name | |||||
self._value = value | |||||
if templates: | |||||
self._templates = templates | |||||
else: | |||||
self._templates = [] | |||||
def __repr__(self): | |||||
return repr(self.value) | |||||
def __str__(self): | |||||
return self.value | |||||
def __lt__(self, other): | |||||
if isinstance(other, Parameter): | |||||
return self.value < other.value | |||||
return self.value < other | |||||
def __le__(self, other): | |||||
if isinstance(other, Parameter): | |||||
return self.value <= other.value | |||||
return self.value <= other | |||||
def __eq__(self, other): | |||||
if isinstance(other, Parameter): | |||||
return (self.value == other.value and | |||||
self.templates == other.templates) | |||||
return self.value == other | |||||
def __ne__(self, other): | |||||
if isinstance(other, Parameter): | |||||
return (self.value != other.value or | |||||
self.templates != other.templates) | |||||
return self.value != other | |||||
def __gt__(self, other): | |||||
if isinstance(other, Parameter): | |||||
return self.value > other.value | |||||
return self.value > other | |||||
def __ge__(self, other): | |||||
if isinstance(other, Parameter): | |||||
return self.value >= other.value | |||||
return self.value >= other | |||||
def __nonzero__(self): | |||||
return bool(self.value) | |||||
def __len__(self): | |||||
return len(self.value) | |||||
def __iter__(self): | |||||
for char in self.value: | |||||
yield char | |||||
def __contains__(self, item): | |||||
return item in self.value or item in self.templates | |||||
@property | |||||
def name(self): | |||||
return self._name | |||||
@property | |||||
def value(self): | |||||
return self._value | |||||
@property | |||||
def templates(self): | |||||
return self._templates |
@@ -0,0 +1,36 @@ | |||||
# -*- coding: utf-8 -*- | |||||
# | |||||
# Copyright (C) 2012 by Ben Kurtovic <ben.kurtovic@verizon.net> | |||||
# | |||||
# Permission is hereby granted, free of charge, to any person obtaining a copy | |||||
# of this software and associated documentation files (the "Software"), to deal | |||||
# in the Software without restriction, including without limitation the rights | |||||
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |||||
# copies of the Software, and to permit persons to whom the Software is | |||||
# furnished to do so, subject to the following conditions: | |||||
# | |||||
# The above copyright notice and this permission notice shall be included in | |||||
# all copies or substantial portions of the Software. | |||||
# | |||||
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |||||
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |||||
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | |||||
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |||||
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |||||
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | |||||
# SOFTWARE. | |||||
from mwtemplateparserfromhell.parameter import Parameter | |||||
from mwtemplateparserfromhell.template import Template | |||||
__all__ = ["Parser"] | |||||
class Parser(object): | |||||
def _tokenize(self, text): | |||||
return text | |||||
def parse(self, text): | |||||
tokens = self._tokenize(text) | |||||
params = [Parameter("1", "bar"), Parameter("2", "baz")] | |||||
templates = [Template(name="foo", params=params)] | |||||
return templates |
@@ -0,0 +1,64 @@ | |||||
# -*- coding: utf-8 -*- | |||||
# | |||||
# Copyright (C) 2012 by Ben Kurtovic <ben.kurtovic@verizon.net> | |||||
# | |||||
# Permission is hereby granted, free of charge, to any person obtaining a copy | |||||
# of this software and associated documentation files (the "Software"), to deal | |||||
# in the Software without restriction, including without limitation the rights | |||||
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |||||
# copies of the Software, and to permit persons to whom the Software is | |||||
# furnished to do so, subject to the following conditions: | |||||
# | |||||
# The above copyright notice and this permission notice shall be included in | |||||
# all copies or substantial portions of the Software. | |||||
# | |||||
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |||||
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |||||
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | |||||
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |||||
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |||||
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | |||||
# SOFTWARE. | |||||
from collections import OrderedDict | |||||
__all__ = ["Template"] | |||||
class Template(object): | |||||
def __init__(self, name, params=None): | |||||
self._name = name | |||||
self._params = OrderedDict() | |||||
if params: | |||||
for param in params: | |||||
self._params[param.name] = param | |||||
def __repr__(self): | |||||
paramlist = [] | |||||
for name, param in self._params.iteritems(): | |||||
paramlist.append('"{0}": "{1}"'.format(name, str(param))) | |||||
params = "{" + ", ".join(paramlist) + "}" | |||||
return "Template(name={0}, params={1})".format(self.name, params) | |||||
def __eq__(self, other): | |||||
if isinstance(other, Template): | |||||
return self.name == other.name and self._params == other._params | |||||
return False | |||||
def __ne__(self, other): | |||||
if isinstance(other, Template): | |||||
return self.name != other.name or self._params != other._params | |||||
return True | |||||
@property | |||||
def name(self): | |||||
return self._name | |||||
@property | |||||
def params(self): | |||||
return self._params.values() | |||||
def get(self, name): | |||||
try: | |||||
return self._params[name] | |||||
except KeyError: # Try lookup by order in param list | |||||
return self._params.values()[name] |
@@ -0,0 +1,53 @@ | |||||
#! /usr/bin/env python | |||||
# -*- coding: utf-8 -*- | |||||
# | |||||
# Copyright (C) 2012 by Ben Kurtovic <ben.kurtovic@verizon.net> | |||||
# | |||||
# Permission is hereby granted, free of charge, to any person obtaining a copy | |||||
# of this software and associated documentation files (the "Software"), to deal | |||||
# in the Software without restriction, including without limitation the rights | |||||
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |||||
# copies of the Software, and to permit persons to whom the Software is | |||||
# furnished to do so, subject to the following conditions: | |||||
# | |||||
# The above copyright notice and this permission notice shall be included in | |||||
# all copies or substantial portions of the Software. | |||||
# | |||||
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |||||
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |||||
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | |||||
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |||||
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |||||
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | |||||
# SOFTWARE. | |||||
from setuptools import setup, find_packages | |||||
from mwtemplateparserfromhell import __version__ | |||||
with open("README.rst") as fp: | |||||
long_docs = fp.read() | |||||
setup( | |||||
name = "mwtemplateparserfromhell", | |||||
packages = find_packages(exclude=("tests",)), | |||||
test_suite = "tests", | |||||
version = __version__, | |||||
author = "Ben Kurtovic", | |||||
author_email = "ben.kurtovic@verizon.net", | |||||
url = "https://github.com/earwig/mwtemplateparserfromhell", | |||||
description = "MWTemplateParserFromHell is a parser for MediaWiki templates.", | |||||
long_description = long_docs, | |||||
download_url = "https://github.com/earwig/mwtemplateparserfromhell/tarball/v{0}".format(__version__), | |||||
keywords = "earwig mwtemplateparserfromhell wikipedia wiki mediawiki template parsing", | |||||
license = "MIT License", | |||||
classifiers = [ | |||||
"Development Status :: 3 - Alpha", | |||||
"Environment :: Console", | |||||
"Intended Audience :: Developers", | |||||
"License :: OSI Approved :: MIT License", | |||||
"Operating System :: OS Independent", | |||||
"Programming Language :: Python", | |||||
"Topic :: Text Processing :: Markup" | |||||
], | |||||
) |
@@ -0,0 +1,99 @@ | |||||
# -*- coding: utf-8 -*- | |||||
# | |||||
# Copyright (C) 2012 by Ben Kurtovic <ben.kurtovic@verizon.net> | |||||
# | |||||
# Permission is hereby granted, free of charge, to any person obtaining a copy | |||||
# of this software and associated documentation files (the "Software"), to deal | |||||
# in the Software without restriction, including without limitation the rights | |||||
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |||||
# copies of the Software, and to permit persons to whom the Software is | |||||
# furnished to do so, subject to the following conditions: | |||||
# | |||||
# The above copyright notice and this permission notice shall be included in | |||||
# all copies or substantial portions of the Software. | |||||
# | |||||
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |||||
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |||||
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | |||||
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |||||
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |||||
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | |||||
# SOFTWARE. | |||||
import unittest | |||||
from mwtemplateparserfromhell.parameter import Parameter | |||||
from mwtemplateparserfromhell.template import Template | |||||
class TestParameter(unittest.TestCase): | |||||
def setUp(self): | |||||
self.name = "foo" | |||||
self.value1 = "bar" | |||||
self.value2 = "{{spam}}" | |||||
self.value3 = "bar{{spam}}" | |||||
self.value4 = "embedded {{eggs|spam|baz=buz}} {{goes}} here" | |||||
self.templates2 = [Template("spam")] | |||||
self.templates3 = [Template("spam")] | |||||
self.templates4 = [Template("eggs", [Parameter("1", "spam"), | |||||
Parameter("baz", "buz")]), | |||||
Template("goes")] | |||||
def test_construct(self): | |||||
Parameter(self.name, self.value1) | |||||
Parameter(self.name, self.value2, self.templates2) | |||||
Parameter(name=self.name, value=self.value3) | |||||
Parameter(name=self.name, value=self.value4, templates=self.templates4) | |||||
def test_name(self): | |||||
params = [ | |||||
Parameter(self.name, self.value1), | |||||
Parameter(self.name, self.value2, self.templates2), | |||||
Parameter(name=self.name, value=self.value3), | |||||
Parameter(name=self.name, value=self.value4, | |||||
templates=self.templates4) | |||||
] | |||||
for param in params: | |||||
self.assertEqual(param.name, self.name) | |||||
def test_value(self): | |||||
tests = [ | |||||
(Parameter(self.name, self.value1), self.value1), | |||||
(Parameter(self.name, self.value2, self.templates2), self.value2), | |||||
(Parameter(name=self.name, value=self.value3), self.value3), | |||||
(Parameter(name=self.name, value=self.value4, | |||||
templates=self.templates4), self.value4) | |||||
] | |||||
for param, correct in tests: | |||||
self.assertEqual(param.value, correct) | |||||
def test_templates(self): | |||||
tests = [ | |||||
(Parameter(self.name, self.value3, self.templates3), | |||||
self.templates3), | |||||
(Parameter(name=self.name, value=self.value4, | |||||
templates=self.templates4), self.templates4) | |||||
] | |||||
for param, correct in tests: | |||||
self.assertEqual(param.templates, correct) | |||||
def test_magic(self): | |||||
params = [Parameter(self.name, self.value1), | |||||
Parameter(self.name, self.value2, self.templates2), | |||||
Parameter(self.name, self.value3, self.templates3), | |||||
Parameter(self.name, self.value4, self.templates4)] | |||||
for param in params: | |||||
self.assertEqual(repr(param), repr(param.value)) | |||||
self.assertEqual(str(param), str(param.value)) | |||||
self.assertIs(param < "eggs", param.value < "eggs") | |||||
self.assertIs(param <= "bar{{spam}}", param.value <= "bar{{spam}}") | |||||
self.assertIs(param == "bar", param.value == "bar") | |||||
self.assertIs(param != "bar", param.value != "bar") | |||||
self.assertIs(param > "eggs", param.value > "eggs") | |||||
self.assertIs(param >= "bar{{spam}}", param.value >= "bar{{spam}}") | |||||
self.assertEquals(bool(param), bool(param.value)) | |||||
self.assertEquals(len(param), len(param.value)) | |||||
self.assertEquals(list(param), list(param.value)) | |||||
self.assertIs("bar" in param, "bar" in param.value) | |||||
if __name__ == "__main__": | |||||
unittest.main(verbosity=2) |
@@ -0,0 +1,58 @@ | |||||
# -*- coding: utf-8 -*- | |||||
# | |||||
# Copyright (C) 2012 by Ben Kurtovic <ben.kurtovic@verizon.net> | |||||
# | |||||
# Permission is hereby granted, free of charge, to any person obtaining a copy | |||||
# of this software and associated documentation files (the "Software"), to deal | |||||
# in the Software without restriction, including without limitation the rights | |||||
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |||||
# copies of the Software, and to permit persons to whom the Software is | |||||
# furnished to do so, subject to the following conditions: | |||||
# | |||||
# The above copyright notice and this permission notice shall be included in | |||||
# all copies or substantial portions of the Software. | |||||
# | |||||
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |||||
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |||||
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | |||||
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |||||
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |||||
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | |||||
# SOFTWARE. | |||||
import unittest | |||||
from mwtemplateparserfromhell.parameter import Parameter | |||||
from mwtemplateparserfromhell.parser import Parser | |||||
from mwtemplateparserfromhell.template import Template | |||||
TESTS = [ | |||||
("", []), | |||||
("abcdef ghijhk", []), | |||||
("abc{this is not a template}def", []), | |||||
("neither is {{this one}nor} {this one {despite}} containing braces", []), | |||||
("this is an acceptable {{template}}", [Template("template")]), | |||||
("{{multiple}}{{templates}}", [Template("multiple"), | |||||
Template("templates")]), | |||||
("multiple {{-}} templates {{+}}!", [Template("-"), Template("+")]), | |||||
("{{{no templates here}}}", []), | |||||
("{ {{templates here}}}", [Template("templates here")]), | |||||
("{{{{I exist}} }}", [Template("I exist")]), | |||||
("{{{{I do not exist}}}}", []), | |||||
("{{foo|bar|baz|eggs=spam}}", | |||||
[Template("foo", [Parameter("1", "bar"), Parameter("2", "baz"), | |||||
Parameter("eggs", "spam")])]), | |||||
("{{abc def|ghi|jk=lmno|pqr|st=uv|wx|yz}}", | |||||
[Template("abc def", [Parameter("1", "ghi"), Parameter("jk", "lmno"), | |||||
Parameter("2", "pqr"), Parameter("st", "uv"), | |||||
Parameter("3", "wx"), Parameter("4", "yz")])]), | |||||
] | |||||
class TestParser(unittest.TestCase): | |||||
def test_parse(self): | |||||
parser = Parser() | |||||
for unparsed, parsed in TESTS: | |||||
self.assertEqual(parser.parse(unparsed), parsed) | |||||
if __name__ == "__main__": | |||||
unittest.main(verbosity=2) |
@@ -0,0 +1,98 @@ | |||||
# -*- coding: utf-8 -*- | |||||
# | |||||
# Copyright (C) 2012 by Ben Kurtovic <ben.kurtovic@verizon.net> | |||||
# | |||||
# Permission is hereby granted, free of charge, to any person obtaining a copy | |||||
# of this software and associated documentation files (the "Software"), to deal | |||||
# in the Software without restriction, including without limitation the rights | |||||
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |||||
# copies of the Software, and to permit persons to whom the Software is | |||||
# furnished to do so, subject to the following conditions: | |||||
# | |||||
# The above copyright notice and this permission notice shall be included in | |||||
# all copies or substantial portions of the Software. | |||||
# | |||||
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |||||
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |||||
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | |||||
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |||||
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |||||
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | |||||
# SOFTWARE. | |||||
from itertools import permutations | |||||
import unittest | |||||
from mwtemplateparserfromhell.parameter import Parameter | |||||
from mwtemplateparserfromhell.template import Template | |||||
class TestTemplate(unittest.TestCase): | |||||
def setUp(self): | |||||
self.name = "foo" | |||||
self.bar = Parameter("1", "bar") | |||||
self.baz = Parameter("2", "baz") | |||||
self.eggs = Parameter("eggs", "spam") | |||||
self.params = [self.bar, self.baz, self.eggs] | |||||
def test_construct(self): | |||||
Template(self.name) | |||||
Template(self.name, self.params) | |||||
Template(name=self.name) | |||||
Template(name=self.name, params=self.params) | |||||
def test_name(self): | |||||
templates = [ | |||||
Template(self.name), | |||||
Template(self.name, self.params), | |||||
Template(name=self.name), | |||||
Template(name=self.name, params=self.params) | |||||
] | |||||
for template in templates: | |||||
self.assertEqual(template.name, self.name) | |||||
def test_params(self): | |||||
for template in (Template(self.name), Template(name=self.name)): | |||||
self.assertEqual(template.params, []) | |||||
for template in (Template(self.name, self.params), | |||||
Template(name=self.name, params=self.params)): | |||||
self.assertEqual(template.params, self.params) | |||||
def test_get(self): | |||||
template = Template(name=self.name, params=self.params) | |||||
self.assertIs(template.get(0), self.bar) | |||||
self.assertIs(template.get(1), self.baz) | |||||
self.assertIs(template.get(2), self.eggs) | |||||
self.assertIs(template.get("1"), self.bar) | |||||
self.assertIs(template.get("2"), self.baz) | |||||
self.assertIs(template.get("eggs"), self.eggs) | |||||
def test_repr(self): | |||||
correct1= 'Template(name=foo, params={})' | |||||
correct2 = 'Template(name=foo, params={"1": "bar", "2": "baz", "eggs": "spam"})' | |||||
tests = [(Template(self.name), correct1), | |||||
(Template(self.name, self.params), correct2)] | |||||
for template, correct in tests: | |||||
self.assertEqual(repr(template), correct) | |||||
self.assertEqual(str(template), correct) | |||||
def test_cmp(self): | |||||
tmp1 = Template(self.name) | |||||
tmp2 = Template(name=self.name) | |||||
tmp3 = Template(self.name, []) | |||||
tmp4 = Template(name=self.name, params=[]) | |||||
tmp5 = Template(self.name, self.params) | |||||
tmp6 = Template(name=self.name, params=self.params) | |||||
for tmpA, tmpB in permutations((tmp1, tmp2, tmp3, tmp4), 2): | |||||
self.assertEqual(tmpA, tmpB) | |||||
for tmpA, tmpB in permutations((tmp5, tmp6), 2): | |||||
self.assertEqual(tmpA, tmpB) | |||||
for tmpA in (tmp5, tmp6): | |||||
for tmpB in (tmp1, tmp2, tmp3, tmp4): | |||||
self.assertNotEqual(tmpA, tmpB) | |||||
self.assertNotEqual(tmpB, tmpA) | |||||
if __name__ == "__main__": | |||||
unittest.main(verbosity=2) |