blob: a7233d4ca6e9150e60abd1f38c5eeb8ea1482f26 [file] [log] [blame]
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))