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.
 
 
 
 

970 lines
36 KiB

  1. # -*- coding: utf-8 -*-
  2. #
  3. # Copyright (C) 2012-2013 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. from __future__ import unicode_literals
  23. from math import log
  24. import re
  25. from . import contexts, tokens
  26. from ..compat import htmlentities
  27. from ..tag_defs import get_html_tag, is_parsable, is_single, is_single_only
  28. __all__ = ["Tokenizer"]
  29. class BadRoute(Exception):
  30. """Raised internally when the current tokenization route is invalid."""
  31. def __init__(self, context=0):
  32. self.context = context
  33. class _TagOpenData(object):
  34. """Stores data about an HTML open tag, like ``<ref name="foo">``."""
  35. CX_NAME = 1 << 0
  36. CX_ATTR_READY = 1 << 1
  37. CX_ATTR_NAME = 1 << 2
  38. CX_ATTR_VALUE = 1 << 3
  39. CX_QUOTED = 1 << 4
  40. CX_NOTE_SPACE = 1 << 5
  41. CX_NOTE_EQUALS = 1 << 6
  42. CX_NOTE_QUOTE = 1 << 7
  43. def __init__(self):
  44. self.context = self.CX_NAME
  45. self.padding_buffer = {"first": "", "before_eq": "", "after_eq": ""}
  46. self.reset = 0
  47. class Tokenizer(object):
  48. """Creates a list of tokens from a string of wikicode."""
  49. USES_C = False
  50. START = object()
  51. END = object()
  52. MARKERS = ["{", "}", "[", "]", "<", ">", "|", "=", "&", "'", "#", "*", ";",
  53. ":", "/", "-", "\n", END]
  54. MAX_DEPTH = 40
  55. MAX_CYCLES = 100000
  56. regex = re.compile(r"([{}\[\]<>|=&'#*;:/\\\"\-!\n])", flags=re.IGNORECASE)
  57. tag_splitter = re.compile(r"([\s\"\\]+)")
  58. def __init__(self):
  59. self._text = None
  60. self._head = 0
  61. self._stacks = []
  62. self._global = 0
  63. self._depth = 0
  64. self._cycles = 0
  65. @property
  66. def _stack(self):
  67. """The current token stack."""
  68. return self._stacks[-1][0]
  69. @property
  70. def _context(self):
  71. """The current token context."""
  72. return self._stacks[-1][1]
  73. @_context.setter
  74. def _context(self, value):
  75. self._stacks[-1][1] = value
  76. @property
  77. def _textbuffer(self):
  78. """The current textbuffer."""
  79. return self._stacks[-1][2]
  80. @_textbuffer.setter
  81. def _textbuffer(self, value):
  82. self._stacks[-1][2] = value
  83. def _push(self, context=0):
  84. """Add a new token stack, context, and textbuffer to the list."""
  85. self._stacks.append([[], context, []])
  86. self._depth += 1
  87. self._cycles += 1
  88. def _push_textbuffer(self):
  89. """Push the textbuffer onto the stack as a Text node and clear it."""
  90. if self._textbuffer:
  91. self._stack.append(tokens.Text(text="".join(self._textbuffer)))
  92. self._textbuffer = []
  93. def _pop(self, keep_context=False):
  94. """Pop the current stack/context/textbuffer, returing the stack.
  95. If *keep_context* is ``True``, then we will replace the underlying
  96. stack's context with the current stack's.
  97. """
  98. self._push_textbuffer()
  99. self._depth -= 1
  100. if keep_context:
  101. context = self._context
  102. stack = self._stacks.pop()[0]
  103. self._context = context
  104. return stack
  105. return self._stacks.pop()[0]
  106. def _can_recurse(self):
  107. """Return whether or not our max recursion depth has been exceeded."""
  108. return self._depth < self.MAX_DEPTH and self._cycles < self.MAX_CYCLES
  109. def _fail_route(self):
  110. """Fail the current tokenization route.
  111. Discards the current stack/context/textbuffer and raises
  112. :py:exc:`~.BadRoute`.
  113. """
  114. context = self._context
  115. self._pop()
  116. raise BadRoute(context)
  117. def _emit(self, token):
  118. """Write a token to the end of the current token stack."""
  119. self._push_textbuffer()
  120. self._stack.append(token)
  121. def _emit_first(self, token):
  122. """Write a token to the beginning of the current token stack."""
  123. self._push_textbuffer()
  124. self._stack.insert(0, token)
  125. def _emit_text(self, text):
  126. """Write text to the current textbuffer."""
  127. self._textbuffer.append(text)
  128. def _emit_all(self, tokenlist):
  129. """Write a series of tokens to the current stack at once."""
  130. if tokenlist and isinstance(tokenlist[0], tokens.Text):
  131. self._emit_text(tokenlist.pop(0).text)
  132. self._push_textbuffer()
  133. self._stack.extend(tokenlist)
  134. def _emit_text_then_stack(self, text):
  135. """Pop the current stack, write *text*, and then write the stack."""
  136. stack = self._pop()
  137. self._emit_text(text)
  138. if stack:
  139. self._emit_all(stack)
  140. self._head -= 1
  141. def _read(self, delta=0, wrap=False, strict=False):
  142. """Read the value at a relative point in the wikicode.
  143. The value is read from :py:attr:`self._head <_head>` plus the value of
  144. *delta* (which can be negative). If *wrap* is ``False``, we will not
  145. allow attempts to read from the end of the string if ``self._head +
  146. delta`` is negative. If *strict* is ``True``, the route will be failed
  147. (with :py:meth:`_fail_route`) if we try to read from past the end of
  148. the string; otherwise, :py:attr:`self.END <END>` is returned. If we try
  149. to read from before the start of the string, :py:attr:`self.START
  150. <START>` is returned.
  151. """
  152. index = self._head + delta
  153. if index < 0 and (not wrap or abs(index) > len(self._text)):
  154. return self.START
  155. try:
  156. return self._text[index]
  157. except IndexError:
  158. if strict:
  159. self._fail_route()
  160. return self.END
  161. def _parse_template(self):
  162. """Parse a template at the head of the wikicode string."""
  163. reset = self._head
  164. try:
  165. template = self._parse(contexts.TEMPLATE_NAME)
  166. except BadRoute:
  167. self._head = reset
  168. raise
  169. self._emit_first(tokens.TemplateOpen())
  170. self._emit_all(template)
  171. self._emit(tokens.TemplateClose())
  172. def _parse_argument(self):
  173. """Parse an argument at the head of the wikicode string."""
  174. reset = self._head
  175. try:
  176. argument = self._parse(contexts.ARGUMENT_NAME)
  177. except BadRoute:
  178. self._head = reset
  179. raise
  180. self._emit_first(tokens.ArgumentOpen())
  181. self._emit_all(argument)
  182. self._emit(tokens.ArgumentClose())
  183. def _parse_template_or_argument(self):
  184. """Parse a template or argument at the head of the wikicode string."""
  185. self._head += 2
  186. braces = 2
  187. while self._read() == "{":
  188. self._head += 1
  189. braces += 1
  190. self._push()
  191. while braces:
  192. if braces == 1:
  193. return self._emit_text_then_stack("{")
  194. if braces == 2:
  195. try:
  196. self._parse_template()
  197. except BadRoute:
  198. return self._emit_text_then_stack("{{")
  199. break
  200. try:
  201. self._parse_argument()
  202. braces -= 3
  203. except BadRoute:
  204. try:
  205. self._parse_template()
  206. braces -= 2
  207. except BadRoute:
  208. return self._emit_text_then_stack("{" * braces)
  209. if braces:
  210. self._head += 1
  211. self._emit_all(self._pop())
  212. if self._context & contexts.FAIL_NEXT:
  213. self._context ^= contexts.FAIL_NEXT
  214. def _handle_template_param(self):
  215. """Handle a template parameter at the head of the string."""
  216. if self._context & contexts.TEMPLATE_NAME:
  217. self._context ^= contexts.TEMPLATE_NAME
  218. elif self._context & contexts.TEMPLATE_PARAM_VALUE:
  219. self._context ^= contexts.TEMPLATE_PARAM_VALUE
  220. elif self._context & contexts.TEMPLATE_PARAM_KEY:
  221. self._emit_all(self._pop(keep_context=True))
  222. self._context |= contexts.TEMPLATE_PARAM_KEY
  223. self._emit(tokens.TemplateParamSeparator())
  224. self._push(self._context)
  225. def _handle_template_param_value(self):
  226. """Handle a template parameter's value at the head of the string."""
  227. self._emit_all(self._pop(keep_context=True))
  228. self._context ^= contexts.TEMPLATE_PARAM_KEY
  229. self._context |= contexts.TEMPLATE_PARAM_VALUE
  230. self._emit(tokens.TemplateParamEquals())
  231. def _handle_template_end(self):
  232. """Handle the end of a template at the head of the string."""
  233. if self._context & contexts.TEMPLATE_PARAM_KEY:
  234. self._emit_all(self._pop(keep_context=True))
  235. self._head += 1
  236. return self._pop()
  237. def _handle_argument_separator(self):
  238. """Handle the separator between an argument's name and default."""
  239. self._context ^= contexts.ARGUMENT_NAME
  240. self._context |= contexts.ARGUMENT_DEFAULT
  241. self._emit(tokens.ArgumentSeparator())
  242. def _handle_argument_end(self):
  243. """Handle the end of an argument at the head of the string."""
  244. self._head += 2
  245. return self._pop()
  246. def _parse_wikilink(self):
  247. """Parse an internal wikilink at the head of the wikicode string."""
  248. self._head += 2
  249. reset = self._head - 1
  250. try:
  251. wikilink = self._parse(contexts.WIKILINK_TITLE)
  252. except BadRoute:
  253. self._head = reset
  254. self._emit_text("[[")
  255. else:
  256. if self._context & contexts.FAIL_NEXT:
  257. self._context ^= contexts.FAIL_NEXT
  258. self._emit(tokens.WikilinkOpen())
  259. self._emit_all(wikilink)
  260. self._emit(tokens.WikilinkClose())
  261. def _handle_wikilink_separator(self):
  262. """Handle the separator between a wikilink's title and its text."""
  263. self._context ^= contexts.WIKILINK_TITLE
  264. self._context |= contexts.WIKILINK_TEXT
  265. self._emit(tokens.WikilinkSeparator())
  266. def _handle_wikilink_end(self):
  267. """Handle the end of a wikilink at the head of the string."""
  268. self._head += 1
  269. return self._pop()
  270. def _parse_external_link(self, brackets):
  271. """Parse an external link at the head of the wikicode string."""
  272. self._emit_text(self._read())
  273. # raise NotImplementedError()
  274. def _parse_heading(self):
  275. """Parse a section heading at the head of the wikicode string."""
  276. self._global |= contexts.GL_HEADING
  277. reset = self._head
  278. self._head += 1
  279. best = 1
  280. while self._read() == "=":
  281. best += 1
  282. self._head += 1
  283. context = contexts.HEADING_LEVEL_1 << min(best - 1, 5)
  284. try:
  285. title, level = self._parse(context)
  286. except BadRoute:
  287. self._head = reset + best - 1
  288. self._emit_text("=" * best)
  289. else:
  290. self._emit(tokens.HeadingStart(level=level))
  291. if level < best:
  292. self._emit_text("=" * (best - level))
  293. self._emit_all(title)
  294. self._emit(tokens.HeadingEnd())
  295. finally:
  296. self._global ^= contexts.GL_HEADING
  297. def _handle_heading_end(self):
  298. """Handle the end of a section heading at the head of the string."""
  299. reset = self._head
  300. self._head += 1
  301. best = 1
  302. while self._read() == "=":
  303. best += 1
  304. self._head += 1
  305. current = int(log(self._context / contexts.HEADING_LEVEL_1, 2)) + 1
  306. level = min(current, min(best, 6))
  307. try: # Try to check for a heading closure after this one
  308. after, after_level = self._parse(self._context)
  309. except BadRoute:
  310. if level < best:
  311. self._emit_text("=" * (best - level))
  312. self._head = reset + best - 1
  313. return self._pop(), level
  314. else: # Found another closure
  315. self._emit_text("=" * best)
  316. self._emit_all(after)
  317. return self._pop(), after_level
  318. def _really_parse_entity(self):
  319. """Actually parse an HTML entity and ensure that it is valid."""
  320. self._emit(tokens.HTMLEntityStart())
  321. self._head += 1
  322. this = self._read(strict=True)
  323. if this == "#":
  324. numeric = True
  325. self._emit(tokens.HTMLEntityNumeric())
  326. self._head += 1
  327. this = self._read(strict=True)
  328. if this[0].lower() == "x":
  329. hexadecimal = True
  330. self._emit(tokens.HTMLEntityHex(char=this[0]))
  331. this = this[1:]
  332. if not this:
  333. self._fail_route()
  334. else:
  335. hexadecimal = False
  336. else:
  337. numeric = hexadecimal = False
  338. valid = "0123456789abcdefABCDEF" if hexadecimal else "0123456789"
  339. if not numeric and not hexadecimal:
  340. valid += "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
  341. if not all([char in valid for char in this]):
  342. self._fail_route()
  343. self._head += 1
  344. if self._read() != ";":
  345. self._fail_route()
  346. if numeric:
  347. test = int(this, 16) if hexadecimal else int(this)
  348. if test < 1 or test > 0x10FFFF:
  349. self._fail_route()
  350. else:
  351. if this not in htmlentities.entitydefs:
  352. self._fail_route()
  353. self._emit(tokens.Text(text=this))
  354. self._emit(tokens.HTMLEntityEnd())
  355. def _parse_entity(self):
  356. """Parse an HTML entity at the head of the wikicode string."""
  357. reset = self._head
  358. self._push()
  359. try:
  360. self._really_parse_entity()
  361. except BadRoute:
  362. self._head = reset
  363. self._emit_text(self._read())
  364. else:
  365. self._emit_all(self._pop())
  366. def _parse_comment(self):
  367. """Parse an HTML comment at the head of the wikicode string."""
  368. self._head += 4
  369. reset = self._head - 1
  370. self._push()
  371. while True:
  372. this = self._read()
  373. if this == self.END:
  374. self._pop()
  375. self._head = reset
  376. self._emit_text("<!--")
  377. return
  378. if this == self._read(1) == "-" and self._read(2) == ">":
  379. self._emit_first(tokens.CommentStart())
  380. self._emit(tokens.CommentEnd())
  381. self._emit_all(self._pop())
  382. self._head += 2
  383. return
  384. self._emit_text(this)
  385. self._head += 1
  386. def _push_tag_buffer(self, data):
  387. """Write a pending tag attribute from *data* to the stack."""
  388. if data.context & data.CX_QUOTED:
  389. self._emit_first(tokens.TagAttrQuote())
  390. self._emit_all(self._pop())
  391. buf = data.padding_buffer
  392. self._emit_first(tokens.TagAttrStart(pad_first=buf["first"],
  393. pad_before_eq=buf["before_eq"], pad_after_eq=buf["after_eq"]))
  394. self._emit_all(self._pop())
  395. data.padding_buffer = {key: "" for key in data.padding_buffer}
  396. def _handle_tag_space(self, data, text):
  397. """Handle whitespace (*text*) inside of an HTML open tag."""
  398. ctx = data.context
  399. end_of_value = ctx & data.CX_ATTR_VALUE and not ctx & (data.CX_QUOTED | data.CX_NOTE_QUOTE)
  400. if end_of_value or (ctx & data.CX_QUOTED and ctx & data.CX_NOTE_SPACE):
  401. self._push_tag_buffer(data)
  402. data.context = data.CX_ATTR_READY
  403. elif ctx & data.CX_NOTE_SPACE:
  404. data.context = data.CX_ATTR_READY
  405. elif ctx & data.CX_ATTR_NAME:
  406. data.context |= data.CX_NOTE_EQUALS
  407. data.padding_buffer["before_eq"] += text
  408. if ctx & data.CX_QUOTED and not ctx & data.CX_NOTE_SPACE:
  409. self._emit_text(text)
  410. elif data.context & data.CX_ATTR_READY:
  411. data.padding_buffer["first"] += text
  412. elif data.context & data.CX_ATTR_VALUE:
  413. data.padding_buffer["after_eq"] += text
  414. def _handle_tag_text(self, text):
  415. """Handle regular *text* inside of an HTML open tag."""
  416. next = self._read(1)
  417. if not self._can_recurse() or text not in self.MARKERS:
  418. self._emit_text(text)
  419. elif text == next == "{":
  420. self._parse_template_or_argument()
  421. elif text == next == "[":
  422. self._parse_wikilink()
  423. elif text == "<":
  424. self._parse_tag()
  425. else:
  426. self._emit_text(text)
  427. def _handle_tag_data(self, data, text):
  428. """Handle all sorts of *text* data inside of an HTML open tag."""
  429. for chunk in self.tag_splitter.split(text):
  430. if not chunk:
  431. continue
  432. if data.context & data.CX_NAME:
  433. if chunk in self.MARKERS or chunk.isspace():
  434. self._fail_route() # Tags must start with text, not spaces
  435. data.context = data.CX_NOTE_SPACE
  436. elif chunk.isspace():
  437. self._handle_tag_space(data, chunk)
  438. continue
  439. elif data.context & data.CX_NOTE_SPACE:
  440. if data.context & data.CX_QUOTED:
  441. data.context = data.CX_ATTR_VALUE
  442. self._pop()
  443. self._head = data.reset - 1 # Will be auto-incremented
  444. return # Break early
  445. self._fail_route()
  446. elif data.context & data.CX_ATTR_READY:
  447. data.context = data.CX_ATTR_NAME
  448. self._push(contexts.TAG_ATTR)
  449. elif data.context & data.CX_ATTR_NAME:
  450. if chunk == "=":
  451. data.context = data.CX_ATTR_VALUE | data.CX_NOTE_QUOTE
  452. self._emit(tokens.TagAttrEquals())
  453. continue
  454. if data.context & data.CX_NOTE_EQUALS:
  455. self._push_tag_buffer(data)
  456. data.context = data.CX_ATTR_NAME
  457. self._push(contexts.TAG_ATTR)
  458. elif data.context & data.CX_ATTR_VALUE:
  459. escaped = self._read(-1) == "\\" and self._read(-2) != "\\"
  460. if data.context & data.CX_NOTE_QUOTE:
  461. data.context ^= data.CX_NOTE_QUOTE
  462. if chunk == '"' and not escaped:
  463. data.context |= data.CX_QUOTED
  464. self._push(self._context)
  465. data.reset = self._head
  466. continue
  467. elif data.context & data.CX_QUOTED:
  468. if chunk == '"' and not escaped:
  469. data.context |= data.CX_NOTE_SPACE
  470. continue
  471. self._handle_tag_text(chunk)
  472. def _handle_tag_close_open(self, data, token):
  473. """Handle the closing of a open tag (``<foo>``)."""
  474. if data.context & (data.CX_ATTR_NAME | data.CX_ATTR_VALUE):
  475. self._push_tag_buffer(data)
  476. self._emit(token(padding=data.padding_buffer["first"]))
  477. self._head += 1
  478. def _handle_tag_open_close(self):
  479. """Handle the opening of a closing tag (``</foo>``)."""
  480. self._emit(tokens.TagOpenClose())
  481. self._push(contexts.TAG_CLOSE)
  482. self._head += 1
  483. def _handle_tag_close_close(self):
  484. """Handle the ending of a closing tag (``</foo>``)."""
  485. strip = lambda tok: tok.text.rstrip().lower()
  486. closing = self._pop()
  487. if len(closing) != 1 or (not isinstance(closing[0], tokens.Text) or
  488. strip(closing[0]) != strip(self._stack[1])):
  489. self._fail_route()
  490. self._emit_all(closing)
  491. self._emit(tokens.TagCloseClose())
  492. return self._pop()
  493. def _handle_blacklisted_tag(self):
  494. """Handle the body of an HTML tag that is parser-blacklisted."""
  495. while True:
  496. this, next = self._read(), self._read(1)
  497. if this is self.END:
  498. self._fail_route()
  499. elif this == "<" and next == "/":
  500. self._handle_tag_open_close()
  501. self._head += 1
  502. return self._parse(push=False)
  503. elif this == "&":
  504. self._parse_entity()
  505. else:
  506. self._emit_text(this)
  507. self._head += 1
  508. def _handle_single_only_tag_end(self):
  509. """Handle the end of an implicitly closing single-only HTML tag."""
  510. padding = self._stack.pop().padding
  511. self._emit(tokens.TagCloseSelfclose(padding=padding, implicit=True))
  512. self._head -= 1 # Offset displacement done by _handle_tag_close_open
  513. return self._pop()
  514. def _handle_single_tag_end(self):
  515. """Handle the stream end when inside a single-supporting HTML tag."""
  516. gen = enumerate(self._stack)
  517. index = next(i for i, t in gen if isinstance(t, tokens.TagCloseOpen))
  518. padding = self._stack[index].padding
  519. token = tokens.TagCloseSelfclose(padding=padding, implicit=True)
  520. self._stack[index] = token
  521. return self._pop()
  522. def _really_parse_tag(self):
  523. """Actually parse an HTML tag, starting with the open (``<foo>``)."""
  524. data = _TagOpenData()
  525. self._push(contexts.TAG_OPEN)
  526. self._emit(tokens.TagOpenOpen())
  527. while True:
  528. this, next = self._read(), self._read(1)
  529. can_exit = (not data.context & (data.CX_QUOTED | data.CX_NAME) or
  530. data.context & data.CX_NOTE_SPACE)
  531. if this is self.END:
  532. if self._context & contexts.TAG_ATTR:
  533. if data.context & data.CX_QUOTED:
  534. # Unclosed attribute quote: reset, don't die
  535. data.context = data.CX_ATTR_VALUE
  536. self._pop()
  537. self._head = data.reset
  538. continue
  539. self._pop()
  540. self._fail_route()
  541. elif this == ">" and can_exit:
  542. self._handle_tag_close_open(data, tokens.TagCloseOpen)
  543. self._context = contexts.TAG_BODY
  544. if is_single_only(self._stack[1].text):
  545. return self._handle_single_only_tag_end()
  546. if is_parsable(self._stack[1].text):
  547. return self._parse(push=False)
  548. return self._handle_blacklisted_tag()
  549. elif this == "/" and next == ">" and can_exit:
  550. self._handle_tag_close_open(data, tokens.TagCloseSelfclose)
  551. return self._pop()
  552. else:
  553. self._handle_tag_data(data, this)
  554. self._head += 1
  555. def _handle_invalid_tag_start(self):
  556. """Handle the (possible) start of an implicitly closing single tag."""
  557. reset = self._head + 1
  558. self._head += 2
  559. try:
  560. if not is_single_only(self.tag_splitter.split(self._read())[0]):
  561. raise BadRoute()
  562. tag = self._really_parse_tag()
  563. except BadRoute:
  564. self._head = reset
  565. self._emit_text("</")
  566. else:
  567. tag[0].invalid = True # Set flag of TagOpenOpen
  568. self._emit_all(tag)
  569. def _parse_tag(self):
  570. """Parse an HTML tag at the head of the wikicode string."""
  571. reset = self._head
  572. self._head += 1
  573. try:
  574. tag = self._really_parse_tag()
  575. except BadRoute:
  576. self._head = reset
  577. self._emit_text("<")
  578. else:
  579. self._emit_all(tag)
  580. def _emit_style_tag(self, tag, markup, body):
  581. """Write the body of a tag and the tokens that should surround it."""
  582. self._emit(tokens.TagOpenOpen(wiki_markup=markup))
  583. self._emit_text(tag)
  584. self._emit(tokens.TagCloseOpen())
  585. self._emit_all(body)
  586. self._emit(tokens.TagOpenClose())
  587. self._emit_text(tag)
  588. self._emit(tokens.TagCloseClose())
  589. def _parse_italics(self):
  590. """Parse wiki-style italics."""
  591. reset = self._head
  592. try:
  593. stack = self._parse(contexts.STYLE_ITALICS)
  594. except BadRoute as route:
  595. self._head = reset
  596. if route.context & contexts.STYLE_PASS_AGAIN:
  597. stack = self._parse(route.context | contexts.STYLE_SECOND_PASS)
  598. else:
  599. return self._emit_text("''")
  600. self._emit_style_tag("i", "''", stack)
  601. def _parse_bold(self):
  602. """Parse wiki-style bold."""
  603. reset = self._head
  604. try:
  605. stack = self._parse(contexts.STYLE_BOLD)
  606. except BadRoute:
  607. self._head = reset
  608. if self._context & contexts.STYLE_SECOND_PASS:
  609. self._emit_text("'")
  610. return True
  611. elif self._context & contexts.STYLE_ITALICS:
  612. self._context |= contexts.STYLE_PASS_AGAIN
  613. self._emit_text("'''")
  614. else:
  615. self._emit_text("'")
  616. self._parse_italics()
  617. else:
  618. self._emit_style_tag("b", "'''", stack)
  619. def _parse_italics_and_bold(self):
  620. """Parse wiki-style italics and bold together (i.e., five ticks)."""
  621. reset = self._head
  622. try:
  623. stack = self._parse(contexts.STYLE_BOLD)
  624. except BadRoute:
  625. self._head = reset
  626. try:
  627. stack = self._parse(contexts.STYLE_ITALICS)
  628. except BadRoute:
  629. self._head = reset
  630. self._emit_text("'''''")
  631. else:
  632. reset = self._head
  633. try:
  634. stack2 = self._parse(contexts.STYLE_BOLD)
  635. except BadRoute:
  636. self._head = reset
  637. self._emit_text("'''")
  638. self._emit_style_tag("i", "''", stack)
  639. else:
  640. self._push()
  641. self._emit_style_tag("i", "''", stack)
  642. self._emit_all(stack2)
  643. self._emit_style_tag("b", "'''", self._pop())
  644. else:
  645. reset = self._head
  646. try:
  647. stack2 = self._parse(contexts.STYLE_ITALICS)
  648. except BadRoute:
  649. self._head = reset
  650. self._emit_text("''")
  651. self._emit_style_tag("b", "'''", stack)
  652. else:
  653. self._push()
  654. self._emit_style_tag("b", "'''", stack)
  655. self._emit_all(stack2)
  656. self._emit_style_tag("i", "''", self._pop())
  657. def _parse_style(self):
  658. """Parse wiki-style formatting (``''``/``'''`` for italics/bold)."""
  659. self._head += 2
  660. ticks = 2
  661. while self._read() == "'":
  662. self._head += 1
  663. ticks += 1
  664. italics = self._context & contexts.STYLE_ITALICS
  665. bold = self._context & contexts.STYLE_BOLD
  666. if ticks > 5:
  667. self._emit_text("'" * (ticks - 5))
  668. ticks = 5
  669. elif ticks == 4:
  670. self._emit_text("'")
  671. ticks = 3
  672. if (italics and ticks in (2, 5)) or (bold and ticks in (3, 5)):
  673. if ticks == 5:
  674. self._head -= 3 if italics else 2
  675. return self._pop()
  676. elif not self._can_recurse():
  677. if ticks == 3:
  678. if self._context & contexts.STYLE_SECOND_PASS:
  679. self._emit_text("'")
  680. return self._pop()
  681. self._context |= contexts.STYLE_PASS_AGAIN
  682. self._emit_text("'" * ticks)
  683. elif ticks == 2:
  684. self._parse_italics()
  685. elif ticks == 3:
  686. if self._parse_bold():
  687. return self._pop()
  688. elif ticks == 5:
  689. self._parse_italics_and_bold()
  690. self._head -= 1
  691. def _handle_list_marker(self):
  692. """Handle a list marker at the head (``#``, ``*``, ``;``, ``:``)."""
  693. markup = self._read()
  694. if markup == ";":
  695. self._context |= contexts.DL_TERM
  696. self._emit(tokens.TagOpenOpen(wiki_markup=markup))
  697. self._emit_text(get_html_tag(markup))
  698. self._emit(tokens.TagCloseSelfclose())
  699. def _handle_list(self):
  700. """Handle a wiki-style list (``#``, ``*``, ``;``, ``:``)."""
  701. self._handle_list_marker()
  702. while self._read(1) in ("#", "*", ";", ":"):
  703. self._head += 1
  704. self._handle_list_marker()
  705. def _handle_hr(self):
  706. """Handle a wiki-style horizontal rule (``----``) in the string."""
  707. length = 4
  708. self._head += 3
  709. while self._read(1) == "-":
  710. length += 1
  711. self._head += 1
  712. self._emit(tokens.TagOpenOpen(wiki_markup="-" * length))
  713. self._emit_text("hr")
  714. self._emit(tokens.TagCloseSelfclose())
  715. def _handle_dl_term(self):
  716. """Handle the term in a description list (``foo`` in ``;foo:bar``)."""
  717. self._context ^= contexts.DL_TERM
  718. if self._read() == ":":
  719. self._handle_list_marker()
  720. else:
  721. self._emit_text("\n")
  722. def _handle_end(self):
  723. """Handle the end of the stream of wikitext."""
  724. if self._context & contexts.FAIL:
  725. if self._context & contexts.TAG_BODY:
  726. if is_single(self._stack[1].text):
  727. return self._handle_single_tag_end()
  728. if self._context & contexts.DOUBLE:
  729. self._pop()
  730. self._fail_route()
  731. return self._pop()
  732. def _verify_safe(self, this):
  733. """Make sure we are not trying to write an invalid character."""
  734. context = self._context
  735. if context & contexts.FAIL_NEXT:
  736. return False
  737. if context & contexts.WIKILINK_TITLE:
  738. if this == "]" or this == "{":
  739. self._context |= contexts.FAIL_NEXT
  740. elif this == "\n" or this == "[" or this == "}":
  741. return False
  742. return True
  743. elif context & contexts.TEMPLATE_NAME:
  744. if this == "{" or this == "}" or this == "[":
  745. self._context |= contexts.FAIL_NEXT
  746. return True
  747. if this == "]":
  748. return False
  749. if this == "|":
  750. return True
  751. if context & contexts.HAS_TEXT:
  752. if context & contexts.FAIL_ON_TEXT:
  753. if this is self.END or not this.isspace():
  754. return False
  755. else:
  756. if this == "\n":
  757. self._context |= contexts.FAIL_ON_TEXT
  758. elif this is self.END or not this.isspace():
  759. self._context |= contexts.HAS_TEXT
  760. return True
  761. elif context & contexts.TAG_CLOSE:
  762. return this != "<"
  763. else:
  764. if context & contexts.FAIL_ON_EQUALS:
  765. if this == "=":
  766. return False
  767. elif context & contexts.FAIL_ON_LBRACE:
  768. if this == "{" or (self._read(-1) == self._read(-2) == "{"):
  769. if context & contexts.TEMPLATE:
  770. self._context |= contexts.FAIL_ON_EQUALS
  771. else:
  772. self._context |= contexts.FAIL_NEXT
  773. return True
  774. self._context ^= contexts.FAIL_ON_LBRACE
  775. elif context & contexts.FAIL_ON_RBRACE:
  776. if this == "}":
  777. if context & contexts.TEMPLATE:
  778. self._context |= contexts.FAIL_ON_EQUALS
  779. else:
  780. self._context |= contexts.FAIL_NEXT
  781. return True
  782. self._context ^= contexts.FAIL_ON_RBRACE
  783. elif this == "{":
  784. self._context |= contexts.FAIL_ON_LBRACE
  785. elif this == "}":
  786. self._context |= contexts.FAIL_ON_RBRACE
  787. return True
  788. def _parse(self, context=0, push=True):
  789. """Parse the wikicode string, using *context* for when to stop."""
  790. if push:
  791. self._push(context)
  792. while True:
  793. this = self._read()
  794. if self._context & contexts.UNSAFE:
  795. if not self._verify_safe(this):
  796. if self._context & contexts.DOUBLE:
  797. self._pop()
  798. self._fail_route()
  799. if this not in self.MARKERS:
  800. self._emit_text(this)
  801. self._head += 1
  802. continue
  803. if this is self.END:
  804. return self._handle_end()
  805. next = self._read(1)
  806. if this == next == "{":
  807. if self._can_recurse():
  808. self._parse_template_or_argument()
  809. else:
  810. self._emit_text("{")
  811. elif this == "|" and self._context & contexts.TEMPLATE:
  812. self._handle_template_param()
  813. elif this == "=" and self._context & contexts.TEMPLATE_PARAM_KEY:
  814. self._handle_template_param_value()
  815. elif this == next == "}" and self._context & contexts.TEMPLATE:
  816. return self._handle_template_end()
  817. elif this == "|" and self._context & contexts.ARGUMENT_NAME:
  818. self._handle_argument_separator()
  819. elif this == next == "}" and self._context & contexts.ARGUMENT:
  820. if self._read(2) == "}":
  821. return self._handle_argument_end()
  822. else:
  823. self._emit_text("}")
  824. elif this == next == "[" and self._can_recurse():
  825. if not self._context & contexts.INVALID_LINK:
  826. self._parse_wikilink()
  827. else:
  828. self._emit_text("[")
  829. elif this == "|" and self._context & contexts.WIKILINK_TITLE:
  830. self._handle_wikilink_separator()
  831. elif this == next == "]" and self._context & contexts.WIKILINK:
  832. return self._handle_wikilink_end()
  833. elif this == "[" and not self._context & contexts.INVALID_LINK: ## or this == ":"
  834. if self._can_recurse():
  835. self._parse_external_link(brackets=this == "[")
  836. else:
  837. self._emit_text("[")
  838. elif this == "=" and not self._global & contexts.GL_HEADING:
  839. if self._read(-1) in ("\n", self.START):
  840. self._parse_heading()
  841. else:
  842. self._emit_text("=")
  843. elif this == "=" and self._context & contexts.HEADING:
  844. return self._handle_heading_end()
  845. elif this == "\n" and self._context & contexts.HEADING:
  846. self._fail_route()
  847. elif this == "&":
  848. self._parse_entity()
  849. elif this == "<" and next == "!":
  850. if self._read(2) == self._read(3) == "-":
  851. self._parse_comment()
  852. else:
  853. self._emit_text(this)
  854. elif this == "<" and next == "/" and self._read(2) is not self.END:
  855. if self._context & contexts.TAG_BODY:
  856. self._handle_tag_open_close()
  857. else:
  858. self._handle_invalid_tag_start()
  859. elif this == "<" and not self._context & contexts.TAG_CLOSE:
  860. if self._can_recurse():
  861. self._parse_tag()
  862. else:
  863. self._emit_text("<")
  864. elif this == ">" and self._context & contexts.TAG_CLOSE:
  865. return self._handle_tag_close_close()
  866. elif this == next == "'":
  867. result = self._parse_style()
  868. if result is not None:
  869. return result
  870. elif self._read(-1) in ("\n", self.START):
  871. if this in ("#", "*", ";", ":"):
  872. self._handle_list()
  873. elif this == next == self._read(2) == self._read(3) == "-":
  874. self._handle_hr()
  875. else:
  876. self._emit_text(this)
  877. elif this in ("\n", ":") and self._context & contexts.DL_TERM:
  878. self._handle_dl_term()
  879. else:
  880. self._emit_text(this)
  881. self._head += 1
  882. def tokenize(self, text):
  883. """Build a list of tokens from a string of wikicode and return it."""
  884. split = self.regex.split(text)
  885. self._text = [segment for segment in split if segment]
  886. return self._parse()