mirror of
https://github.com/arsenetar/dupeguru.git
synced 2024-11-16 20:29:02 +00:00
113 lines
3.9 KiB
Python
113 lines
3.9 KiB
Python
# Created By: Virgil Dupras
|
|
# Created On: 2006/11/16
|
|
# Copyright 2011 Hardcoded Software (http://www.hardcoded.net)
|
|
#
|
|
# This software is licensed under the "BSD" License as described in the "LICENSE" file,
|
|
# which should be included with this package. The terms are also available at
|
|
# http://www.hardcoded.net/licenses/bsd_license
|
|
|
|
import logging
|
|
from appscript import app, k, CommandError
|
|
import time
|
|
import os.path as op
|
|
|
|
from cocoa import as_fetch
|
|
from hscommon.trans import tr
|
|
|
|
from core.app import JobType
|
|
from core.scanner import ScanType
|
|
from core_me.app import DupeGuru as DupeGuruBase
|
|
from .app import JOBID2TITLE, PyDupeGuruBase
|
|
|
|
JobType.RemoveDeadTracks = 'jobRemoveDeadTracks'
|
|
JobType.ScanDeadTracks = 'jobScanDeadTracks'
|
|
|
|
JOBID2TITLE.update({
|
|
JobType.RemoveDeadTracks: tr("Removing dead tracks from your iTunes Library"),
|
|
JobType.ScanDeadTracks: tr("Scanning the iTunes Library"),
|
|
})
|
|
|
|
class DupeGuruME(DupeGuruBase):
|
|
def __init__(self, view, appdata):
|
|
appdata = op.join(appdata, 'dupeGuru Music Edition')
|
|
DupeGuruBase.__init__(self, view, appdata)
|
|
self.dead_tracks = []
|
|
|
|
def remove_dead_tracks(self):
|
|
def do(j):
|
|
a = app('iTunes')
|
|
a.activate(timeout=0)
|
|
for index, track in enumerate(j.iter_with_progress(self.dead_tracks)):
|
|
if index % 100 == 0:
|
|
time.sleep(.1)
|
|
try:
|
|
track.delete(timeout=0)
|
|
except CommandError as e:
|
|
logging.warning('Error while trying to remove a track from iTunes: %s' % str(e))
|
|
|
|
self.view.start_job(JobType.RemoveDeadTracks, do)
|
|
|
|
def scan_dead_tracks(self):
|
|
def do(j):
|
|
a = app('iTunes')
|
|
a.activate(timeout=0)
|
|
try:
|
|
[source] = [s for s in a.sources(timeout=0) if s.kind(timeout=0) == k.library]
|
|
[library] = source.library_playlists(timeout=0)
|
|
except ValueError:
|
|
logging.warning('Some unexpected iTunes configuration encountered')
|
|
return
|
|
self.dead_tracks = []
|
|
tracks = as_fetch(library.file_tracks, k.file_track)
|
|
for index, track in enumerate(j.iter_with_progress(tracks)):
|
|
if index % 100 == 0:
|
|
time.sleep(.1)
|
|
if track.location(timeout=0) == k.missing_value:
|
|
self.dead_tracks.append(track)
|
|
logging.info('Found %d dead tracks' % len(self.dead_tracks))
|
|
|
|
self.view.start_job(JobType.ScanDeadTracks, do)
|
|
|
|
class PyDupeGuru(PyDupeGuruBase):
|
|
def __init__(self):
|
|
self._init(DupeGuruME)
|
|
|
|
def removeDeadTracks(self):
|
|
self.model.remove_dead_tracks()
|
|
|
|
def scanDeadTracks(self):
|
|
self.model.scan_dead_tracks()
|
|
|
|
#---Information
|
|
def deadTrackCount(self) -> int:
|
|
return len(self.model.dead_tracks)
|
|
|
|
#---Properties
|
|
def setMinMatchPercentage_(self, percentage: int):
|
|
self.model.scanner.min_match_percentage = percentage
|
|
|
|
def setScanType_(self, scan_type: int):
|
|
try:
|
|
self.model.scanner.scan_type = [
|
|
ScanType.Filename,
|
|
ScanType.Fields,
|
|
ScanType.FieldsNoOrder,
|
|
ScanType.Tag,
|
|
ScanType.Contents,
|
|
ScanType.ContentsAudio,
|
|
][scan_type]
|
|
except IndexError:
|
|
pass
|
|
|
|
def setWordWeighting_(self, words_are_weighted: bool):
|
|
self.model.scanner.word_weighting = words_are_weighted
|
|
|
|
def setMatchSimilarWords_(self, match_similar_words: bool):
|
|
self.model.scanner.match_similar_words = match_similar_words
|
|
|
|
def enable_scanForTag_(self, enable: bool, scan_tag: str):
|
|
if enable:
|
|
self.model.scanner.scanned_tags.add(scan_tag)
|
|
else:
|
|
self.model.scanner.scanned_tags.discard(scan_tag)
|