Safe file handling, closes #18

This commit is contained in:
Nimesh Ghelani 2016-01-23 23:02:19 +05:30
parent fd80fdb221
commit e02ef0fff6

View file

@ -420,9 +420,8 @@ class Playlist(Record):
return fullPath
def populate(self, filename):
f = open(filename, "rb")
with open(filename, 'rb') as f:
data = f.readlines()
f.close()
extension = os.path.splitext(filename)[1].lower()
if extension == '.pls':
@ -505,9 +504,8 @@ class Shuffler(object):
self.lists.append(os.path.abspath(os.path.join(dirpath, filename)))
def write_database(self):
f = open(os.path.join(self.base, "iPod_Control", "iTunes", "iTunesSD"), "wb")
with open(os.path.join(self.base, "iPod_Control", "iTunes", "iTunesSD"), "wb") as f:
f.write(self.tunessd.construct())
f.close()
#
# Read all files from the directory