From d4dadf6ca35c0aeff17409bb9c1563fb15c03b56 Mon Sep 17 00:00:00 2001 From: Stefano Date: Tue, 2 Jan 2018 22:35:27 +0100 Subject: [PATCH] Fixed the number of domains in each line. Fixed the number of domains in each line and added the support to inline comments (they will be ignored as the comment lines). Code refactoring. --- updateHostsFile.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/updateHostsFile.py b/updateHostsFile.py index 23b951743..76a76acd1 100644 --- a/updateHostsFile.py +++ b/updateHostsFile.py @@ -671,16 +671,13 @@ def compress_file(input_file, target_ip, output_file): lines_index = 0 for line in input_file.readlines(): line = line.decode("UTF-8") - if line.startswith(('#', '\n')): - continue if line.startswith(target_ip): - current_len = len(lines[lines_index]) - if current_len < 128 and (current_len + len(line[7:])) < 192: - lines[lines_index] += line[7:-1] + if lines[lines_index].count(' ') < 9: + lines[lines_index] += ' ' + line[7:line.find('#')].strip() else: lines[lines_index] += '\n' - lines.append(line[:-1]) + lines.append(line[:line.find('#')].strip()) lines_index += 1 for line in lines: