ocarina/libsaria/collection/lens.py

59 lines
1.1 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