コード例 #1
0
ファイル: test_sw_urllib.py プロジェクト: dlitz/stillweb
    def test_rebase_url(self):
        """rebase_url"""
        from StillWeb.sw_urllib import rebase_url

        # Simple rebasing
        self.assertEqual("http://a/new/b/c", rebase_url(url="http://a/b/c", src="http://a/", dest="http://a/new/"))
        self.assertEqual("http://a/new/b/c", rebase_url("http://a/old/b/c", "http://a/old/", "http://a/new/"))

        # When the URL can't be rebased
        self.assertEqual("http://xyz/foo", rebase_url("http://xyz/foo", "http://a/old/", "http://a/new/"))
        self.assertEqual("http://a/foo", rebase_url("/foo", "http://a/bar/", "http://xyz/new/"))
        self.assertRaises(ValueError, rebase_url, "/foo", "http://a/bar/", "http://xyz/new/", must_rebase=True)
コード例 #2
0
ファイル: LinkRewriter.py プロジェクト: dlitz/stillweb
    def cb(url, criterion):
        # Resolve the link URL with respect to the current (fake) URL
        link_url = rfc3986_urljoin(fake_current_url, url)

        # Convert the fake URL into an absolute real URL
        link_url = rebase_url(link_url, fake_base_url, real_base_url)

        if not always_absolute:
            # Convert the absolute URL into a relative URL
            link_url = relative_url(link_url, real_current_url)

        return link_url
コード例 #3
0
ファイル: sw_util.py プロジェクト: dlitz/stillweb
    def __init__(self, framework, orig_target_url):
        if not orig_target_url.startswith("/"):
            raise AssertionError("orig_target_url must start with /")

        self.orig_target_url = orig_target_url
        self.target_url = strip_index_from_url(orig_target_url)
        self.pathtuple = pathtuple_from_target_url(orig_target_url)
        if "output_dir" in framework.plugins['vars'].vars:
            self.output_dir = framework.plugins['vars'].vars['output_dir']
            self.output_filename = os.path.join(self.output_dir, *self.pathtuple)
        if "source_dir" in framework.plugins['vars'].vars:
            self.source_dir = framework.plugins['vars'].vars['source_dir']
            self.source_filename = os.path.join(self.source_dir, *self.pathtuple)

        if "base_url" in framework.plugins['vars'].vars:
            self.base_url = framework.plugins['vars'].vars['base_url']
            self.current_url = rebase_url(self.target_url, generate_fake_url(), self.base_url)
コード例 #4
0
ファイル: LinkRewriter.py プロジェクト: dlitz/stillweb
def rewrite_links(node, match_criteria, target_url, base_url, always_absolute=False):
    # We generate a fake URL so links like <a href="/">...</a> will resolve
    # to the top-level URL of the *site* rather than of the *server*.
    fake_base_url = generate_fake_url()
    fake_current_url = rfc3986_urljoin(fake_base_url, target_url)
    real_base_url = base_url
    real_current_url = rebase_url(fake_current_url, fake_base_url, real_base_url)

    def cb(url, criterion):
        # Resolve the link URL with respect to the current (fake) URL
        link_url = rfc3986_urljoin(fake_current_url, url)

        # Convert the fake URL into an absolute real URL
        link_url = rebase_url(link_url, fake_base_url, real_base_url)

        if not always_absolute:
            # Convert the absolute URL into a relative URL
            link_url = relative_url(link_url, real_current_url)

        return link_url

    LinkRewriter(match_criteria).rewrite_links(node, cb)