1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
# Copyright 2015 Google Inc. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Join logical lines together.
Determine how many lines can be joined into one line. For instance, we could
join these statements into one line:
if a == 42:
continue
like this:
if a == 42: continue
There are a few restrictions:
1. The lines should have been joined in the original source.
2. The joined lines must not go over the column boundary if placed on the same
line.
3. They need to be very simple statements.
Note: Because we don't allow the use of a semicolon to separate statements, it
follows that there can only be at most two lines to join.
"""
from yapf.yapflib import style
_CLASS_OR_FUNC = frozenset({'def', 'class'})
def CanMergeMultipleLines(lines, last_was_merged=False):
"""Determine if multiple lines can be joined into one.
Arguments:
lines: (list of LogicalLine) This is a splice of LogicalLines from the full
code base.
last_was_merged: (bool) The last line was merged.
Returns:
True if two consecutive lines can be joined together. In reality, this will
only happen if two consecutive lines can be joined, due to the style guide.
"""
# The indentation amount for the starting line (number of spaces).
indent_amt = lines[0].depth * style.Get('INDENT_WIDTH')
if len(lines) == 1 or indent_amt > style.Get('COLUMN_LIMIT'):
return False
if (len(lines) >= 3 and lines[2].depth >= lines[1].depth and
lines[0].depth != lines[2].depth):
# If lines[2]'s depth is greater than or equal to line[1]'s depth, we're not
# looking at a single statement (e.g., if-then, while, etc.). A following
# line with the same depth as the first line isn't part of the lines we
# would want to combine.
return False # Don't merge more than two lines together.
if lines[0].first.value in _CLASS_OR_FUNC:
# Don't join lines onto the starting line of a class or function.
return False
limit = style.Get('COLUMN_LIMIT') - indent_amt
if lines[0].last.total_length < limit:
limit -= lines[0].last.total_length
if lines[0].first.value == 'if':
return _CanMergeLineIntoIfStatement(lines, limit)
if last_was_merged and lines[0].first.value in {'elif', 'else'}:
return _CanMergeLineIntoIfStatement(lines, limit)
# TODO(morbo): Other control statements?
return False
def _CanMergeLineIntoIfStatement(lines, limit):
"""Determine if we can merge a short if-then statement into one line.
Two lines of an if-then statement can be merged if they were that way in the
original source, fit on the line without going over the column limit, and are
considered "simple" statements --- typically statements like 'pass',
'continue', and 'break'.
Arguments:
lines: (list of LogicalLine) The lines we are wanting to merge.
limit: (int) The amount of space remaining on the line.
Returns:
True if the lines can be merged, False otherwise.
"""
if len(lines[1].tokens) == 1 and lines[1].last.is_multiline_string:
# This might be part of a multiline shebang.
return True
if lines[0].lineno != lines[1].lineno:
# Don't merge lines if the original lines weren't merged.
return False
if lines[1].last.total_length >= limit:
# Don't merge lines if the result goes over the column limit.
return False
return style.Get('JOIN_MULTIPLE_LINES')
|
