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.
 
 
 
 

73 rivejä
2.4 KiB

  1. # -*- coding: utf-8 -*-
  2. #
  3. # Copyright (C) 2012-2015 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 __children__(self):
  36. yield self.title
  37. def __strip__(self, normalize, collapse):
  38. return self.title.strip_code(normalize, collapse)
  39. def __showtree__(self, write, get, mark):
  40. write("=" * self.level)
  41. get(self.title)
  42. write("=" * self.level)
  43. @property
  44. def title(self):
  45. """The title of the heading, as a :class:`.Wikicode` object."""
  46. return self._title
  47. @property
  48. def level(self):
  49. """The heading level, as an integer between 1 and 6, inclusive."""
  50. return self._level
  51. @title.setter
  52. def title(self, value):
  53. self._title = parse_anything(value)
  54. @level.setter
  55. def level(self, value):
  56. value = int(value)
  57. if value < 1 or value > 6:
  58. raise ValueError(value)
  59. self._level = value