Use textwrap.dedent

This commit is contained in:
Peter Melnichenko 2016-07-12 12:27:02 +03:00
parent f45c1352f2
commit d45a05939f

View File

@ -17,6 +17,7 @@ import subprocess
import sys import sys
import tarfile import tarfile
import tempfile import tempfile
import textwrap
import zipfile import zipfile
try: try:
@ -682,9 +683,8 @@ class FilePatch(object):
class Patch(object): class Patch(object):
def __init__(self, src): def __init__(self, src):
# The first and the last lines are empty. # The first and the last lines are empty.
lines = src.splitlines()[1:-1] lines = textwrap.dedent(src[1:-1]).splitlines()
indent_length = len(lines[0]) - len(lines[0].lstrip()) lines = [line if line else " " for line in lines]
lines = [line[indent_length:] or " " for line in lines]
self.file_patches = [] self.file_patches = []
file_lines = None file_lines = None
file_name = None file_name = None