git.ozlabs.org Git - patchwork/blob - patchwork/views/__init__.py

   1 # Patchwork - automated patch tracking system
   2 # Copyright (C) 2008 Jeremy Kerr <jk@ozlabs.org>
   3 #
   4 # This file is part of the Patchwork package.
   5 #
   6 # Patchwork is free software; you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation; either version 2 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # Patchwork is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 # GNU General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with Patchwork; if not, write to the Free Software
  18 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19
  20
  21 from base import *
  22 from patchwork.utils import Order, get_patch_ids, bundle_actions, set_bundle
  23 from patchwork.paginator import Paginator
  24 from patchwork.forms import MultiplePatchForm
  25 from patchwork.models import Comment
  26 import re
  27 import datetime
  28
  29 try:
  30     from email.mime.nonmultipart import MIMENonMultipart
  31     from email.encoders import encode_7or8bit
  32     from email.parser import HeaderParser
  33     from email.header import Header
  34     import email.utils
  35 except ImportError:
  36     # Python 2.4 compatibility
  37     from email.MIMENonMultipart import MIMENonMultipart
  38     from email.Encoders import encode_7or8bit
  39     from email.Parser import HeaderParser
  40     from email.Header import Header
  41     import email.Utils
  42     email.utils = email.Utils
  43
  44 def generic_list(request, project, view,
  45         view_args = {}, filter_settings = [], patches = None,
  46         editable_order = False):
  47
  48     context = PatchworkRequestContext(request,
  49             list_view = view,
  50             list_view_params = view_args)
  51
  52     context.project = project
  53     order = Order(request.REQUEST.get('order'), editable = editable_order)
  54
  55     # Explicitly set data to None because request.POST will be an empty dict
  56     # when the form is not submitted, but passing a non-None data argument to
  57     # a forms.Form will make it bound and we don't want that to happen unless
  58     # there's been a form submission.
  59     data = None
  60     if request.method == 'POST':
  61         data = request.POST
  62     user = request.user
  63     properties_form = None
  64     if project.is_editable(user):
  65
  66         # we only pass the post data to the MultiplePatchForm if that was
  67         # the actual form submitted
  68         data_tmp = None
  69         if data and data.get('form', '') == 'patchlistform':
  70             data_tmp = data
  71
  72         properties_form = MultiplePatchForm(project, data = data_tmp)
  73
  74     if request.method == 'POST' and data.get('form') == 'patchlistform':
  75         action = data.get('action', '').lower()
  76
  77         # special case: the user may have hit enter in the 'create bundle'
  78         # text field, so if non-empty, assume the create action:
  79         if data.get('bundle_name', False):
  80             action = 'create'
  81
  82         ps = Patch.objects.filter(id__in = get_patch_ids(data))
  83
  84         if action in bundle_actions:
  85             errors = set_bundle(user, project, action, data, ps, context)
  86
  87         elif properties_form and action == properties_form.action:
  88             errors = process_multiplepatch_form(properties_form, user,
  89                                                 action, ps, context)
  90         else:
  91             errors = []
  92
  93         if errors:
  94             context['errors'] = errors
  95
  96     for (filterclass, setting) in filter_settings:
  97         if isinstance(setting, dict):
  98             context.filters.set_status(filterclass, **setting)
  99         elif isinstance(setting, list):
 100             context.filters.set_status(filterclass, *setting)
 101         else:
 102             context.filters.set_status(filterclass, setting)
 103
 104     if patches is None:
 105         patches = Patch.objects.filter(project=project)
 106
 107     patches = context.filters.apply(patches)
 108     if not editable_order:
 109         patches = order.apply(patches)
 110
 111     # we don't need the content or headers for a list; they're text fields
 112     # that can potentially contain a lot of data
 113     patches = patches.defer('content', 'headers')
 114
 115     # but we will need to follow the state and submitter relations for
 116     # rendering the list template
 117     patches = patches.select_related('state', 'submitter')
 118
 119     paginator = Paginator(request, patches)
 120
 121     context.update({
 122             'page':             paginator.current_page,
 123             'patchform':        properties_form,
 124             'project':          project,
 125             'order':            order,
 126             })
 127
 128     return context
 129
 130
 131 def process_multiplepatch_form(form, user, action, patches, context):
 132     errors = []
 133     if not form.is_valid() or action != form.action:
 134         return ['The submitted form data was invalid']
 135
 136     if len(patches) == 0:
 137         context.add_message("No patches selected; nothing updated")
 138         return errors
 139
 140     changed_patches = 0
 141     for patch in patches:
 142         if not patch.is_editable(user):
 143             errors.append("You don't have permissions to edit patch '%s'"
 144                             % patch.name)
 145             continue
 146
 147         changed_patches += 1
 148         form.save(patch)
 149
 150     if changed_patches == 1:
 151         context.add_message("1 patch updated")
 152     elif changed_patches > 1:
 153         context.add_message("%d patches updated" % changed_patches)
 154     else:
 155         context.add_message("No patches updated")
 156
 157     return errors
 158
 159 class PatchMbox(MIMENonMultipart):
 160     patch_charset = 'utf-8'
 161     def __init__(self, _text):
 162         MIMENonMultipart.__init__(self, 'text', 'plain',
 163                         **{'charset': self.patch_charset})
 164         self.set_payload(_text.encode(self.patch_charset))
 165         encode_7or8bit(self)
 166
 167 def patch_to_mbox(patch):
 168     postscript_re = re.compile('\n-{2,3} ?\n')
 169
 170     comment = None
 171     try:
 172         comment = Comment.objects.get(patch = patch, msgid = patch.msgid)
 173     except Exception:
 174         pass
 175
 176     body = ''
 177     if comment:
 178         body = comment.content.strip() + "\n"
 179
 180     parts = postscript_re.split(body, 1)
 181     if len(parts) == 2:
 182         (body, postscript) = parts
 183         body = body.strip() + "\n"
 184         postscript = postscript.rstrip()
 185     else:
 186         postscript = ''
 187
 188     for comment in Comment.objects.filter(patch = patch) \
 189             .exclude(msgid = patch.msgid):
 190         body += comment.patch_responses()
 191
 192     if postscript:
 193         body += '---\n' + postscript + '\n'
 194
 195     if patch.content:
 196         body += '\n' + patch.content
 197
 198     delta = patch.date - datetime.datetime.utcfromtimestamp(0)
 199     utc_timestamp = delta.seconds + delta.days*24*3600
 200
 201     mail = PatchMbox(body)
 202     mail['Subject'] = patch.name
 203     mail['From'] = email.utils.formataddr((
 204                     str(Header(patch.submitter.name, mail.patch_charset)),
 205                     patch.submitter.email))
 206     mail['X-Patchwork-Id'] = str(patch.id)
 207     mail['Message-Id'] = patch.msgid
 208     mail.set_unixfrom('From patchwork ' + patch.date.ctime())
 209
 210
 211     copied_headers = ['To', 'Cc', 'Date']
 212     orig_headers = HeaderParser().parsestr(str(patch.headers))
 213     for header in copied_headers:
 214         if header in orig_headers:
 215             mail[header] = orig_headers[header]
 216
 217     if 'Date' not in mail:
 218         mail['Date'] = email.utils.formatdate(utc_timestamp)
 219
 220     return mail