From 62f12d031a19c0773f789e2adb17d54abfcf055a Mon Sep 17 00:00:00 2001
From: Connor Olding <cloningdonor@gmail.com>
Date: Sun, 2 Nov 2014 13:29:44 -0800
Subject: [PATCH] more cleanup

---
 mutaext.py | 18 ++++++++---------
 unsync.py  | 59 ++++++++++++++++++++----------------------------------
 2 files changed, 31 insertions(+), 46 deletions(-)

diff --git a/mutaext.py b/mutaext.py
index 504cabb..ce3a785 100644
--- a/mutaext.py
+++ b/mutaext.py
@@ -25,23 +25,23 @@ def rating2byte(r):
     return 0
 
 def rating_get(id3, key):
-    try:
+    if 'TXXX:RATING' in id3:
         rating = id3['TXXX:RATING']
-    except KeyError:
+        return list(rating.text)
+    else:
         try:
             _, popm = next(popms(id3))
         except StopIteration:
             return []
         else:
             return [str(byte2rating(popm.rating))]
-    else:
-        return list(rating.text)
 
 def _canconv(r):
     try:
-        if int(r) != str(int(r)):
+        ir = int(r)
+        if ir != str(ir):
             return False
-        return int(r) >= 1 and int(r) <= 5
+        return ir >= 1 and ir <= 5
     except (ValueError, TypeError):
         return False
 
@@ -77,9 +77,6 @@ EasyID3.RegisterTextKey('albumartist', 'TPE2')
 EasyID3.RegisterKey('rating', rating_get, rating_set, rating_delete)
 
 class SyncFile(MutableMapping):
-    """Dumb OOP crap that just adds more code
-    and a sense of self-importance"""
-
     def __init__(self, path):
         self.md = mutagenx.File(path, easy=True)
         self.path = path
@@ -113,3 +110,6 @@ class SyncFile(MutableMapping):
 
     def __len__(self):
         return len([k for k in self.__iter__()])
+
+    def save(self):
+        return self.md.save()
diff --git a/unsync.py b/unsync.py
index 1adfa36..5c49a80 100755
--- a/unsync.py
+++ b/unsync.py
@@ -1,25 +1,22 @@
 #!/bin/python
 
-import os
-import os.path
+import os, os.path
 import sys
-import shutil
-import tempfile
+from shutil import copy2
+from tempfile import mkstemp
 from zlib import crc32
 
 import mutaext
 import convert
 from mutaext import SyncFile
 
-# BUG: doesn't work with my .m4a files?
-#goodexts = ('.mp3', '.mp2', '.m4a', '.flac', '.ogg')
 goodexts = ('.mp3', '.flac', '.ogg')
 
 matchtags = ['artist', 'album', 'title']
-alltags = [\
-    'albumartist', 'composer', 'comment' \
-    'tracknumber', 'discnumber', \
-    'genre', 'date', \
+alltags = [
+    'albumartist', 'composer', 'comment',
+    'tracknumber', 'discnumber',
+    'genre', 'date',
 ]
 alltags.extend(mutaext.replaygain_tags)
 alltags.extend(mutaext.extra_tags)
@@ -29,15 +26,12 @@ lament = lambda *args, **kwargs: print(*args, file=sys.stderr, **kwargs)
 walkfiles = lambda w: (os.path.join(r, f) for r, _, fs in w for f in fs)
 extof = lambda p: os.path.splitext(p)[1].lower()
 filterext = lambda ps, es: (p for p in ps if extof(p) in es)
-ansty = lambda u: str(u.decode('ascii', errors='replace').replace('\ufffd', '?'))
 
 def shouldsync(md):
     rating = md.get('rating', '')
     sync = md.get('sync', '')
-    try:
+    if rating.isnumeric():
         rating = int(rating)
-    except ValueError:
-        pass
     if sync:
         sync = sync.lower()
 
@@ -80,8 +74,8 @@ def makefilename(md):
     album = md['album']
 
     fn = "%(artist)s - %(album)s - %(title)s" % locals()
-    crc = crc32(fn.encode('utf-8')) & 0xFFFFFFFF
     # FAT is a pain to deal with so just use nondescript filenames
+    crc = crc32(fn.encode('utf-8')) & 0xFFFFFFFF
     fn = '{:08X}.ogg'.format(crc)
 
     return fn
@@ -108,7 +102,7 @@ def run(args):
     if inonly:
         return 0
 
-    lament("Matching...")
+    lament("Matching tags...")
 
     outdir = args[2]
     for p in paths(outdir):
@@ -119,42 +113,33 @@ def run(args):
             os.remove(p)
         elif updatemetadata(md, match):
             print("UPD", p)
-            md.md.save()
+            md.save()
+
+    lament("Syncing files...")
 
-    lament("[beginning tosync]")
     for md in tosync:
-        #print("SYN", md.path)
         fn = makefilename(md)
         fout = os.path.join(outdir, fn)
 
         if md.seen:
-            try:
-                _from = md.seen
-                _to = fout
-                if type(_from) != type(_to):
-                    raise TypeError
-
-                if _from != _to:
-                    print("MOV", _from)
-                    os.rename(_from, _to)
-                continue
-            except:
-                lament(type(_from), type(_to))
-                lament("_from:", [_from])
-                lament("_to:", [_to])
-                raise
+            _from = md.seen
+            _to = fout
+            if _from != _to:
+                print("MOV", _from)
+                os.rename(_from, _to)
+            continue
 
         print("ADD", md.path)
 
-        _, ftemp = tempfile.mkstemp()
+        _, ftemp = mkstemp()
         try:
             convert.ogg(md.path, ftemp)
             mdnew = SyncFile(ftemp)
             for tag in alltags:
                 if tag in md:
                     mdnew[tag] = md[tag]
-            mdnew.md.save()
-            shutil.copy2(ftemp, fout)
+            mdnew.save()
+            copy2(ftemp, fout)
         finally:
             os.remove(ftemp)