code-git/share/roundup/templates/devel/extensions/local_replace.py0000644000000000000000000000302311647305455024140 0ustar 00000000000000import re substitutions = [ (re.compile('debian:\#(?P\d+)'), 'debian#\g' ), (re.compile('\#(?P\s*)(?P\d+)'), "#\g\g" ), (re.compile('(?P\s+)revision(?P\s*)(?P\d+)'), "\grevision\g\g"), (re.compile('(?P\s+)rev(?P\s*)(?P\d+)'), "\grev\g\g"), (re.compile('(?P\s+)(?Pr|r\s+)(?P\d+)'), "\g\g\g"), ] def local_replace(message): for cre, replacement in substitutions: message = cre.sub(replacement, message) return message def init(instance): instance.registerUtil('localReplace', local_replace) if "__main__" == __name__: print " debian:#222", local_replace(" debian:#222") print " revision 222", local_replace(" revision 222") print " wordthatendswithr 222", local_replace(" wordthatendswithr 222") print " r222", local_replace(" r222") print " r 222", local_replace(" r 222") print " #555", local_replace(" #555") code-git/share/roundup/templates/devel/extensions/spambayes.py0000644000000000000000000000520111647305455023337 0ustar 00000000000000import re, math from roundup.cgi.actions import Action from roundup.cgi.exceptions import * import xmlrpclib, socket REVPAT = re.compile(r'(r[0-9]+\b|rev(ision)? [0-9]+\b)') def extract_classinfo(db, classname, nodeid): node = db.getnode(classname, nodeid) authorage = node['creation'].timestamp() - \ db.getnode('user', node.get('author', node.get('creator')))['creation'].timestamp() authorid = node.get('author', node.get('creator')) content = db.getclass(classname).get(nodeid, 'content') tokens = ["klass:%s" % classname, "author:%s" % authorid, "authorage:%d" % int(math.log(authorage)), "hasrev:%s" % (REVPAT.search(content) is not None)] return (content, tokens) def train_spambayes(db, content, tokens, is_spam): spambayes_uri = db.config.detectors['SPAMBAYES_URI'] server = xmlrpclib.ServerProxy(spambayes_uri, verbose=False) try: server.train({'content':content}, tokens, {}, is_spam) return (True, None) except (socket.error, xmlrpclib.Error), e: return (False, str(e)) class SpambayesClassify(Action): permissionType = 'SB: May Classify' def handle(self): (content, tokens) = extract_classinfo(self.db, self.classname, self.nodeid) if self.form.has_key("trainspam"): is_spam = True elif self.form.has_key("trainham"): is_spam = False (status, errmsg) = train_spambayes(self.db, content, tokens, is_spam) node = self.db.getnode(self.classname, self.nodeid) props = {} if status: if node.get('spambayes_misclassified', False): props['spambayes_misclassified'] = True props['spambayes_score'] = 1.0 s = " SPAM" if not is_spam: props['spambayes_score'] = 0.0 s = " HAM" self.client.ok_message.append(self._('Message classified as') + s) else: self.client.error_message.append(self._('Unable to classify message, got error:') + errmsg) klass = self.db.getclass(self.classname) klass.set(self.nodeid, **props) self.db.commit() def sb_is_spam(obj): cutoff_score = float(obj._db.config.detectors['SPAMBAYES_SPAM_CUTOFF']) try: score = obj['spambayes_score'] except KeyError: return False return score >= cutoff_score def init(instance): instance.registerAction("spambayes_classify", SpambayesClassify) instance.registerUtil('sb_is_spam', sb_is_spam) code-git/share/roundup/templates/devel/extensions/timestamp.py0000644000000000000000000000157211647305455023365 0ustar 00000000000000import time, struct, base64 from roundup.cgi.actions import RegisterAction from roundup.cgi.exceptions import * def timestamp(): return base64.encodestring(struct.pack("i", time.time())).strip() def unpack_timestamp(s): return struct.unpack("i",base64.decodestring(s))[0] class Timestamped: def check(self): try: created = unpack_timestamp(self.form['opaque'].value) except KeyError: raise FormError, "somebody tampered with the form" if time.time() - created < 4: raise FormError, "responding to the form too quickly" return True class TimestampedRegister(Timestamped, RegisterAction): def permission(self): self.check() RegisterAction.permission(self) def init(instance): instance.registerUtil('timestamp', timestamp) instance.registerAction('register', TimestampedRegister) code-git/share/roundup/templates/devel/extensions/timezone.py0000644000000000000000000000161311647305455023210 0ustar 00000000000000# Utility for replacing the simple input field for the timezone with # a select-field that lists the available values. import cgi try: import pytz except ImportError: pytz = None def tzfield(prop, name, default): if pytz: value = prop.plain() if '' == value: value = default else: try: value = "Etc/GMT%+d" % int(value) except ValueError: pass l = ['') return '\n'.join(l) else: return prop.field() def init(instance): instance.registerUtil('tzfield', tzfield)