mirror of
https://github.com/Hopiu/linkchecker.git
synced 2026-03-17 14:30:30 +00:00
103 lines
3.4 KiB
Python
103 lines
3.4 KiB
Python
# -*- coding: iso-8859-1 -*-
|
|
# Copyright (C) 2014 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.
|
|
"""
|
|
Module for plugin management.
|
|
"""
|
|
from .. import loader, log, LOG_PLUGIN
|
|
|
|
|
|
class _PluginBase(object):
|
|
"""Basic plugin class featuring plugin identification and
|
|
helper functions."""
|
|
|
|
def __init__(self, config):
|
|
"""Add plugin-specific configuration."""
|
|
pass
|
|
|
|
def check(self, url_data):
|
|
"""Common check method run for all plugins."""
|
|
pass
|
|
|
|
@classmethod
|
|
def read_config(cls, configparser):
|
|
"""Read configuration file options."""
|
|
pass
|
|
|
|
|
|
class _ConnectionPlugin(_PluginBase):
|
|
"""Plugins run after connection checks."""
|
|
pass
|
|
|
|
|
|
class _ContentPlugin(_PluginBase):
|
|
"""Plugins run for valid URLs with content."""
|
|
pass
|
|
|
|
|
|
def get_plugin_modules(folders, package='plugins',
|
|
parentpackage='linkcheck.dummy'):
|
|
"""Get plugin modules for given folders."""
|
|
for folder in folders:
|
|
for module in loader.get_folder_modules(folder, parentpackage):
|
|
yield module
|
|
for module in loader.get_package_modules(package):
|
|
yield module
|
|
|
|
|
|
def get_plugin_classes(modules):
|
|
"""Get plugin classes for given modules."""
|
|
classes = (_ConnectionPlugin, _ContentPlugin)
|
|
return loader.get_plugins(modules, classes)
|
|
|
|
|
|
class PluginManager(object):
|
|
"""Manage all connection and content plugins."""
|
|
|
|
def __init__(self, config):
|
|
"""Load enabled plugins."""
|
|
self.connection_plugins = []
|
|
self.content_plugins = []
|
|
folders = config["pluginfolders"]
|
|
modules = get_plugin_modules(folders)
|
|
self.load_modules(modules, config)
|
|
|
|
def load_modules(self, modules, config):
|
|
"""Load plugin modules."""
|
|
for pluginclass in get_plugin_classes(modules):
|
|
name = pluginclass.__name__
|
|
if name in config["enabledplugins"]:
|
|
if issubclass(pluginclass, _ConnectionPlugin):
|
|
log.debug(LOG_PLUGIN, "Enable connection plugin %s", name)
|
|
self.connection_plugins.append(pluginclass(config[name]))
|
|
else:
|
|
log.debug(LOG_PLUGIN, "Enable content plugin %s", name)
|
|
self.content_plugins.append(pluginclass(config[name]))
|
|
|
|
def run_connection_plugins(self, url_data):
|
|
"""Run all connection plugins."""
|
|
run_plugins(self.connection_plugins, url_data)
|
|
|
|
def run_content_plugins(self, url_data):
|
|
"""Run all content plugins."""
|
|
run_plugins(self.content_plugins, url_data)
|
|
|
|
|
|
def run_plugins(plugins, url_data):
|
|
"""Run the check(url_data) method of given plugins."""
|
|
for plugin in plugins:
|
|
log.debug(LOG_PLUGIN, "Run plugin %s", plugin.__class__.__name__)
|
|
plugin.check(url_data)
|