upstream/IPod-Shuffle-4g
1
0
Fork 1
mirror of https://github.com/nims11/IPod-Shuffle-4g.git synced 2025-12-07 16:08:00 +09:00
Code Issues Projects Releases Packages Wiki Activity Actions

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
def populate(self, filename):
f = open(filename, "rb")
data = f.readlines()
f.close()
with open(filename, 'rb') as f:
data = f.readlines()
extension = os.path.splitext(filename)[1].lower()
if extension == '.pls':
@ -529,9 +528,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")
f.write(self.tunessd.construct())
f.close()
with open(os.path.join(self.base, "iPod_Control", "iTunes", "iTunesSD"), "wb") as f:
f.write(self.tunessd.construct())
#
# Read all files from the directory