Merge pull request #312 from sigmavirus24/bug/311

Fix #311. Add regex to check for field assignment

Author: florentx

pep8.py

@@ -112,6 +112,7 @@
 KEYWORD_REGEX = re.compile(r'(\s*)\b(?:%s)\b(\s*)' % r'|'.join(KEYWORDS))
 OPERATOR_REGEX = re.compile(r'(?:[^,\s])(\s*)(?:[-+*/|!<=>%&^]+)(\s*)')
 LAMBDA_REGEX = re.compile(r'\blambda\b')
+IDENTIFIER_REGEX = re.compile('\s*[^.[\]]+\s=')
 HUNK_REGEX = re.compile(r'^@@ -\d+(?:,\d+)? \+(\d+)(?:,(\d+))? @@.*$')
 
 # Work around Python < 2.6 behaviour, which does not generate NL after
@@ -923,7 +924,9 @@ def compound_statements(logical_line):
              before.count('[') <= before.count(']') and   # [1:2] (slice)
              before.count('(') <= before.count(')'))):    # (annotation)
             if LAMBDA_REGEX.search(before):
-                yield 0, "E731 do not assign a lambda expression, use a def"
+                if IDENTIFIER_REGEX.match(before):
+                    yield 0, ("E731 do not assign a lambda expression, use a"
+                              " def")
                 break
             if before.startswith('def '):
                 yield 0, "E704 multiple statements on one line (def)"

testsuite/E73.py

@@ -5,3 +5,14 @@
 #: E731:2:5
 while False:
     this = lambda y, z: 2 * x
+#: Okay
+f = object()
+f.method = lambda: 'Method'
+#: Okay
+f = {}
+f['a'] = lambda x: x ** 2
+#: Okay
+f = []
+f.append(lambda x: x ** 2)
+#: Okay
+lambda: 'no-op'