[PATCH 5/9] py3: "Modernize" code base
Stephen Finucane
stephen.finucane at intel.com
Mon Nov 30 09:10:46 AEDT 2015
Run code through the 'modernize' application to fix Python 3
compatibility while also retaining Python 2 backwards compatibility.
There are some key changes made to the autogenerated code:
* Don't wrap 'items()' in 'list' for for loops - it's not necessary
* Don't wrap 'keys()' in 'list' - just drop 'keys()'
* Use Django's version of six rather than the upstream one
Many of the issues found are based upon the changed definitions of the
map, keys and items functions, along with the removal of the iteritems
function and reduce keyword.
Signed-off-by: Stephen Finucane <stephen.finucane at intel.com>
---
patchwork/bin/parsemail.py | 13 +++++----
patchwork/filters.py | 24 ++++++++---------
patchwork/paginator.py | 1 +
patchwork/parser.py | 6 +++--
patchwork/templatetags/listurl.py | 6 ++---
patchwork/templatetags/syntax.py | 9 ++++---
patchwork/tests/test_bundles.py | 7 ++---
patchwork/tests/test_checks.py | 2 +-
patchwork/tests/test_list.py | 3 ++-
patchwork/tests/test_person.py | 3 ++-
patchwork/tests/test_xmlrpc.py | 4 +--
patchwork/tests/utils.py | 2 +-
patchwork/utils.py | 2 +-
patchwork/views/mail.py | 2 +-
patchwork/views/patch.py | 2 +-
patchwork/views/xmlrpc.py | 55 +++++++++++++++++++++------------------
16 files changed, 77 insertions(+), 64 deletions(-)
diff --git a/patchwork/bin/parsemail.py b/patchwork/bin/parsemail.py
index 866743f..2bc080c 100755
--- a/patchwork/bin/parsemail.py
+++ b/patchwork/bin/parsemail.py
@@ -27,6 +27,7 @@ import datetime
from email import message_from_file
from email.header import Header, decode_header
from email.utils import parsedate_tz, mktime_tz
+from functools import reduce
import logging
import operator
import re
@@ -36,6 +37,8 @@ import django
from django.conf import settings
from django.contrib.auth.models import User
from django.utils.log import AdminEmailHandler
+from django.utils import six
+from django.utils.six.moves import map
from patchwork.models import (
Patch, Project, Person, Comment, State, get_default_initial_patch_state)
@@ -67,7 +70,7 @@ def clean_header(header):
return frag_str.decode(frag_encoding)
return frag_str.decode()
- fragments = map(decode, decode_header(header))
+ fragments = list(map(decode, decode_header(header)))
return normalise_space(u' '.join(fragments))
@@ -161,7 +164,7 @@ def mail_headers(mail):
return reduce(operator.__concat__,
['%s: %s\n' % (k, Header(v, header_name=k,
continuation_ws='\t').encode())
- for (k, v) in mail.items()])
+ for (k, v) in list(mail.items())])
def find_pull_request(content):
@@ -177,7 +180,7 @@ def find_pull_request(content):
def try_decode(payload, charset):
try:
- payload = unicode(payload, charset)
+ payload = six.text_type(payload, charset)
except UnicodeDecodeError:
return None
return payload
@@ -195,7 +198,7 @@ def find_content(project, mail):
payload = part.get_payload(decode=True)
subtype = part.get_content_subtype()
- if not isinstance(payload, unicode):
+ if not isinstance(payload, six.text_type):
charset = part.get_content_charset()
# Check that we have a charset that we understand. Otherwise,
@@ -491,7 +494,7 @@ def main(args):
def list_logging_levels():
"""Give a summary of all available logging levels."""
- return sorted(VERBOSITY_LEVELS.keys(),
+ return sorted(list(VERBOSITY_LEVELS.keys()),
key=lambda x: VERBOSITY_LEVELS[x])
parser.add_argument('infile', nargs='?', type=argparse.FileType('r'),
diff --git a/patchwork/filters.py b/patchwork/filters.py
index a260ef1..4f5a584 100644
--- a/patchwork/filters.py
+++ b/patchwork/filters.py
@@ -19,11 +19,11 @@
from __future__ import absolute_import
-from urllib import quote
-
from django.contrib.auth.models import User
from django.utils.html import escape
from django.utils.safestring import mark_safe
+from django.utils import six
+from django.utils.six.moves.urllib.parse import quote
from patchwork.models import Person, State
@@ -55,7 +55,7 @@ class Filter(object):
pass
def parse(self, dict):
- if self.param not in dict.keys():
+ if self.param not in dict:
return
self._set_key(dict[self.param])
@@ -278,7 +278,7 @@ class ArchiveFilter(Filter):
def _set_key(self, str):
self.archive_state = False
self.applied = True
- for (k, v) in self.param_map.iteritems():
+ for (k, v) in self.param_map.items():
if str == v:
self.archive_state = k
if self.archive_state == None:
@@ -411,7 +411,7 @@ filterclasses = [SubmitterFilter, \
class Filters:
def __init__(self, request):
- self._filters = map(lambda c: c(self), filterclasses)
+ self._filters = [c(self) for c in filterclasses]
self.dict = request.GET
self.project = None
@@ -441,29 +441,27 @@ class Filters:
def querystring(self, remove = None):
params = dict(self.params())
- for (k, v) in self.dict.iteritems():
+ for (k, v) in self.dict.items():
if k not in params:
params[k] = v
if remove is not None:
- if remove.param in params.keys():
+ if remove.param in list(params.keys()):
del params[remove.param]
- pairs = params.iteritems()
-
def sanitise(s):
- if not isinstance(s, basestring):
- s = unicode(s)
+ if not isinstance(s, six.string_types):
+ s = six.text_type(s)
return quote(s.encode('utf-8'))
return '?' + '&'.join(['%s=%s' % (sanitise(k), sanitise(v))
- for (k, v) in pairs])
+ for (k, v) in list(params.items())])
def querystring_without_filter(self, filter):
return self.querystring(filter)
def applied_filters(self):
- return filter(lambda x: x.applied, self._filters)
+ return [x for x in self._filters if x.applied]
def available_filters(self):
return self._filters
diff --git a/patchwork/paginator.py b/patchwork/paginator.py
index dbbcbaa..310aac5 100644
--- a/patchwork/paginator.py
+++ b/patchwork/paginator.py
@@ -21,6 +21,7 @@ from __future__ import absolute_import
from django.conf import settings
from django.core import paginator
+from django.utils.six.moves import range
DEFAULT_PATCHES_PER_PAGE = 100
diff --git a/patchwork/parser.py b/patchwork/parser.py
index a7fa89c..84f704a 100644
--- a/patchwork/parser.py
+++ b/patchwork/parser.py
@@ -26,6 +26,8 @@ from collections import Counter
import hashlib
import re
+from django.utils.six.moves import map
+
_hunk_re = re.compile('^\@\@ -\d+(?:,(\d+))? \+\d+(?:,(\d+))? \@\@')
_filename_re = re.compile('^(---|\+\+\+) (\S+)')
@@ -113,7 +115,7 @@ def parse_patch(text):
return 1
return int(x)
- lc = map(fn, match.groups())
+ lc = list(map(fn, match.groups()))
state = 4
patchbuf += buf + line
@@ -217,7 +219,7 @@ def hash_patch(str):
if not x:
return 1
return int(x)
- line_nos = map(fn, hunk_match.groups())
+ line_nos = list(map(fn, hunk_match.groups()))
line = '@@ -%d +%d @@' % tuple(line_nos)
elif line[0] in prefixes:
diff --git a/patchwork/templatetags/listurl.py b/patchwork/templatetags/listurl.py
index 118fd58..7e0e3d7 100644
--- a/patchwork/templatetags/listurl.py
+++ b/patchwork/templatetags/listurl.py
@@ -39,7 +39,7 @@ class ListURLNode(template.defaulttags.URLNode):
def __init__(self, kwargs):
super(ListURLNode, self).__init__(None, [], {}, False)
self.params = {}
- for (k, v) in kwargs.iteritems():
+ for (k, v) in kwargs.items():
if k in list_params:
self.params[k] = v
@@ -68,14 +68,14 @@ class ListURLNode(template.defaulttags.URLNode):
except Exception:
pass
- for (k, v) in self.params.iteritems():
+ for (k, v) in self.params.items():
params[smart_str(k, 'ascii')] = v.resolve(context)
if not params:
return str
return str + '?' + '&'.join(
- ['%s=%s' % (k, escape(v)) for (k, v) in params.iteritems()])
+ ['%s=%s' % (k, escape(v)) for (k, v) in list(params.items())])
@register.tag
diff --git a/patchwork/templatetags/syntax.py b/patchwork/templatetags/syntax.py
index 3988f61..9d06044 100644
--- a/patchwork/templatetags/syntax.py
+++ b/patchwork/templatetags/syntax.py
@@ -24,6 +24,7 @@ import re
from django import template
from django.utils.html import escape
from django.utils.safestring import mark_safe
+from django.utils.six.moves import map
register = template.Library()
@@ -32,17 +33,17 @@ def _compile(t):
(r, str) = t
return (re.compile(r, re.M | re.I), str)
-_patch_span_res = map(_compile, [
+_patch_span_res = list(map(_compile, [
('^(Index:?|diff|\-\-\-|\+\+\+|\*\*\*) .*$', 'p_header'),
('^\+.*$', 'p_add'),
('^-.*$', 'p_del'),
('^!.*$', 'p_mod'),
- ])
+ ]))
_patch_chunk_re = \
re.compile('^(@@ \-\d+(?:,\d+)? \+\d+(?:,\d+)? @@)(.*)$', re.M | re.I)
-_comment_span_res = map(_compile, [
+_comment_span_res = list(map(_compile, [
('^\s*Signed-off-by: .*$', 'signed-off-by'),
('^\s*Acked-by: .*$', 'acked-by'),
('^\s*Nacked-by: .*$', 'nacked-by'),
@@ -50,7 +51,7 @@ _comment_span_res = map(_compile, [
('^\s*Reviewed-by: .*$', 'reviewed-by'),
('^\s*From: .*$', 'from'),
('^\s*>.*$', 'quote'),
- ])
+ ]))
_span = '<span class="%s">%s</span>'
diff --git a/patchwork/tests/test_bundles.py b/patchwork/tests/test_bundles.py
index 233e129..4b982a5 100644
--- a/patchwork/tests/test_bundles.py
+++ b/patchwork/tests/test_bundles.py
@@ -29,6 +29,7 @@ from django.utils.http import urlencode
from patchwork.models import Patch, Bundle, BundlePatch, Person
from patchwork.tests.utils import defaults, create_user, find_in_context
+from django.utils.six.moves import range, zip
def bundle_url(bundle):
return '/bundle/%s/%s/' % (bundle.owner.username, bundle.name)
@@ -573,11 +574,11 @@ class BundleInitialOrderTest(BundleTestBase):
bundle.delete()
def testBundleForwardOrder(self):
- ids = map(lambda p: p.id, self.patches)
+ ids = [p.id for p in self.patches]
self._testOrder(ids, self.patches)
def testBundleReverseOrder(self):
- ids = map(lambda p: p.id, self.patches)
+ ids = [p.id for p in self.patches]
ids.reverse()
self._testOrder(ids, self.patches)
@@ -611,7 +612,7 @@ class BundleReorderTest(BundleTestBase):
# check if order field is still sequential:
order_numbers = [ bp.order for bp in bps ]
- expected_order = range(1, len(neworder)+1) # [1 ... len(neworder)]
+ expected_order = list(range(1, len(neworder)+1)) # [1 ... len(neworder)]
self.failUnlessEqual(order_numbers, expected_order)
def testBundleReorderAll(self):
diff --git a/patchwork/tests/test_checks.py b/patchwork/tests/test_checks.py
index 8cf997b..d3bae77 100644
--- a/patchwork/tests/test_checks.py
+++ b/patchwork/tests/test_checks.py
@@ -81,7 +81,7 @@ class PatchChecksTest(TransactionTestCase):
self.assertEqual(self.patch.check_set.count(), total)
- for state in state_counts.keys():
+ for state in state_counts:
self.assertEqual(counts[state], state_counts[state])
# also check the ones we didn't explicitly state
diff --git a/patchwork/tests/test_list.py b/patchwork/tests/test_list.py
index 69c3d24..0f23d13 100644
--- a/patchwork/tests/test_list.py
+++ b/patchwork/tests/test_list.py
@@ -28,6 +28,7 @@ import unittest
from django.core.urlresolvers import reverse
from django.test.client import Client
from django.test import TestCase
+from django.utils.six.moves import zip
from patchwork.models import Person, Patch
from patchwork.tests.utils import defaults, create_user, find_in_context
@@ -83,7 +84,7 @@ class PatchOrderTest(TestCase):
ids = self._extract_patch_ids(response)
self.assertTrue(bool(ids))
patches = [ Patch.objects.get(id = i) for i in ids ]
- pairs = zip(patches, patches[1:])
+ pairs = list(zip(patches, patches[1:]))
[ test_fn(p1, p2) for (p1, p2) in pairs ]
def testDateOrder(self):
diff --git a/patchwork/tests/test_person.py b/patchwork/tests/test_person.py
index 7d82ce8..ef35da6 100644
--- a/patchwork/tests/test_person.py
+++ b/patchwork/tests/test_person.py
@@ -24,6 +24,7 @@ import unittest
from django.test.client import Client
from django.test import TestCase
+from django.utils.six.moves import map, range
from patchwork.models import EmailConfirmation, Person, Bundle
@@ -33,7 +34,7 @@ class SubmitterCompletionTest(TestCase):
Person(name = "Test Name", email = "test1 at example.com"),
Person(email = "test2 at example.com"),
]
- map(lambda p: p.save(), self.people)
+ list(map(lambda p: p.save(), self.people))
def testNameComplete(self):
response = self.client.get('/submitter/', {'q': 'name'})
diff --git a/patchwork/tests/test_xmlrpc.py b/patchwork/tests/test_xmlrpc.py
index 06522ff..0d032ed 100644
--- a/patchwork/tests/test_xmlrpc.py
+++ b/patchwork/tests/test_xmlrpc.py
@@ -18,11 +18,11 @@
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
import unittest
-import xmlrpclib
from django.conf import settings
from django.core.urlresolvers import reverse
from django.test import LiveServerTestCase
+from django.utils.six.moves import xmlrpc_client
from patchwork.models import Person, Patch
from patchwork.tests.utils import defaults
@@ -36,7 +36,7 @@ class XMLRPCTest(LiveServerTestCase):
def setUp(self):
self.url = (self.live_server_url +
reverse('patchwork.views.xmlrpc.xmlrpc'))
- self.rpc = xmlrpclib.Server(self.url)
+ self.rpc = xmlrpc_client.Server(self.url)
def testGetRedirect(self):
response = self.client.patch(self.url)
diff --git a/patchwork/tests/utils.py b/patchwork/tests/utils.py
index 061a64e..fb32493 100644
--- a/patchwork/tests/utils.py
+++ b/patchwork/tests/utils.py
@@ -98,7 +98,7 @@ def read_patch(filename, encoding = None):
if encoding is not None:
f = codecs.open(file_path, encoding = encoding)
else:
- f = file(file_path)
+ f = open(file_path)
return f.read()
diff --git a/patchwork/utils.py b/patchwork/utils.py
index f151c12..0fbf317 100644
--- a/patchwork/utils.py
+++ b/patchwork/utils.py
@@ -76,7 +76,7 @@ class Order(object):
str = str[1:]
reversed = True
- if str not in self.order_map.keys():
+ if str not in self.order_map:
return
self.order = str
diff --git a/patchwork/views/mail.py b/patchwork/views/mail.py
index 5e38bba..af8249a 100644
--- a/patchwork/views/mail.py
+++ b/patchwork/views/mail.py
@@ -108,7 +108,7 @@ def optinout(request, action, description):
conf_settings.DEFAULT_FROM_EMAIL, [email])
context['email'] = mail
context['email_sent'] = True
- except Exception, ex:
+ except Exception as ex:
context['error'] = 'An error occurred during confirmation . ' + \
'Please try again later.'
context['admins'] = conf_settings.ADMINS
diff --git a/patchwork/views/patch.py b/patchwork/views/patch.py
index 3c43cd7..cb017fb 100644
--- a/patchwork/views/patch.py
+++ b/patchwork/views/patch.py
@@ -64,7 +64,7 @@ def patch(request, patch_id):
bundle.append_patch(patch)
bundle.save()
context.add_message('Patch added to bundle "%s"' % bundle.name)
- except Exception, ex:
+ except Exception as ex:
context.add_message("Couldn't add patch '%s' to bundle %s: %s" \
% (patch.name, bundle.name, ex.message))
diff --git a/patchwork/views/xmlrpc.py b/patchwork/views/xmlrpc.py
index 9d7ef07..64909e2 100644
--- a/patchwork/views/xmlrpc.py
+++ b/patchwork/views/xmlrpc.py
@@ -23,16 +23,21 @@
from __future__ import absolute_import
import base64
-from DocXMLRPCServer import XMLRPCDocGenerator
-from SimpleXMLRPCServer import SimpleXMLRPCDispatcher
+# NOTE(stephenfin) six does not seem to support this
+try:
+ from DocXMLRPCServer import XMLRPCDocGenerator
+except ImportError:
+ from xmlrpc.server import XMLRPCDocGenerator
import sys
-import xmlrpclib
from django.core import urlresolvers
from django.contrib.auth import authenticate
from django.http import (
HttpResponse, HttpResponseRedirect, HttpResponseServerError)
from django.views.decorators.csrf import csrf_exempt
+from django.utils import six
+from django.utils.six.moves import map, xmlrpc_client
+from django.utils.six.moves.xmlrpc_server import SimpleXMLRPCDispatcher
from patchwork.models import Patch, Project, Person, State, Check
from patchwork.views import patch_to_mbox
@@ -52,7 +57,7 @@ class PatchworkXMLRPCDispatcher(SimpleXMLRPCDispatcher,
def _dumps(obj, *args, **kwargs):
kwargs['allow_none'] = self.allow_none
kwargs['encoding'] = self.encoding
- return xmlrpclib.dumps(obj, *args, **kwargs)
+ return xmlrpc_client.dumps(obj, *args, **kwargs)
self.dumps = _dumps
@@ -90,7 +95,7 @@ class PatchworkXMLRPCDispatcher(SimpleXMLRPCDispatcher,
return authenticate(username=username, password=password)
def _dispatch(self, request, method, params):
- if method not in self.func_map.keys():
+ if method not in list(self.func_map.keys()):
raise Exception('method "%s" is not supported' % method)
auth_required, fn = self.func_map[method]
@@ -106,18 +111,18 @@ class PatchworkXMLRPCDispatcher(SimpleXMLRPCDispatcher,
def _marshaled_dispatch(self, request):
try:
- params, method = xmlrpclib.loads(request.body)
+ params, method = six.moves.xmlrpc_client.loads(request.body)
response = self._dispatch(request, method, params)
# wrap response in a singleton tuple
response = (response,)
response = self.dumps(response, methodresponse=1)
- except xmlrpclib.Fault, fault:
+ except six.moves.xmlrpc_client.Fault as fault:
response = self.dumps(fault)
except:
# report exception back to server
response = self.dumps(
- xmlrpclib.Fault(1, '%s:%s' % (sys.exc_type, sys.exc_value)),
+ six.moves.xmlrpc_client.Fault(1, '%s:%s' % (sys.exc_info()[0], sys.exc_info()[1])),
)
return response
@@ -225,7 +230,7 @@ def person_to_dict(obj):
'id': obj.id,
'email': obj.email,
'name': name,
- 'user': unicode(obj.user).encode('utf-8'),
+ 'user': six.text_type(obj.user).encode('utf-8'),
}
@@ -261,18 +266,18 @@ def patch_to_dict(obj):
"""
return {
'id': obj.id,
- 'date': unicode(obj.date).encode('utf-8'),
+ 'date': six.text_type(obj.date).encode('utf-8'),
'filename': obj.filename(),
'msgid': obj.msgid,
'name': obj.name,
- 'project': unicode(obj.project).encode('utf-8'),
+ 'project': six.text_type(obj.project).encode('utf-8'),
'project_id': obj.project_id,
- 'state': unicode(obj.state).encode('utf-8'),
+ 'state': six.text_type(obj.state).encode('utf-8'),
'state_id': obj.state_id,
'archived': obj.archived,
- 'submitter': unicode(obj.submitter).encode('utf-8'),
+ 'submitter': six.text_type(obj.submitter).encode('utf-8'),
'submitter_id': obj.submitter_id,
- 'delegate': unicode(obj.delegate).encode('utf-8'),
+ 'delegate': six.text_type(obj.delegate).encode('utf-8'),
'delegate_id': max(obj.delegate_id, 0),
'commit_ref': max(obj.commit_ref, ''),
}
@@ -333,10 +338,10 @@ def check_to_dict(obj):
object which is OK to send to the client."""
return {
'id': obj.id,
- 'date': unicode(obj.date).encode('utf-8'),
- 'patch': unicode(obj.patch).encode('utf-8'),
+ 'date': six.text_type(obj.date).encode('utf-8'),
+ 'patch': six.text_type(obj.patch).encode('utf-8'),
'patch_id': obj.patch_id,
- 'user': unicode(obj.user).encode('utf-8'),
+ 'user': six.text_type(obj.user).encode('utf-8'),
'user_id': obj.user_id,
'state': obj.get_state_display(),
'target_url': obj.target_url,
@@ -401,7 +406,7 @@ def project_list(search_str=None, max_count=0):
if max_count > 0:
return map(project_to_dict, projects)[:max_count]
else:
- return map(project_to_dict, projects)
+ return list(map(project_to_dict, projects))
except Project.DoesNotExist:
return []
@@ -453,7 +458,7 @@ def person_list(search_str=None, max_count=0):
if max_count > 0:
return map(person_to_dict, people)[:max_count]
else:
- return map(person_to_dict, people)
+ return list(map(person_to_dict, people))
except Person.DoesNotExist:
return []
@@ -594,9 +599,9 @@ def patch_list(filt=None):
patches = Patch.objects.filter(**dfilter)
if max_count > 0:
- return map(patch_to_dict, patches[:max_count])
+ return [patch_to_dict(patch) for patch in patches[:max_count]]
else:
- return map(patch_to_dict, patches)
+ return [patch_to_dict(patch) for patch in patches]
except Patch.DoesNotExist:
return []
@@ -746,7 +751,7 @@ def patch_set(user, patch_id, params):
if not patch.is_editable(user):
raise Exception('No permissions to edit this patch')
- for (k, v) in params.iteritems():
+ for (k, v) in params.items():
if k not in ok_params:
continue
@@ -789,7 +794,7 @@ def state_list(search_str=None, max_count=0):
if max_count > 0:
return map(state_to_dict, states)[:max_count]
else:
- return map(state_to_dict, states)
+ return list(map(state_to_dict, states))
except State.DoesNotExist:
return []
@@ -915,9 +920,9 @@ def check_list(filt=None):
checks = Check.objects.filter(**dfilter)
if max_count > 0:
- return map(check_to_dict, checks[:max_count])
+ return list(map(check_to_dict, checks[:max_count]))
else:
- return map(check_to_dict, checks)
+ return list(map(check_to_dict, checks))
except Check.DoesNotExist:
return []
--
2.0.0
More information about the Patchwork
mailing list