Django,自定义模板过滤器-regex问题

如果字符串中除wiki链接外还包含其他文本,则由于您正在使用 re.match 而不是 re.search . 重赛 在字符串的开头匹配。 重新搜索 匹配字符串中的任何位置。见 matching vs. searching .

另外,你的正则表达式使用贪婪 * ,因此,如果一行包含多个wiki链接,它将不起作用。使用 *? 相反,让它不贪婪:

re.search(r'\[\[(.*?)\|(.*?)\]\]', value)

编辑:

关于如何修复代码的提示,我建议您使用 re.sub with a callback . 其优点是:

• 如果您在同一行中有多个wiki链接,它将正常工作。
• 一次越过绳子就足够了。你不需要通行证就可以找到维基链接,也不需要另一个通行证就可以进行替换。

以下是实施的示意图:

import re

WIKILINK_RE = re.compile(r'\[\[(.*?)\|(.*?)\]\]')

def wikilink(value):
  def wikilink_sub_callback(match_obj):
    alias = match_obj.group(1).strip()
    text = match_obj.group(2).strip()
    if(alias exists in Page):
      class_attr = ''
    else:
      class_attr = ' class="redlink"'
    return '<a href="%s"%s>%s</a>' % (alias, class_attr, text)

  return WIKILINK_RE.sub(wikilink_sub_callback, value)

这类问题很快就落在一组小的单元测试中。

可以单独测试的过滤器片段(通过一点代码重组):

• 确定值是否包含要查找的模式
• 如果有匹配的页,会生成什么字符串
• 生成的字符串是没有匹配的页

这将帮助你隔离出哪里出了问题。您可能会发现需要重新连接regexps以考虑周围的可选空格。

而且,乍一看,您的过滤器似乎是可利用的。您声称结果是安全的,但是您没有为脚本标记之类的错误过滤alt文本。

代码:

import re

def page_exists(alias):
    if alias == 'ThisIsAWikiLink':
        return True

    return False

def wikilink(value):
    if value == None:
        return None

    for alias, text in re.findall('\[\[\s*(.*?)\s*\|\s*(.*?)\s*\]\]',value):
        if page_exists(alias):
            value = re.sub('\[\[\s*%s\s*\|\s*%s\s*\]\]' % (alias,text), '<a href="/Sites/wiki/%s">%s</a>' % (alias, text),value)            
        else:
            value = re.sub('\[\[\s*%s\s*\|\s*%s\s*\]\]' % (alias,text), '<a href="/Sites/wiki/%s/edit/" class="redtext">%s</a>' % (alias, text), value)

    return value

样品结果:

>>> import wikilink
>>> wikilink.wikilink(None)
>>> wikilink.wikilink('')
''
>>> wikilink.wikilink('Test')
'Test'
>>> wikilink.wikilink('[[ThisIsAWikiLink | This is the alt text]]')
'<a href="/Sites/wiki/ThisIsAWikiLink">This is the alt text</a>'
>>> wikilink.wikilink('[[ThisIsABadWikiLink | This is the alt text]]')
'<a href="/Sites/wiki/ThisIsABadWikiLink/edit/" class="redtext">This is the alt text</a>'
>>> wikilink.wikilink('[[ThisIsAWikiLink | This is the alt text]]\n[[ThisIsAWikiLink | This is another instance]]')
'<a href="/Sites/wiki/ThisIsAWikiLink">This is the alt text</a>\n<a href="/Sites/wiki/ThisIsAWikiLink">This is another instance</a>'
>>> wikilink.wikilink('[[ThisIsAWikiLink | This is the alt text]]\n[[ThisIsAWikiLink | This is another instance]]')

一般意见:

• 芬德尔 你要找的是魔法的功能吗?
• 变化 存在的页面 运行所需的任何查询
• 易受HTML注入攻击(如上述Dave W.Smith所述)
• 在每次迭代中必须重新编译regex效率很低
• 每次查询数据库效率低下

我认为使用这种方法会很快遇到性能问题。

这是工作代码,以防有人需要它:

from django import template
from django.template.defaultfilters import stringfilter
from sites.wiki.models import Page
import re

register = template.Library()

@register.filter
@stringfilter
def wikilink(value):
  WIKILINK_RE = re.compile(r'\[\[ ?(.*?) ?\| ?(.*?) ?\]\]')

  def wikilink_sub_callback(match_obj):
    alias = match_obj.group(1).strip()
    text = match_obj.group(2).strip()

    class_attr = ''
    try:
        Page.objects.get(alias=alias)
    except Page.DoesNotExist:
        class_attr = ' class="redlink"'
    return '<a href="%s"%s>%s</a>' % (alias, class_attr, text)

  return WIKILINK_RE.sub(wikilink_sub_callback, value)
wikilink.is_safe = True

非常感谢您的回答!