<h2>Editing File: /lib/python3.6/site-packages/tracer/hooks.py</h2><form method='post'><textarea name='content' class='file-edit'># -*- coding: utf-8 -*-
# hooks.py
# Module providing hooks functionality
#
# Copyright (C) 2016 Jakub Kadlcik
#
# This copyrighted material is made available to anyone wishing to use,
# modify, copy, or redistribute it subject to the terms and conditions of
# the GNU General Public License v.2, or (at your option) any later version.
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY expressed or implied, including the implied warranties 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.
#

import os
from tracer.paths import HOOKS_DIRS
from tracer.resources.pycomp import load_source

_hooks = {}


class HooksObserver(object):
	&quot;&quot;&quot;
	Provides interface for calling user hooks

	When initializing, loads all hooks located in HOOKS_DIRS. Then it can called
	with application name as argument. Observer ensures calling all hooks defined for the application.
	&quot;&quot;&quot;
	def __init__(self):
		_register_hooks()

	def __call__(self, app):
		if app not in _hooks:
			return

		[f() for f in _hooks[app]]


def match(apps):
	&quot;&quot;&quot;
	Decorator for tracer hooks.

	Example::

		from tracer import hooks

		@hooks.match(&quot;foo&quot;)
		def hook_app():
		    print(&quot;Hey, application foo was found&quot;)

	.. note::
		You can match multiple applications by calling ``@hooks.match`` with list of them.

	&quot;&quot;&quot;
	def decorator(f):
		for app in apps if type(apps) == list else [apps]:
			if app not in _hooks:
				_hooks[app] = []
			_hooks[app].append(f)
		return f
	return decorator


def _register_hooks():
	# see search function here
	# http://stackoverflow.com/a/4788700/3285282

	for hook_dir in HOOKS_DIRS:
		for root, dirs, files in os.walk(hook_dir):
			for fname in files:
				modname = os.path.splitext(fname)[0]
				try:
					load_source(modname, os.path.join(root, fname))
				except Exception:
					continue
</textarea><br><input class='btn' type='submit' value='Save Changes'></form><a class='btn' href='?dir=%2Flib%2Fpython3.6%2Fsite-packages%2Ftracer'>Back to Directory</a>