dc1b6dd4a6
I can add songs to a playlist and populate the playlist afterwards (and on startup).
65 lines
1.2 KiB
Python
65 lines
1.2 KiB
Python
# Bryan Schumaker (10/1/2010)
|
|
|
|
import libsaria
|
|
import collection
|
|
|
|
from libsaria import data
|
|
|
|
save = data.save
|
|
load = data.load
|
|
|
|
sep = libsaria.path.sep
|
|
walk = libsaria.path.walk
|
|
join = libsaria.path.join
|
|
splitext = libsaria.path.splitext
|
|
|
|
badfiles = set()
|
|
|
|
def set_badfiles():
|
|
global badfiles
|
|
bf = load("badfiles")
|
|
if bf != None:
|
|
badfiles = bf
|
|
|
|
|
|
class Library(collection.Collection):
|
|
def __init__(self):
|
|
collection.Collection.__init__(self, "library.dl_tree")
|
|
|
|
def scan(self, path):
|
|
print "Library scanning %s" % path
|
|
set_badfiles()
|
|
self.reset()
|
|
self.update(path)
|
|
self.save()
|
|
save(badfiles, "badfiles", "")
|
|
self.disp()
|
|
|
|
def update(self, path):
|
|
global badfiles
|
|
FileRef = libsaria.collection.FileRef
|
|
|
|
for root,dirs,files in walk(path):
|
|
stripped_root = root.strip(sep)
|
|
split_root = stripped_root.split(sep)
|
|
for file in files:
|
|
ext = splitext(file)[1]
|
|
if ext in badfiles:
|
|
continue
|
|
path = join(root, file)
|
|
try:
|
|
ref = FileRef(path)
|
|
except:
|
|
badfiles.add(ext)
|
|
continue
|
|
try:
|
|
self.insert_allocate(split_root + [file], ref)
|
|
except UnicodeEncodeError:
|
|
pass
|
|
|
|
|
|
class Playlist(collection.Collection):
|
|
def __init__(self):
|
|
collection.Collection.__init__(self, "playlist.dl_tree")
|
|
|