64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
# Copyright 2021 (c) Anna Schumaker.
|
|
import db
|
|
from gi.repository import GObject
|
|
from . import metadata
|
|
|
|
class Task(GObject.GObject):
|
|
def run_task(self): raise NotImplementedError
|
|
|
|
|
|
class CommitTask(Task):
|
|
def run_task(self): db.commit()
|
|
|
|
|
|
class FileTask(Task):
|
|
def __init__(self, library, filepath):
|
|
Task.__init__(self)
|
|
self.library = library
|
|
self.filepath = filepath
|
|
|
|
def run_task(self):
|
|
if db.track.Table.lookup(self.filepath):
|
|
return
|
|
|
|
with metadata.Metadata(self.filepath) as meta:
|
|
artist = db.artist.Table.find(meta.artist(), meta.artistsort())
|
|
album = db.album.Table.find(artist, meta.album())
|
|
disc = db.disc.Table.find(album, meta.discnumber(), meta.discsubtitle())
|
|
year = db.year.Table.find(meta.year())
|
|
track = db.track.Table.insert(self.library, artist, album, disc, year,
|
|
meta.tracknumber(), meta.length(),
|
|
meta.title(), self.filepath)
|
|
for genre in meta.genres():
|
|
db.genre.Map.insert(db.genre.Table.find(genre), track)
|
|
db.playlist.TempMap.insert(db.playlist.Table.find("New Tracks"), track)
|
|
|
|
|
|
class ImportTask(FileTask):
|
|
def __init__(self, library, filepath, playcount, lastplayed):
|
|
FileTask.__init__(self, library, filepath)
|
|
self.playcount = playcount
|
|
self.lastplayed = lastplayed
|
|
|
|
def run_task(self):
|
|
FileTask.run_task(self)
|
|
if track := db.track.Table.lookup(self.filepath):
|
|
track.last_played(self.playcount, self.lastplayed)
|
|
|
|
|
|
class DirectoryTask(Task):
|
|
def __init__(self, library, dirpath):
|
|
Task.__init__(self)
|
|
self.library = library
|
|
self.dirpath = dirpath
|
|
|
|
def run_task(self):
|
|
res = [ ]
|
|
for f in self.dirpath.iterdir():
|
|
if f.is_file():
|
|
res.append(FileTask(self.library, f))
|
|
elif f.is_dir():
|
|
res.append(DirectoryTask(self.library, f))
|
|
res.append(CommitTask())
|
|
return res
|