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.
 
 
 
 

69 line
2.8 KiB

  1. # -*- coding: utf-8 -*-
  2. #
  3. # Copyright (C) 2012 Ben Kurtovic <ben.kurtovic@verizon.net>
  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. """
  23. This module contains accessory functions that wrap around existing ones to
  24. provide additional functionality.
  25. """
  26. from __future__ import unicode_literals
  27. from .compat import bytes, str
  28. from .nodes import Node
  29. from .smart_list import SmartList
  30. def parse_anything(value):
  31. """Return a :py:class:`~.Wikicode` for *value*, allowing multiple types.
  32. This differs from :py:func:`mwparserfromhell.parse` in that we accept more
  33. than just a string to be parsed. Unicode objects (strings in py3k), strings
  34. (bytes in py3k), integers (converted to strings), ``None``, existing
  35. :py:class:`~.Node` or :py:class:`~.Wikicode` objects, as well as an
  36. iterable of these types, are supported. This is used to parse input
  37. on-the-fly by various methods of :py:class:`~.Wikicode` and others like
  38. :py:class:`~.Template`, such as :py:meth:`wikicode.insert()
  39. <.Wikicode.insert>` or setting :py:meth:`template.name <.Template.name>`.
  40. """
  41. from . import parse
  42. from .wikicode import Wikicode
  43. if isinstance(value, Wikicode):
  44. return value
  45. elif isinstance(value, Node):
  46. return Wikicode(SmartList([value]))
  47. elif isinstance(value, str):
  48. return parse(value)
  49. elif isinstance(value, bytes):
  50. return parse(value.decode("utf8"))
  51. elif isinstance(value, int):
  52. return parse(str(value))
  53. elif value is None:
  54. return Wikicode(SmartList())
  55. try:
  56. nodelist = SmartList()
  57. for item in value:
  58. nodelist += parse_anything(item).nodes
  59. except TypeError:
  60. error = "Needs string, Node, Wikicode, int, None, or iterable of these, but got {0}: {1}"
  61. raise ValueError(error.format(type(value).__name__, value))
  62. return Wikicode(nodelist)