-
Notifications
You must be signed in to change notification settings - Fork 0
/
handle_text_file.py
49 lines (41 loc) · 1.38 KB
/
handle_text_file.py
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
#!/usr/bin/python3
'''
handle_text_file.py
Reads lines in a file and store it in a array.
Ignoring empty lines and lines starting with #
The function can be included in other scripts like this:
from handle_text_file import read_lines
lines = read_lines(MY_FILE)
result = save_to_text_file('output.txt', lines)
'''
import os
def read_lines(filename, verbose=0):
# Check if the file exists
if not os.path.exists(filename):
if verbose > 0:
print(f"Error: The file '{filename}' does not exist.")
return []
lines = []
with open(filename, 'r') as file:
for line in file:
stripped_line = line.strip()
if stripped_line and not stripped_line.startswith('#'):
lines.append(stripped_line)
return lines
def save_to_text_file(filename, data, verbose=0):
try:
with open(filename, 'w') as file:
for line in data:
file.write(f"{line}\n")
if verbose > 0:
print(f"Data successfully written to {filename}")
return True
except Exception as e:
if verbose > 0:
print(f"An error occurred while writing to the file: {e}")
return False
### Start
if __name__ == "__main__":
file_content = read_lines('your_file.txt', 1)
print(file_content)
result = save_to_text_file('output.txt', ['Line 1', 'Line 2', 'Line 3'])