mirror of
https://github.com/Hopiu/linkchecker.git
synced 2026-03-18 15:00:28 +00:00
151 lines
4.8 KiB
Python
151 lines
4.8 KiB
Python
# -*- coding: iso-8859-1 -*-
|
|
# Copyright (C) 2000-2010 Bastian Kleineidam
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License along
|
|
# with this program; if not, write to the Free Software Foundation, Inc.,
|
|
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
"""
|
|
Main functions for link checking.
|
|
"""
|
|
|
|
import os
|
|
import cgi
|
|
import logging
|
|
import urllib
|
|
from .. import strformat, url as urlutil
|
|
|
|
|
|
def absolute_url (base_url, base_ref, parent_url):
|
|
"""
|
|
Search for the absolute url to detect the link type. This does not
|
|
join any url fragments together!
|
|
|
|
@param base_url: base url from a link tag
|
|
@type base_url: string or None
|
|
@param base_ref: base url from <base> tag
|
|
@type base_ref: string or None
|
|
@param parent_url: url of parent document
|
|
@type parent_url: string or None
|
|
"""
|
|
if base_url and urlutil.url_is_absolute(base_url):
|
|
return base_url
|
|
elif base_ref and urlutil.url_is_absolute(base_ref):
|
|
return base_ref
|
|
elif parent_url and urlutil.url_is_absolute(parent_url):
|
|
return parent_url
|
|
return u""
|
|
|
|
|
|
def get_url_from (base_url, recursion_level, aggregate,
|
|
parent_url=None, base_ref=None, line=0, column=0,
|
|
name=u""):
|
|
"""
|
|
Get url data from given base data.
|
|
|
|
@param base_url: base url from a link tag
|
|
@type base_url: string or None
|
|
@param recursion_level: current recursion level
|
|
@type recursion_level: number
|
|
@param aggregate: aggregate object
|
|
@type aggregate: aggregate.Consumer
|
|
@param parent_url: parent url
|
|
@type parent_url: string or None
|
|
@param base_ref: base url from <base> tag
|
|
@type base_ref string or None
|
|
@param line: line number
|
|
@type line: number
|
|
@param column: column number
|
|
@type column: number
|
|
@param name: link name
|
|
@type name: string
|
|
"""
|
|
if base_url is not None:
|
|
base_url = strformat.unicode_safe(base_url)
|
|
if parent_url is not None:
|
|
parent_url = strformat.unicode_safe(parent_url)
|
|
if base_ref is not None:
|
|
base_ref = strformat.unicode_safe(base_ref)
|
|
name = strformat.unicode_safe(name)
|
|
url = absolute_url(base_url, base_ref, parent_url).lower()
|
|
if not (url or name):
|
|
# use filename as base url, with slash as path seperator
|
|
name = base_url.replace("\\", "/")
|
|
klass = get_urlclass_from(url)
|
|
return klass(base_url, recursion_level, aggregate,
|
|
parent_url=parent_url, base_ref=base_ref,
|
|
line=line, column=column, name=name)
|
|
|
|
|
|
def get_urlclass_from (url):
|
|
"""Return checker class for given URL."""
|
|
if url.startswith("http:"):
|
|
klass = httpurl.HttpUrl
|
|
elif url.startswith("ftp:"):
|
|
klass = ftpurl.FtpUrl
|
|
elif url.startswith("file:"):
|
|
klass = fileurl.FileUrl
|
|
elif url.startswith("telnet:"):
|
|
klass = telneturl.TelnetUrl
|
|
elif url.startswith("mailto:"):
|
|
klass = mailtourl.MailtoUrl
|
|
elif url.startswith("https:"):
|
|
klass = httpsurl.HttpsUrl
|
|
elif url.startswith(("nntp:", "news:", "snews:")):
|
|
klass = nntpurl.NntpUrl
|
|
elif unknownurl.is_unknown_url(url):
|
|
# unknown url
|
|
klass = unknownurl.UnknownUrl
|
|
else:
|
|
# assume local file
|
|
klass = fileurl.FileUrl
|
|
return klass
|
|
|
|
|
|
def get_index_html (urls):
|
|
"""
|
|
Construct artificial index.html from given URLs.
|
|
|
|
@param urls: URL strings
|
|
@type urls: iterator of string
|
|
"""
|
|
lines = ["<html>", "<body>"]
|
|
for entry in urls:
|
|
name = cgi.escape(entry)
|
|
try:
|
|
url = cgi.escape(urllib.quote(entry))
|
|
except KeyError:
|
|
# Some unicode entries raise KeyError.
|
|
url = name
|
|
lines.append('<a href="%s">%s</a>' % (url, name))
|
|
lines.extend(["</body>", "</html>"])
|
|
return os.linesep.join(lines)
|
|
|
|
|
|
class StoringHandler (logging.Handler):
|
|
"""Store all emitted log messages in a size-limited list.
|
|
Used by the CSS syntax checker."""
|
|
|
|
def __init__ (self, maxrecords=100):
|
|
logging.Handler.__init__(self)
|
|
self.storage = []
|
|
self.maxrecords = maxrecords
|
|
|
|
def emit (self, record):
|
|
if len(self.storage) >= self.maxrecords:
|
|
self.storage.pop()
|
|
self.storage.append(record)
|
|
|
|
|
|
# all the URL classes
|
|
from . import (fileurl, unknownurl, ftpurl, httpurl,
|
|
httpsurl, mailtourl, telneturl, nntpurl)
|