#!/usr/bin/env python

# -*- coding: utf-8 -*-

# Copyright (C) 2013 Ben Ockmore

# This file is part of Warp Tagger.

# Warp Tagger is free software: you can redistribute it

# and/or modify it under the terms of the GNU General Public License as

# published by the Free Software Foundation, either version 3 of the

# License, or (at your option) any later version.

# Warp Tagger is distributed in the hope that it will be

# useful, but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License

# along with Warp Tagger. If not, see <http://www.gnu.org/licenses/>.

import os

import time

from collections import deque

import mutagen.oggvorbis

import mutagen.flac

import musicbrainzngs as ws

import Warp.compatid3

import Warp.track

import Warp.release

import Warp.plugin

import Warp.utils

num_total_songs = 0

albums_fetch_queue = deque()

albums = {}

options = {}

skipped_files = list()

num_passes = 0

num_completed_releases = 0

last_num_completed_releases = None

no_new_albums = False

ValidFileTypes = [

]

def PrintHeader():

return

def FetchNextRelease():

return None

def RequestRelease( release_id ):

return albums[release_id]

def InterpretOptionValue( value ):

return unicode( value.replace( "\"", "" ).replace( "\n", "" ) )

def ReadOptions( file_name ):

Warp.utils.safeprint( u"\n" )

##################################

### Metadata Syncing Functions ###

##################################

def SyncMetadata( release ):

return

########################

### Main Script Loop ###

########################

PrintHeader()

Warp.plugin.LoadPlugins()

ws.set_rate_limit() # Disable the default rate limiting, as I do my own, and don't know whether this is blocking/non-blocking.

ws.set_useragent( "Warp Tagger", "0.2", "ben.sput@gmail.com" )

if os.path.exists( "./options" ):

ReadOptions( "./options" )

elif os.path.exists( "./options.default" ):

ReadOptions( "./options.default" )

# result = ws.get_release_by_id("dee5c657-7960-4a3c-a01e-66865ba24320",["artist-credits","recordings","labels","release-groups","media"])["release"]

# Warp.utils.safeprint( json.dumps(result, sort_keys=True, indent=4, ensure_ascii = False) )

last_fetch_time = 0

# raise SystemError

if os.path.isdir( options["library_folder"] ):

options["library_folder"] = os.path.realpath( options["library_folder"] )

else:

options["library_folder"] = os.path.realpath( "./" )

ignores = list()

for dirname, dirnames, filenames in os.walk( options["library_folder"] ):

if "warp-ignore" in filenames:

del filenames[:]

del dirnames[:]

for filename in filenames:

if os.path.splitext( filename )[1][1:] in ValidFileTypes: # Compares extension to valid extensions.

num_total_songs += 1

Warp.utils.safeprint( u"Found {} songs to update.".format( num_total_songs ) )

Warp.utils.safeprint( u"Updating...\n" )

while num_completed_releases != last_num_completed_releases:

last_num_completed_releases = len( albums )

num_passes += 1

Warp.utils.safeprint( u"Starting Pass #{}".format( num_passes ) )

for dirname, dirnames, filenames in os.walk( options["library_folder"] ):

if "warp-ignore" in filenames:

del filenames[:]

del dirnames[:]

for filename in filenames:

if ( Warp.track.Track.num_loaded > 1000 ) or ( Warp.release.Release.num_loaded > 100 ):

no_new_albums = True

release_id = None

audio = None

track = None

file_ext = os.path.splitext( filename )[1][1:]

is_audio_file = False

abs_file_path = os.path.realpath( os.path.join( dirname, filename ) )

if file_ext == "mp3":

is_audio_file = True

try:

audio = Warp.compatid3.CompatID3( abs_file_path )

except mutagen.id3.ID3NoHeaderError:

pass

else:

if "TXXX:MusicBrainz Album Id" in audio:

release_id = str( audio["TXXX:MusicBrainz Album Id"] )

elif "TXXX:musicbrainz_albumid" in audio:

release_id = str( audio["TXXX:musicbrainz_albumid"] )

elif "TXXX:MUSICBRAINZ_ALBUMID" in audio:

release_id = str( audio["TXXX:MUSICBRAINZ_ALBUMID"] )

else:

Warp.utils.safeprint( u"Song: {} doesn't have tag:".format( audio.filename ) )

for key, value in audio.items():

if str( key )[0:4] != "APIC":

Warp.utils.safeprint( u"{} : {}".format( key, value ) )

if release_id is not None:

track = Warp.track.MP3Track( audio, file_ext, options )

elif file_ext == "flac":

is_audio_file = True

try:

audio = mutagen.flac.FLAC( abs_file_path )

except mutagen.flac.FLACNoHeaderError:

pass

else:

if "musicbrainz_albumid" in audio:

release_id = str( audio["musicbrainz_albumid"][0] )

track = Warp.track.FLACTrack( audio, file_ext, options )

elif file_ext == "ogg":

is_audio_file = True

try:

audio = mutagen.oggvorbis.OggVorbis( abs_file_path )

except mutagen.oggvorbis.OggVorbisHeaderError:

pass

else:

is_audio_file = True

if "musicbrainz_albumid" in audio:

is_audio_file = True

release_id = str( audio["musicbrainz_albumid"][0] )

track = Warp.track.OggTrack( audio, file_ext, options )

if release_id != None:

if ( no_new_albums == False ) or ( ( no_new_albums == True ) and ( release_id in albums.keys() ) ):

release = RequestRelease( release_id )

if release is not None:

track.release = release

release.AddSong( track )

elif is_audio_file:

if abs_file_path not in skipped_files:

skipped_files.append( abs_file_path )

Warp.track.Track.num_processed += 1

while len( albums_fetch_queue ) > 0:

fetched_release = FetchNextRelease()

if fetched_release != None:

SyncMetadata( fetched_release )

fetched_release.Close()

Warp.utils.safeprint( u"\nPass #{}: {} songs remaining to process and {}/{} processed.\n".format( num_passes, Warp.track.Track.num_loaded, Warp.track.Track.num_processed, num_total_songs ) )

num_completed_releases = len( albums )

no_new_albums = False

Warp.utils.safeprint( u"\n-------------------------------------------------" )

Warp.utils.safeprint( u"Checked {} MP3s, {} OGGs and {} FLACs.".format( Warp.track.MP3Track.count, Warp.track.OggTrack.count, Warp.track.FLACTrack.count ) )

if len( skipped_files ) > 0:

Warp.utils.safeprint( u"Skipped {} files with no MB Release ID:".format( len( skipped_files ) ) )

for s in skipped_files:

Warp.utils.safeprint( u" - {}".format( s ) )