mirror of
https://github.com/spl0k/supysonic.git
synced 2024-09-20 03:11:04 +00:00
75 lines
1.7 KiB
Python
Executable File
75 lines
1.7 KiB
Python
Executable File
# coding: utf-8
|
|
|
|
import os.path
|
|
import datetime
|
|
import eyeD3
|
|
|
|
import db
|
|
|
|
def seconds_to_time(secs):
|
|
th = secs / 3600
|
|
tm = (secs % 3600) / 60
|
|
ts = secs % 60
|
|
return datetime.time(int(th), int(tm), int(ts))
|
|
|
|
class Scanner:
|
|
def __init__(self, session):
|
|
self.__session = session
|
|
self.__added_artists = 0
|
|
self.__added_albums = 0
|
|
self.__added_tracks = 0
|
|
self.__deleted_artists = 0
|
|
self.__deleted_albums = 0
|
|
self.__deleted_tracks = 0
|
|
|
|
def scan(self, folder):
|
|
for root, subfolders, files in os.walk(folder.path):
|
|
for f in files:
|
|
if f.endswith('.mp3'):
|
|
self.__scan_file(os.path.join(root, f))
|
|
|
|
def prune(self, folder):
|
|
pass
|
|
|
|
def __scan_file(self, path):
|
|
tag = eyeD3.Tag()
|
|
tag.link(path)
|
|
|
|
al = self.__find_album(tag.getArtist(), tag.getAlbum())
|
|
tr = al.tracks.filter(db.Track.path == path).first()
|
|
if tr is None:
|
|
tr = db.Track(path = path)
|
|
self.__added_tracks += 1
|
|
|
|
tr.disc = (tag.getDiscNum() or (1, 1))[0]
|
|
tr.number = tag.getTrackNum()[0]
|
|
tr.title = tag.getTitle()
|
|
tr.duration = seconds_to_time(eyeD3.Mp3AudioFile(path).getPlayTime())
|
|
tr.album = al
|
|
|
|
def __find_album(self, artist, album):
|
|
ar = self.__find_artist(artist)
|
|
al = ar.albums.filter(db.Album.name == album).first()
|
|
if not al is None:
|
|
return al
|
|
|
|
al = db.Album(name = album, artist = ar)
|
|
self.__added_albums += 1
|
|
|
|
return al
|
|
|
|
def __find_artist(self, artist):
|
|
ar = self.__session.query(db.Artist).filter(db.Artist.name == artist).first()
|
|
if not ar is None:
|
|
return ar
|
|
|
|
ar = db.Artist(name = artist)
|
|
self.__session.add(ar)
|
|
self.__added_artists += 1
|
|
|
|
return ar
|
|
|
|
def stats(self):
|
|
return (self.__added_artists, self.__added_albums, self.__added_tracks), (self.__deleted_artists, self.__deleted_albums, self.__deleted_tracks)
|
|
|