| @@ -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) | |||||