keithsalisbury-subtrac 0.1.1

data/lib/subtrac/trac-plugins/timingandestimationplugin/timingandestimationplugin/reports_filter.py

@@ -0,0 +1,150 @@
+from trac.web.api import ITemplateStreamFilter
+from trac.core import *
+from genshi.core import *
+from genshi.builder import tag
+from sets import Set as set
+from genshi.filters.transform import Transformer
+import re
+
+from trac.ticket.report import ReportModule
+from trac.util.datefmt import format_datetime, format_time
+import csv
+from trac.web.api import RequestDone
+
+billing_report_regex = re.compile("\{(?P<reportid>\d*)\}")
+def report_id_from_text(text):
+    m = billing_report_regex.match(text)
+    if m:
+        return int(m.groupdict()["reportid"])
+
+def get_billing_reports(comp):
+    cur = comp.env.get_db_cnx().cursor()
+    try:
+        cur.execute("SELECT id FROM custom_report")
+        return set([x[0] for x in cur.fetchall()])
+    except Exception, e:
+        # if we can't get the billing reports (e.g. the
+        # TimingAndEstimationPlugin isn't installed), silently continue
+        # without hiding anything.
+        return set()
+
+class RowFilter(object):
+    """A genshi filter that operates on table rows, completely hiding any that
+    are in the billing_reports table."""
+
+    def __init__(self, comp):
+        self.component = comp
+        self.billing_reports = get_billing_reports(comp)
+        self.component.log.debug('self.billing_reports= %r' % self.billing_reports)
+
+    def __call__(self, row_stream):
+        events = list(row_stream)
+        report_url = Stream(events) \
+                        .select('td[@class="report"]/a/@href').render()
+        id = int(report_url.split('/')[-1])
+
+        if not id in self.billing_reports:
+            for kind,data,pos in Stream(events):
+                yield kind,data,pos
+
+class ReportsFilter(Component):
+    """Remove all billing reports from the reports list."""
+    implements(ITemplateStreamFilter)
+
+    def filter_stream(self, req, method, filename, stream, data):
+        if not filename == 'report_view.html':
+            return stream
+        self.log.debug("Applying Reports Filter to remove T&E reports")
+        return stream | Transformer(
+            '//table[@class="listing reports"]/tbody/tr'
+            ).filter(RowFilter(self))
+
+class ReportScreenFilter(Component):
+    """Hides TandE reports even when you just go to the url"""
+    implements(ITemplateStreamFilter)
+    def __init__(self):
+        self.billing_reports = get_billing_reports(self)
+        self.log.debug('ReportScreenFilter: self.billing_reports= %r' % self.billing_reports)
+
+    def filter_stream(self, req, method, filename, stream, data):
+        if not filename == 'report_view.html':
+            return stream
+        reportid = [None]
+
+        def idhelper(strm):
+            header = strm[0][1]
+            if not reportid[0]:
+                self.log.debug("ReportScreenFilter: helper: %s %s %s"%(strm,header,report_id_from_text(header)))
+                reportid[0] = report_id_from_text(header) 
+            for kind, data, pos in strm:
+                yield kind, data, pos       
+                
+        def permhelper(strm):
+            id = reportid[0]
+            self.log.debug("ReportScreenFilter: id:%s, in bill: %s   has perm:%s" % (id, id in self.billing_reports, req.perm.has_permission("TIME_VIEW")))
+            if id and id in self.billing_reports and not req.perm.has_permission("TIME_VIEW"):
+                self.log.debug("ReportScreenFilter: No time view, prevent render")
+                msg = "YOU MUST HAVE TIME_VIEW PERMSSIONS TO VIEW THIS REPORT"
+                for kind, data, pos in tag.span(msg).generate():
+                    yield kind, data, pos
+            else:
+                for kind, data, pos in strm:
+                    yield kind, data, pos
+
+        self.log.debug("ReportScreenFilter: About to prevent rendering of billing reports without permissions")
+        stream = stream | Transformer('//div[@id="content"]/h1/text()').filter(idhelper)
+        stream = stream | Transformer('//div[@id="content"]').filter(permhelper)
+        return stream
+
+## ENFORCE PERMISSIONS ON report exports
+
+billing_report_fname_regex = re.compile("report_(?P<reportid>\d*)")
+def report_id_from_filename(text):
+    if text:
+        m = billing_report_fname_regex.match(text)
+        if m:
+            return int(m.groupdict()["reportid"])
+    return -1;
+
+
+def _send_csv(self, req, cols, rows, sep=',', mimetype='text/plain',
+              filename=None):
+    req.send_response(200)
+    req.send_header('Content-Type', mimetype + ';charset=utf-8')
+    if filename:
+        req.send_header('Content-Disposition', 'filename=' + filename)
+    req.end_headers()
+    
+    id = report_id_from_filename(filename)
+    reports = get_billing_reports(self)
+    if id in reports and not req.perm.has_permission("TIME_VIEW"):
+        raise RequestDone
+    
+    def iso_time(t):
+        return format_time(t, 'iso8601')
+
+    def iso_datetime(dt):
+        return format_datetime(dt, 'iso8601')
+
+    col_conversions = {
+        'time': iso_time,
+        'datetime': iso_datetime,
+        'changetime': iso_datetime,
+        'date': iso_datetime,
+        'created': iso_datetime,
+        'modified': iso_datetime,
+        }
+    
+    converters = [col_conversions.get(c.strip('_'), unicode) for c in cols]
+    
+    writer = csv.writer(req, delimiter=sep)
+    writer.writerow([unicode(c).encode('utf-8') for c in cols])
+    for row in rows:
+        row = list(row)
+        for i in xrange(len(row)):
+            row[i] = converters[i](row[i]).encode('utf-8')
+        writer.writerow(row)
+
+    raise RequestDone
+
+ReportModule._send_csv = _send_csv

data/lib/subtrac/trac-plugins/timingandestimationplugin/timingandestimationplugin/statuses.py

@@ -0,0 +1,25 @@
+import dbhelper
+from trac.ticket.default_workflow import get_workflow_config
+from sets import Set
+
+def status_variables(statuses):
+    return ', '.join(['$'+i.upper().replace("_","").replace(" ","") for i in list(statuses)])
+
+def get_statuses(com):
+    config = com.config
+    stats = get_statuses_from_workflow(config)
+    status_sql = """
+    SELECT DISTINCT status FROM ticket WHERE status <> '' ;
+    """
+    stats |= Set(dbhelper.get_column_as_list(com, status_sql))
+    stats.difference_update(['', None])
+    return stats
+
+def get_statuses_from_workflow(config):
+    wf = get_workflow_config(config)
+    x = Set()
+    for key, value in wf.items():
+        x.add(value['newstate'])
+        x |= Set(value['oldstates'])
+    x.difference_update([u'*'])
+    return x

data/lib/subtrac/trac-plugins/timingandestimationplugin/timingandestimationplugin/tande_filters.py

@@ -0,0 +1,131 @@
+from trac.web.api import ITemplateStreamFilter
+from trac.perm import IPermissionRequestor
+from trac.core import *
+from genshi.core import *
+from genshi.builder import tag
+from sets import Set as set
+from genshi.filters.transform import Transformer
+from blackmagic import *
+from trac.ticket.query import QueryModule
+
+from StringIO import StringIO
+import csv
+from trac.mimeview.api import Context
+from trac.resource import Resource
+
+## MONKEY PATCH THE QUERY MODULE CSV EXPORT FN TO ENFORCE PERMISSIONS
+def new_csv_export(self, req, query, sep=',', mimetype='text/plain'):
+    self.log.debug("T&E plugin has overridden QueryModule.csv_export so to enforce field permissions")
+
+    ## find the columns that should be hidden
+    hidden_fields = []
+    fields = self.config.getlist(csection, 'fields', [])
+    self.log.debug('QueryModule.csv_export: found : %s' % fields)
+    for field in fields:
+        perms = self.config.getlist(csection, '%s.permission' % field, [])
+        #self.log.debug('QueryModule.csv_export: read permission config: %s has %s' % (field, perms))
+        for (perm, denial) in [s.split(":") for s in perms] :
+            perm = perm.upper()
+            #self.log.debug('QueryModule.csv_export: testing permission: %s:%s should act= %s' %
+            #               (field, perm, (not req.perm.has_permission(perm) or perm == "ALWAYS")))
+            if (not req.perm.has_permission(perm) or perm == "ALWAYS") and denial.lower() in ["remove","hide"]:
+                hidden_fields.append(field)
+    ## END find the columns that should be hidden
+    
+    content = StringIO()
+    cols = query.get_columns()
+    writer = csv.writer(content, delimiter=sep)
+    writer = csv.writer(content, delimiter=sep, quoting=csv.QUOTE_MINIMAL)
+    writer.writerow([unicode(c).encode('utf-8') for c in cols if c not in hidden_fields])
+    
+    context = Context.from_request(req)
+    results = query.execute(req, self.env.get_db_cnx())
+    self.log.debug('QueryModule.csv_export: hidding columns %s' %  hidden_fields)
+    for result in results:
+        ticket = Resource('ticket', result['id'])
+        if 'TICKET_VIEW' in req.perm(ticket):
+            values = []
+            for col in cols:
+                if col not in hidden_fields:
+                    value = result[col]
+                    if col in ('cc', 'reporter'):
+                        value = Chrome(self.env).format_emails(context(ticket),
+                                                               value)
+                    values.append(unicode(value).encode('utf-8'))
+            writer.writerow(values)
+    return (content.getvalue(), '%s;charset=utf-8' % mimetype)
+
+QueryModule.export_csv = new_csv_export
+
+class TicketFormatFilter(Component):
+    """Filtering the streams to alter the base format of the ticket"""
+    implements(ITemplateStreamFilter)
+
+    def filter_stream(self, req, method, filename, stream, data):
+        self.log.debug("TicketFormatFilter executing") 
+        if not filename == 'ticket.html':
+            self.log.debug("TicketFormatFilter not the correct template")
+            return stream
+        
+        self.log.debug("TicketFormatFilter disabling totalhours and removing header hours")
+        stream = disable_field(stream, "totalhours")
+        stream = remove_header(stream, "hours")
+        return stream 
+
+
+class QueryColumnPermissionFilter(Component):
+    """ Filtering the stream to remove """
+    implements(ITemplateStreamFilter)    
+    
+    ## ITemplateStreamFilter
+    
+    def filter_stream(self, req, method, filename, stream, data):
+        if not filename == "query.html":
+            self.log.debug('Not a query returning')
+            return stream
+
+        def make_col_helper(field):
+            def column_helper (column_stream):
+                s =  Stream(column_stream)
+                val = s.select('//input/@value').render()
+                if val.lower() != field.lower(): #if we are the field just skip it
+                    #identity stream filter
+                    for kind, data, pos in s:
+                        yield kind, data, pos        
+            return column_helper
+
+        fields = self.config.getlist(csection, 'fields', [])
+        for field in fields:
+            self.log.debug('found : %s' % field)
+            perms = self.config.getlist(csection, '%s.permission' % field, [])
+            self.log.debug('read permission config: %s has %s' % (field, perms))
+            for (perm, denial) in [s.split(":") for s in perms] :
+                perm = perm.upper()
+                self.log.debug('testing permission: %s:%s should act= %s' %
+                               (field, perm, (not req.perm.has_permission(perm) or perm == "ALWAYS")))
+                if (not req.perm.has_permission(perm) or perm == "ALWAYS") and denial.lower() in ["remove","hide"]:
+                    # remove from the list of addable 
+                    stream = stream | Transformer(
+                        '//select[@id="add_filter"]/option[@value="%s"]' % field
+                        ).replace(" ")
+
+                    # remove from the list of columns
+                    stream = stream | Transformer(
+                        '//fieldset[@id="columns"]/div/label'
+                        ).filter(make_col_helper(field))
+                    
+                    #remove from the results table
+                    stream = stream | Transformer(
+                        '//th[@class="%s"]' % field
+                        ).replace(" ")
+                    stream = stream | Transformer(
+                        '//td[@class="%s"]' % field
+                        ).replace(" ")
+
+                    # remove from the filters
+                    stream = stream | Transformer(
+                        '//tr[@class="%s"]' % field
+                        ).replace(" ")
+
+                    
+        return stream

data/lib/subtrac/trac-plugins/timingandestimationplugin/timingandestimationplugin/templates/billing.cs

@@ -0,0 +1,84 @@
+<?cs include "header.cs"?>
+<?cs include "macros.cs"?>
+
+<form method="post" action="<?cs var:billing_info.href ?>" >
+<div id="content" class="billing">
+  <a href="<?cs var:billing_info.usermanual_href ?>" ><?cs var:billing_info.usermanual_title ?></a>
+  <div id="messages" >
+    <?cs each:item = billing_info.messages ?>
+      <div class="message" ><?cs var:item ?></div>
+    <?cs /each ?>
+  </div>
+
+  <table border="0" cellspacing="0" cellpadding="0" class="minorsection">
+    <tr>
+      <td colspan="2" >
+            <div class="minorsection">
+	    <div class="label" >Billing Status:</div>
+              <label for="billable">Billable: </label>
+              <input id="billable" name="billable" type="checkbox" checked="true" /> 
+
+            &nbsp;&nbsp;|&nbsp;&nbsp;
+              <label for="unbillable">Not Billable: </label>
+              <input id="unbillable" name="unbillable" type="checkbox"  /> 
+            </div>
+            <div class="minorsection">
+	    <div class="label" >Status:</div>
+              <label for="new">New: </label>
+              <input id="new" name="new" type="checkbox" checked="true" /> 
+            &nbsp;&nbsp;|&nbsp;&nbsp;
+              <label for="assigned">Assigned: </label>
+              <input id="assigned" name="assigned" type="checkbox" checked="true" /> 
+            &nbsp;&nbsp;|&nbsp;&nbsp;
+              <label for="reopened">Reopened: </label>
+              <input id="reopened" name="reopened" type="checkbox" checked="true" /> 
+            &nbsp;&nbsp;|&nbsp;&nbsp;
+              <label for="closed">Closed: </label>
+              <input id="closed" name="closed" type="checkbox" checked="true" /> 
+            </div>
+
+      </td>
+    </tr><tr class="minorsection">
+      <td class="minorsectionleft" valign="top"><label for="startdate" >Start Date:</label></td>
+      <td class="minorsectionright"><input id="startdate" name="startdate" type="text" /> or:<br />
+          <label for="startbilling" >Choose an old billing date:</label><br />
+          <select id="startbilling" name="startbilling" >
+            <option value="" ></option>
+              <?cs each:item = billing_info.billdates ?>
+                <option value="<?cs var:item.value ?>" ><?cs var:item.text ?></option>
+              <?cs /each ?>
+          </select>
+      </td>
+    </tr><tr class="minorsection">
+      <td class="minorsectionleft" valign="top"><label for="enddate" >End Date:</label>
+      </td>
+      <td class="minorsectionright">
+          <input id="enddate" name="enddate" type="text" /> or:<br />
+          <label for="endbilling" >Choose an old billing date:</label><br />
+          <select id="endbilling" name="endbilling" >
+            <option value="" ></option>
+              <?cs each:item = billing_info.billdates ?>
+                <option value="<?cs var:item.value ?>" ><?cs var:item.text ?></option>
+              <?cs /each ?>
+          </select>
+      </td>
+    </tr>
+  </table>
+  <ul id="reportlinks">
+  <?cs each:report_group = billing_info.reports ?>
+    <li><?cs var:report_group.title ?>
+      <ul>
+        <?cs each:report = report_group.reports ?>
+          <li><a href="" onmouseover="linkify(this, '<?cs var:billing_info.report_base_href ?>/<?cs var:report.id ?>')"  >
+            <?cs var:report.title ?>
+          </a></li>     
+       <?cs /each ?>
+      </ul>
+    </li>
+  <?cs /each ?>
+  </ul>
+  <input type="submit" name="setbillingtime" value="Set Billing Time" onclick="return confirm('Are you sure that you want to create a billed date?')" />&nbsp;
+
+</div>
+</form>
+<?cs include "footer.cs"?>

data/lib/subtrac/trac-plugins/timingandestimationplugin/timingandestimationplugin/templates/billing.html

@@ -0,0 +1,104 @@
+<!DOCTYPE html
+    PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN"
+    "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
+<html xmlns="http://www.w3.org/1999/xhtml"
+      xmlns:py="http://genshi.edgewall.org/"
+      xmlns:xi="http://www.w3.org/2001/XInclude">
+  <xi:include href="layout.html" />
+  <xi:include href="macros.html" />
+
+      <head>
+	<title>Time Tracking Management for Trac.11</title>
+    <script type="text/javascript" src="${chrome.htdocs_location}js/wikitoolbar.js"></script>
+    <script type="text/javascript" py:choose="">
+      $(document).ready(function() {
+          $("div.description").find("h1,h2,h3,h4,h5,h6").addAnchor("Link to this section");
+        });
+    </script>
+      </head>
+
+      <body>
+<form method="post" action="${billing_info.href}"  >
+<div id="content" class="billing">
+  <a href="$billing_info.usermanual_href" >$billing_info.usermanual_title</a>
+  <div id="messages" >
+    <py:for each="item in billing_info.messages" >
+      <div class="message" >$item</div>
+    </py:for>
+  </div>
+
+  <table border="0" cellspacing="0" cellpadding="0" class="minorsection">
+    <tr>
+      <td colspan="2" >
+            <div class="minorsection">
+	    <div class="label" >Billing Status:</div>
+              <label for="billable">Billable: </label>
+              <input id="billable" name="billable" type="checkbox" checked="true" /> 
+
+            &nbsp;&nbsp;|&nbsp;&nbsp;
+              <label for="unbillable">Not Billable: </label>
+              <input id="unbillable" name="unbillable" type="checkbox"  /> 
+            </div>
+
+		  <div class="minorsection">
+		    <div class="label" >Status:</div>
+		    <py:for each="status in statuses" >
+		      <label for="$status">${status.capitalize()}: </label>
+		      <input id="$status" name="$status" type="checkbox" checked="true" /> 
+		    </py:for>
+		    <script>
+		      <py:for each="status in statuses" >
+			   addBillingField("$status", "checkbox", true);
+		      </py:for>
+		    </script>
+		  </div>
+
+      </td>
+    </tr><tr class="minorsection">
+      <td class="minorsectionleft" valign="top"><label for="startdate" >Start Date:</label></td>
+      <td class="minorsectionright"><input id="startdate" name="startdate" type="text" /> or:<br />
+          <label for="startbilling" >Choose an old billing date:</label><br />
+          <select id="startbilling" name="startbilling" >
+            <option value="" ></option>
+              <py:for each="item in billing_info.billdates" >
+                <option value="$item.value" >$item.text</option>
+              </py:for>
+          </select>
+      </td>
+    </tr><tr class="minorsection">
+      <td class="minorsectionleft" valign="top"><label for="enddate" >End Date:</label>
+      </td>
+      <td class="minorsectionright">
+          <input id="enddate" name="enddate" type="text" /> or:<br />
+          <label for="endbilling" >Choose an old billing date:</label><br />
+          <select id="endbilling" name="endbilling" >
+            <option value="" ></option>
+              <py:for each="item in billing_info.billdates" >
+                <option value="$item.value" >$item.text</option>
+              </py:for>
+          </select>
+      </td>
+    </tr>
+  </table>
+  <ul id="reportlinks">
+  <py:for each="key in reports" >
+    <li>${reports[key]["title"]}
+      <ul>
+        <py:for each="report in reports[key]['reports']">
+          <li><a href="" onmouseover="linkify(this, '$billing_info.report_base_href/$report.id')"  >
+            $report.title
+          </a></li>     
+       </py:for>
+      </ul>
+    </li>
+  </py:for>
+
+  </ul>
+  <input 
+   py:if="is_time_admin"
+   type="submit" name="setbillingtime" value="Set Billing Time" onclick="return confirm('Are you sure that you want to create a billed date?')" />&nbsp;
+
+</div>
+</form>
+</body>
+</html>