Skip to content

Commit 7ffb8c4

Browse files
committed
Reduce branches in initialize_runs()
1 parent 14762c1 commit 7ffb8c4

File tree

1 file changed

+14
-16
lines changed

1 file changed

+14
-16
lines changed

src/main/main.py

Lines changed: 14 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -146,38 +146,36 @@ def initialize_runs(handler: PackageHandler, today: datetime) -> int:
146146
'end': '<!-- Index End - WARNING: Do not delete or modify this markdown comment. -->'
147147
}
148148

149-
start_line_readme = None
150-
end_line_readme = None
151-
152149
index_header = {
153150
'labels': f'| Day | Title | Solution | Site | Difficulty | {env_vars["nb_name"]} |',
154151
'sep': f'| ----- | ------- | ---------- | ------ | ------------ | { "-" * (len(env_vars["nb_name"]) + 2) } |'
155152
}
156153

154+
start_line_readme = None
155+
end_line_readme = None
156+
lines_readme = []
157+
lines_template = []
158+
157159
with open('README.md', 'r+', encoding='utf-8') as file:
158-
lines = file.readlines()
160+
lines_readme = file.readlines()
159161

160-
for i in range(len(lines)-1, -1, -1):
162+
for i in range(len(lines_readme)-1, -1, -1):
161163

162-
if index_block['start'] in lines[i]:
164+
if index_block['start'] in lines_readme[i]:
163165
start_line_readme = i
164166

165-
if index_block['end'] in lines[i]:
167+
if index_block['end'] in lines_readme[i]:
166168
end_line_readme = i
167169

168-
for j in range(start_line_readme + 1, end_line_readme):
169-
170-
if j == start_line_readme + 1:
171-
lines[j] = f'{index_header["labels"]}\n'
170+
lines_readme[start_line_readme + 1] = f'{index_header["labels"]}\n'
171+
lines_readme[start_line_readme + 2] = f'{index_header["sep"]}\n'
172172

173-
if j == start_line_readme + 2:
174-
lines[j] = f'{index_header["sep"]}\n'
173+
for j in range(start_line_readme + 3, end_line_readme):
175174

176-
if j > start_line_readme + 2:
177-
lines[j] = ''
175+
lines_readme[j] = ''
178176

179177
file.seek(0)
180-
file.writelines(lines)
178+
file.writelines(lines_readme)
181179
file.truncate()
182180

183181
# HANDLE SETUP - EXTRA COLUMN (aka NB) - TEMPLATE

0 commit comments

Comments
 (0)