blob: a7233d4ca6e9150e60abd1f38c5eeb8ea1482f26 (
plain)
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
|
import os
import re
import errno
def write_file(path, data):
# In case data is None, return immediately
if data is None:
return
wdata = data.rstrip() + "\n"
with open(path, "w") as f:
f.write(wdata)
def append_file(path, data):
# In case data is None, return immediately
if data is None:
return
wdata = data.rstrip() + "\n"
with open(path, "a") as f:
f.write(wdata)
def read_file(path):
data = None
with open(path) as f:
data = f.read()
return data
def remove_from_file(path, data):
# In case data is None, return immediately
if data is None:
return
try:
rdata = read_file(path)
except IOError as e:
# if file does not exit, just quit, otherwise raise an exception
if e.errno == errno.ENOENT:
return
else:
raise
contents = rdata.strip().splitlines()
for r in data.strip().splitlines():
try:
contents.remove(r)
except ValueError:
pass
write_file(path, "\n".join(contents))
|
