noneinfo01/IPod-Shuffle-4g
1
0
Fork 0
forked from upstream/IPod-Shuffle-4g
Code Pull requests Activity

Safe file handling, closes #18

Browse source
This commit is contained in:
Nimesh Ghelani 2016-01-23 23:02:19 +05:30 • committed by NicoHood
parent bcde228a62
commit 15b305739a
1 changed files with 4 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

10
shuffle.py
View file

@ -440,9 +440,8 @@ class Playlist(Record):
return fullPath return fullPath
def populate(self, filename): def populate(self, filename):
f = open(filename, "rb") with open(filename, 'rb') as f:
data = f.readlines() data = f.readlines()
f.close()
extension = os.path.splitext(filename)[1].lower() extension = os.path.splitext(filename)[1].lower()
if extension == '.pls': if extension == '.pls':
@ -529,9 +528,8 @@ class Shuffler(object):
self.lists.append(os.path.abspath(os.path.join(dirpath, filename))) self.lists.append(os.path.abspath(os.path.join(dirpath, filename)))
def write_database(self): 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.write(self.tunessd.construct())
f.close()
# #
# Read all files from the directory # Read all files from the directory