upstream/IPod-Shuffle-4g
1
0
Fork 1
mirror of https://github.com/nims11/IPod-Shuffle-4g.git synced 2025-12-08 00:18:01 +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
parent fd80fdb221
commit e02ef0fff6
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

@ -420,9 +420,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':
@ -505,9 +504,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