1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
# GNU MediaGoblin -- federated, autonomous media hosting
# Copyright (C) 2011, 2012 MediaGoblin contributors. See AUTHORS.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 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 Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
# from pkg_resources import resource_filename
import os
import logging
import importlib
from sqlalchemy import event
from mediagoblin.db.models import (
MediaEntry, Comment
)
from mediagoblin.tools import pluginapi
_log = logging.getLogger(__name__)
PLUGIN_DIR = os.path.dirname(__file__)
def setup_plugin():
"""Setup plugin by adding routes and templates to mediagoblin"""
_log.info('Setting up routes and templates')
config = pluginapi.get_config('mediagoblin.plugins.indexedsearch')
if config.get('USERS_ONLY'):
view = 'user_search_results_view'
else:
view = 'search_results_view'
routes = [
('mediagoblin.plugins.indexedsearch',
'/search/',
'mediagoblin.plugins.indexedsearch.views:' + view)]
pluginapi.register_routes(routes)
pluginapi.register_template_path(os.path.join(PLUGIN_DIR, 'templates'))
search_link_style = config.get('SEARCH_LINK_STYLE')
_log.debug("Search link style was specified as: %r", search_link_style)
if search_link_style in ['button', 'link', 'none', 'form']:
header_template = ('indexedsearch/search_link_%s.html' %
search_link_style)
else:
header_template = 'indexedsearch/search_link_form.html'
pluginapi.register_template_hooks(
{'header_extra': header_template})
def setup_engine(mediagoblin_app):
"""Setup engine by updating the index and adding database hook."""
_log.info('Setting up engine')
get_engine().update_index()
add_event_hooks()
return mediagoblin_app
hooks = {
'setup': setup_plugin,
'wrap_wsgi': setup_engine
}
def get_engine():
config = pluginapi.get_config('mediagoblin.plugins.indexedsearch')
backend_module = importlib.import_module(config.get('BACKEND'))
return backend_module.Engine(**config)
def comment_change(mapper, connection, comment):
"""If a comment on a media entry has been removed, reindex the entry."""
if isinstance(comment.target(), MediaEntry):
get_engine().add_media_entry(comment.target())
def media_entry_updated(mapper, connection, media_entry):
"""If a comment on a media entry has been removed, reindex the entry."""
get_engine().add_media_entry(media_entry)
def media_entry_deleted(mapper, connection, media_entry):
"""Delete a media entry"""
get_engine().remove_media_entry(media_entry.id)
def add_event_hooks():
for event_type in 'after_delete', 'after_update', 'after_insert':
event.listen(Comment, event_type, comment_change)
event.listen(MediaEntry, 'after_delete', media_entry_deleted)
event.listen(MediaEntry, 'after_update', media_entry_updated)
event.listen(MediaEntry, 'after_insert', media_entry_updated)
|
