A Python parser for MediaWiki wikicode https://mwparserfromhell.readthedocs.io/
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 

75 lines
2.5 KiB

  1. # -*- coding: utf-8 -*-
  2. #
  3. # Copyright (C) 2012-2013 Ben Kurtovic <ben.kurtovic@gmail.com>
  4. #
  5. # Permission is hereby granted, free of charge, to any person obtaining a copy
  6. # of this software and associated documentation files (the "Software"), to deal
  7. # in the Software without restriction, including without limitation the rights
  8. # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  9. # copies of the Software, and to permit persons to whom the Software is
  10. # furnished to do so, subject to the following conditions:
  11. #
  12. # The above copyright notice and this permission notice shall be included in
  13. # all copies or substantial portions of the Software.
  14. #
  15. # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16. # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17. # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  18. # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19. # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20. # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  21. # SOFTWARE.
  22. from __future__ import unicode_literals
  23. from . import Node
  24. from ..compat import str
  25. from ..utils import parse_anything
  26. __all__ = ["Heading"]
  27. class Heading(Node):
  28. """Represents a section heading in wikicode, like ``== Foo ==``."""
  29. def __init__(self, title, level):
  30. super(Heading, self).__init__()
  31. self._title = title
  32. self._level = level
  33. def __unicode__(self):
  34. return ("=" * self.level) + str(self.title) + ("=" * self.level)
  35. def __iternodes__(self, getter):
  36. yield None, self
  37. for child in getter(self.title):
  38. yield self.title, child
  39. def __strip__(self, normalize, collapse):
  40. return self.title.strip_code(normalize, collapse)
  41. def __showtree__(self, write, get, mark):
  42. write("=" * self.level)
  43. get(self.title)
  44. write("=" * self.level)
  45. @property
  46. def title(self):
  47. """The title of the heading, as a :py:class:`~.Wikicode` object."""
  48. return self._title
  49. @property
  50. def level(self):
  51. """The heading level, as an integer between 1 and 6, inclusive."""
  52. return self._level
  53. @title.setter
  54. def title(self, value):
  55. self._title = parse_anything(value)
  56. @level.setter
  57. def level(self, value):
  58. value = int(value)
  59. if value < 1 or value > 6:
  60. raise ValueError(value)
  61. self._level = value