From 85b6721d3e6d3a2bd1ce7e955d97fa64d22bf233 Mon Sep 17 00:00:00 2001
From: achbed <github@achbed.org>
Date: Mon, 1 Dec 2014 02:01:49 -0600
Subject: [PATCH] Added code to search for "COUNTRY" custom tag in MP3 files

Signed-off-by: achbed <github@achbed.org>
---
 deefuzzer/tools/mp3.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/deefuzzer/tools/mp3.py b/deefuzzer/tools/mp3.py
index 72e8a30..500c37f 100644
--- a/deefuzzer/tools/mp3.py
+++ b/deefuzzer/tools/mp3.py
@@ -46,6 +46,8 @@ from utils import *
 
 EasyID3.valid_keys["comment"]="COMM::'XXX'"
 EasyID3.valid_keys["copyright"]="TCOP::'XXX'"
+EasyID3.valid_keys["country"]="TXXX:COUNTRY:'XXX'"
+EasyID3.RegisterTXXXKey("country","COUNTRY")
 
 class Mp3:
     """A MP3 file object"""
@@ -62,6 +64,7 @@ class Mp3:
                     'album': 'TALB',
                     'date': 'TDRC',
                     'comment': 'COMM',
+                    'country': 'COUNTRY',
                     'genre': 'TCON',
                     'copyright': 'TCOP',
                     }
@@ -77,6 +80,7 @@ class Mp3:
                     'album': '',
                     'date': '',
                     'comment': '',
+                    'country': '',
                     'genre': '',
                     'copyright': '',
                     }
@@ -109,7 +113,10 @@ class Mp3:
             try:
                 metadata[key] = self.mp3[key][0]
             except:
-                metadata[key] = ''
+                try:
+                    metadata[key] = self.mp3[self.keys2id3[key]][0]
+                except:
+                    metadata[key] = ''
         return metadata
 
     def write_tags(self):
-- 
2.39.5