root/peerreviewplugin/branches/2.2-dev/codereview/peerReviewNew.py

#       
# Copyright (C) 2005-2006 Team5 
# All rights reserved.  
#       
# This software is licensed as described in the file COPYING.txt, which 
# you should have received as part of this distribution.        
#       
# Author: Team5
#

# Provides functionality to create a new code review.
# Works with peerReviewNew.cs

from trac.core import *
from trac.web.chrome import INavigationContributor,ITemplateProvider
from trac.web.main import IRequestHandler
from trac import util
from trac.util import escape
from codereview.CodeReviewStruct import *
from codereview.dbBackend import *
from codereview.ReviewerStruct import *
from trac.web.chrome import add_stylesheet
import time
import itertools

class UserbaseModule(Component):
    implements(IRequestHandler, ITemplateProvider, INavigationContributor)
    

    # INavigationContributor methods
    def get_active_navigation_item(self, req):
        return 'peerReviewMain'

    def get_navigation_items(self, req):
        return [] 
       

    # IRequestHandler methods
    def match_request(self, req):
        return req.path_info == '/peerReviewNew'
                                        

    def process_request(self, req):
        
        data = {}

        if req.perm.has_permission('CODE_REVIEW_MGR'):
            data['manager'] = 1
        else:
            req.perm.assert_permission('CODE_REVIEW_DEV')
            data['manager'] = 0

        data['main'] = "no"
        data['create'] = "yes"
        data['search'] = "no"
        data['option'] = "no"

        db = self.env.get_db_cnx()
        dbBack = dbBackend(db)
        allUsers = dbBack.getPossibleUsers()

        reviewID = req.args.get('resubmit')
        data['oldid'] = -1

        # if we tried resubmitting and the reviewID is not a valid number or not a valid code review, error
        if reviewID != None and (not reviewID.isdigit() or dbBack.getCodeReviewsByID(reviewID) == None):
            data['error.type'] = "TracError"
            data['error.title'] = "Resubmit ID error"
            data['error.message'] = "Invalid resubmit ID supplied - unable to load page correctly."
            return 'error.html', data, None

        # if we are resubmitting a code review and we are the author or the manager
        if reviewID != None and (dbBack.getCodeReviewsByID(reviewID).Author == util.get_reporter_id(req) or req.perm.has_permission('CODE_REVIEW_MGR')):
            review = dbBack.getCodeReviewsByID(reviewID)
            data['new'] = "no"
            data['oldid'] = reviewID
            # get code review data and populate
            userStructs = dbBack.getReviewers(reviewID)
            returnUsers = ""
            popUsers = []
            for struct in userStructs:
                returnUsers+=struct.Reviewer + "#"
                popUsers.append(struct.Reviewer)

            files = dbBack.getReviewFiles(reviewID)
            returnFiles = ""
            popFiles = []
            # Set up the file information
            for struct in files:
                returnFiles+= "%s,%s,%s, %s#" % (struct.Path, struct.Version, struct.LineStart, struct.LineEnd ) 
                tempFiles = []
                tempFiles.append(struct.Path)
                tempFiles.append(struct.Version)
                tempFiles.append(struct.LineStart)
                tempFiles.append(struct.LineEnd)
                popFiles.append(tempFiles);

            data['files'] = returnFiles
            data['name'] = review.Name
            data['notes'] = review.Notes
            data['reviewers'] = returnUsers
            data['prevUsers'] = popUsers
            data['prevFiles'] = popFiles

            # Figure out the users that were not included
            # in the previous code review so that they can be
            # added to the dropdown to select more users
            # (only check if all users were not included in previous code review)
            notUsers = []
            if len(popUsers) != len(allUsers): 
                for user in allUsers:
                    match = "no"
                    for candidate in popUsers:
                        if candidate == user:
                            match = "yes"
                            break
                    if match == "no":
                        notUsers.append(user)
                data['notPrevUsers'] = notUsers
                data['emptyList'] = 0
            else:
                data['notPrevUsers'] = []
                data['emptyList'] = 1
        #if we resubmitting a code review, and are neiter the author and the manager
        elif reviewID != None and not dbBack.getCodeReviewsByID(reviewID).Author == util.get_reporter_id(req) and not req.perm.has_permission('CODE_REVIEW_MGR'):
            data['error.type'] = "TracError"
            data['error.title'] = "Access error"
            data['error.message'] = "You need to be a manager or the author of this code review to resubmit it."
            return 'error.html', data, None
        #if we are not resubmitting
        else:
            if req.args.get('reqAction') == 'createCodeReview':
                oldid = req.args.get('oldid')
                if oldid != None:
                    review = dbBack.getCodeReviewsByID(oldid)
                    review.Status = "Closed"
                    review.save(db)
                returnid = self.createCodeReview(req)
                #If no errors then redirect to the viewCodeReview page
                req.redirect(self.env.href.peerReviewView() + '?Review=' + str(returnid))
            else:
                data['new'] = "yes"

        if data['new'] == "yes":
            data['reviewersSelectedValue'] = {'value': ''}
            data['filesSelectedValue'] = {'value': ''} 
        else:
            data['reviewersSelectedValue'] = {'value': returnUsers}
            data['filesSelectedValue'] = {'value': returnFiles} 

        data['users'] = allUsers
        data['cycle'] = itertools.cycle

        return 'peerReviewNew.html', data, None

    # ITemplateProvider methods
    def get_templates_dirs(self):
        """
        Return the absolute path of the directory containing the provided
        ClearSilver templates.
        """
        from pkg_resources import resource_filename
        return [resource_filename(__name__, 'templates')]


    # Needed to be filled out based on interface
    def get_htdocs_dirs(self):
        from pkg_resources import resource_filename
        return [('hw', resource_filename(__name__, 'htdocs'))]


    # Takes the information given when the page is posted
    # and creates a new code review struct in the database
    # and populates it with the information.  Also creates
    # new reviewer structs and file structs for the review.
    def createCodeReview(self, req):
        struct = CodeReviewStruct(None)
        struct.Author = util.get_reporter_id(req)
        struct.Status = 'Open for review'
        struct.DateCreate = int(time.time())
        struct.Name = req.args.get('Name')
        struct.Notes = req.args.get('Notes')
        id = struct.save(self.env.get_db_cnx())
        self.log.debug('BEN %s', id)       
        # loop here through all the reviewers
        # and create new reviewer structs based on them
        string = req.args.get('ReviewersSelected')
        tokens = string.split('#')
        for token in tokens:
            if token != "":
                struct = ReviewerStruct(None)
                struct.IDReview = id
                struct.Reviewer = token
                struct.Status = 0
                struct.Vote = "-1"
                struct.save(self.env.get_db_cnx())

        # loop here through all included files
        # and create new file structs based on them
        files = req.args.get('FilesSelected')
        items = files.split('#')
        for item in items:
            if item != "":
                segment = item.split(',')
                struct = ReviewFileStruct(None)
                struct.IDReview = id
                struct.Path = segment[0]
                struct.Version = segment[1]
                struct.LineStart = segment[2]
                struct.LineEnd = segment[3]
                struct.save(self.env.get_db_cnx())

        return id