Question

我有http代理项目。我使用了一些重写规则来更改所有操作并将href更改为此/follow/+URL。我还将所有src的网址更改为此/proxy/+URL，这有助于我在链接之间分配网址，例如图片。它适用于<img>标记。我遇到的问题是<iframe src=...>标记。在我应用我的代码后，它将src更改为'/ proxy / + URL'，但我想将其更改为/follow/+URL，我不知道如何做到这一点。

这是我的代码

from urlparse import urlparse, urlunparse
import re, urllib


REWRITE_LINKS = re.compile(r'((?P<attr>action|href|src)=["\'](?P<uri>\S+?)["\'])', re.IGNORECASE)


def rewrite_links(content, mimetype = '', uri = ''):
    uri = str(uri)
    urip = urlparse(uri)
    server_root = str(urlunparse((urip[0], urip[1], '/', '', '', '')))

    working_dir = str(urlunparse((urip[0], urip[1], urip[2], '/', '', '')))
    working_dir = '/'.join(working_dir.split('/')[:-1])

    def repl_html(match):
        attr, value = match.groupdict()['attr'], match.groupdict()['uri']

        if value in ('',) or value.startswith('javascript:') or value.startswith('#'):
            pass
        else:
            if value.find('://') == -1:
                if value.startswith('./'):
                    value = working_dir + value[2:]
                elif value.startswith('../'):
                    value = '/'.join(working_dir.split('/')[:-1]) + value[3:]
                elif value.startswith('/'):
                    value = server_root + value[1:]
                else:
                    value = server_root + value
            #value = value.replace('/','|')
            if attr.lower() == 'src':
                value = '/proxy/' + value
            else:
                value = '/follow/' + value

        return ' %s="%s" ' % (attr, value)

    if mimetype.startswith('text/html'):
        content = REWRITE_LINKS.sub(repl_html, content)
    elif mimetype.startswith('text/css'):
        pass
    elif mimetype.startswith('application/x-javascript'):
        pass
    else:
        pass

    return content

你有什么建议吗？

Answer 1

您可以尝试添加额外的命名组：

<(?P<tagname>[^\s]+)\s[^>]*

到REWRITE_LINKS这样的行：

REWRITE_LINKS = re.compile(r'(<(?P<tagname>[^\s]+)\s[^>]*(?P<attr>action|href|src)=["\'](?P<uri>\S+?)["\'])', re.IGNORECASE)

然后您可以在if attr.lower() == 'src':条件语句

中查询此匹配项

如何匹配iframe标签的一部分？

1 个答案: