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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
# -*- coding: utf-8 -*-
#
# Copyright (C) 2006 Alec Thomas <alec@swapoff.org>
# Copyright (C) 2011 Steffen Hoffmann <hoff.st@web.de>
#
# This software is licensed as described in the file COPYING, which
# you should have received as part of this distribution.
#
import re
from trac.config import BoolOption, ListOption
from trac.core import Component, implements
from trac.perm import PermissionError
from trac.resource import Resource
from trac.ticket.api import TicketSystem
from trac.ticket.model import Ticket
from trac.util import get_reporter_id
from trac.util.compat import set, sorted
from trac.util.text import to_unicode
from tractags.api import DefaultTagProvider, ITagProvider, TagSystem, _
class TicketTagProvider(DefaultTagProvider):
"""A tag provider using ticket fields as sources of tags.
Currently does NOT support custom fields.
"""
fields = ListOption('tags', 'ticket_fields', 'keywords',
doc=_("List of ticket fields to expose as tags."))
ignore_closed_tickets = BoolOption('tags', 'ignore_closed_tickets', True,
_("Do not collect tags from closed tickets."))
# custom_fields = ListOption('tags', 'custom_ticket_fields',
# doc=_("List of custom ticket fields to expose as tags."))
realm = 'ticket'
def check_permission(self, perm, action):
map = {'view': 'TICKET_VIEW', 'modify': 'TICKET_CHGPROP'}
return super(TicketTagProvider, self).check_permission(perm, action) \
and map[action] in perm
# ITagProvider methods
def get_tagged_resources(self, req, tags):
if not self.check_permission(req.perm, 'view'):
return
db = self.env.get_db_cnx()
fields = ["COALESCE(%s, '')" % f for f in self.fields]
ignore = ''
if self.ignore_closed_tickets:
ignore = " WHERE status != 'closed'"
sql = """
SELECT *
FROM (SELECT id, %s, %s AS std_fields
FROM ticket%s) s
""" % (','.join(self.fields), db.concat(*fields), ignore)
args = []
constraints = []
if tags:
for tag in tags:
constraints.append("std_fields %s" % db.like())
args.append('%' + db.like_escape(tag) + '%')
else:
constraints.append("std_fields != ''")
if constraints:
sql += " WHERE " + '(' + ' OR '.join(constraints) + ')'
sql += " ORDER BY id"
self.env.log.debug(sql)
cursor = db.cursor()
cursor.execute(sql, args)
for row in cursor:
id, ttags = row[0], ' '.join([f for f in row[1:-1] if f])
perm = req.perm('ticket', id)
if 'TICKET_VIEW' not in perm or 'TAGS_VIEW' not in perm:
continue
ticket_tags = TagSystem(self.env).split_into_tags(ttags)
tags = set([to_unicode(x) for x in tags])
if (not tags or ticket_tags.intersection(tags)):
yield Resource('ticket', id), ticket_tags
def get_resource_tags(self, req, resource):
assert resource.realm == self.realm
if not self.check_permission(req.perm, 'view'):
return
ticket = Ticket(self.env, resource.id)
return self._ticket_tags(ticket)
def set_resource_tags(self, req, resource, tags, comment=u''):
assert resource.realm == self.realm
if not self.check_permission(req.perm(resource), 'modify'):
raise PermissionError(resource=resource, env=self.env)
split_into_tags = TagSystem(self.env).split_into_tags
ticket = Ticket(self.env, resource.id)
all = self._ticket_tags(ticket)
keywords = split_into_tags(ticket['keywords'])
tags.difference_update(all.difference(keywords))
ticket['keywords'] = u' '.join(sorted(map(to_unicode, tags)))
ticket.save_changes(get_reporter_id(req), comment)
def remove_resource_tags(self, req, resource, comment=u''):
assert resource.realm == self.realm
if not self.check_permission(req.perm(resource), 'modify'):
raise PermissionError(resource=resource, env=self.env)
ticket = Ticket(self.env, resource.id)
ticket['keywords'] = u''
ticket.save_changes(get_reporter_id(req), comment)
def describe_tagged_resource(self, req, resource):
if not self.check_permission(req.perm, 'view'):
return ''
ticket = Ticket(self.env, resource.id)
if ticket.exists:
ticket_system = TicketSystem(self.env)
return ticket_system.get_resource_description(ticket.resource,
format='summary')
else:
return ''
# Private methods
def _ticket_tags(self, ticket):
split_into_tags = TagSystem(self.env).split_into_tags
return split_into_tags(
' '.join(filter(None, [ticket[f] for f in self.fields])))
|
