#!/usr/bin/python
import os.path
import re
import os
import sys
import copy
helptext = """
Tool for encoding tags of mp3 files in the russian 1-byte charsets to unicode
Usage: tag2utf [DIRECTORIES]
(By default files will be searched in the current dirrectory)
Modes:
--restore : programm will try to restore tags, that was broken by not right user choise
--help, --version, --usage - view this text
Version 0.12
Author: Kopats Andrei
hlamer@tut.by
Bugfix: Yarmak Vladislav
chayt@smtp.ru
This program is distributed under the terms of the GPL License.
TODO:
undo changes,
Charsets will be in the config file or command line, for encoding not only from cp1251 and koi8-r
GUI
If you need to encode tags from different charset using this version, you can modify script, it's very easy to do.
"""
charsets = {'cp1251':'c','koi8-r':'k' }
#modify it if you want decode tags from other encodings
"""
Default mode - just to convert tags, that have 1-byte encoding now
Backup mode used for restore wrongly converted tags (and optionaly decode to right charset. """
restoreMode = False
try:
import eyeD3
except:
print 'You need to install python-eyed3 package.'
sys.exit()
mp3FileName = re.compile ('.*(mp3|MP3)$')
def recodingNeed (strs):
"""recoding needed if tags have symbols with 256>ords >127
"""
# needed = False
for string in strs:
for i in range (len(string)):
if 256>ord (string)>127:
return True #nonunicode nonascii
return False
def passDir (rootdir):
tags = []
songs = []
titles = []
artists = []
albums = []
for song in os.listdir(rootdir) :
if ( os.path.isfile(os.path.join(rootdir,song))
and mp3FileName.match (song)):
filename = os.path.join(rootdir,song)
tag = eyeD3.Tag()
try:
if not tag.link(filename):
continue #somthing wrong whith this file
except:
print '\n',filename,':error, may be tag is corrupted.\n '
continue
if recodingNeed ([tag.getTitle(),tag.getArtist(),tag.getAlbum()]) or restoreMode:
if not os.access(filename,os.W_OK):
print 'Warning! Have not access for writing file '\
,filename , ' Sciped!'
continue
tags.append (tag)
songs.append (song)
if not restoreMode : #normal mode
titles.append (getTagStr (tag.getTitle()))
artists.append (getTagStr (tag.getArtist()))
albums.append (getTagStr (tag.getAlbum()))
else: #restore mode
titles.append (tag.getTitle())
artists.append (tag.getArtist())
albums.append (tag.getAlbum())
if len (tags) > 0:
print len(tags),' file(s) finded in the ',rootdir
askUser (tags,songs,titles,artists,albums)
def getTagStr (tagUnicStr):
#gets the 1byte 8bits string, as writed in the tag, from the unicode, returned by tag.get*
ls = []
for i in range (0,len(tagUnicStr)):
if (ord (tagUnicStr) in range(256)):
ls.append (chr (ord (tagUnicStr)))
else:
ls.append(tagUnicStr)
Str8 = ''.join(ls)
return Str8
def updateTags (tags,titles,artists,albums,charset, wrongCharset = ''):
for i in range (len(tags)):
tags.setVersion (eyeD3.ID3_V2_4)
tags.setTextEncoding (eyeD3.UTF_8_ENCODING)
if not ( restoreMode and wrongCharset != ''): #normal mode
if (recodingNeed(artists)):
tags.setArtist(artists.decode(charset))
else:
tags.setArtist(artists)
if (recodingNeed(albums)):
tags.setAlbum(albums.decode(charset))
else:
tags.setAlbum(albums)
if (recodingNeed(titles)):
tags.setTitle(titles.decode(charset))
else:
tags.setTitle(titles)
else: #restore mode
tags.setArtist(artists.encode(wrongCharset).decode(charset))
tags.setAlbum (albums.encode(wrongCharset).decode(charset))
tags.setTitle (titles.encode(wrongCharset).decode(charset))
tags.update()
def askUser (tags,songs, titles,artists,albums):
charsetListStr = ''
if not restoreMode: #normal mode
for charset in charsets.keys():
print '\n'+'['+charsets[charset]+']'+' If charset of tags is '+ charset+ ':'
for i in range (len(songs)):
sys.stdout.write(songs)
outlst = [' ']
if (recodingNeed(titles)):
outlst.append(titles.decode(charset))
else:
outlst.append(titles)
outlst.append(' ')
if (recodingNeed(artists)):
outlst.append(artists.decode(charset))
else:
outlst.append(artists)
outlst.append(' ')
if (recodingNeed(albums)):
outlst.append(albums.decode(charset))
else:
outlst.append(albums)
print "".join(outlst)
else: #backup mode
for wrongCharset in charsets.keys(): #charset, to which file was wrongly converted
for tagCharset in charsets.keys(): #right charset of tag
if wrongCharset == tagCharset:
continue
print '\n'+'['+charsets[wrongCharset]+charsets[tagCharset]+']', \
' If charset is '+ tagCharset+' (wrongly converted to '+ wrongCharset+ ')',':'
for i in range (len(songs)):
try:
print songs,' ',\
titles.encode(wrongCharset).decode(tagCharset),' ',\
artists.encode(wrongCharset).decode(tagCharset),' ',\
albums.encode(wrongCharset).decode(tagCharset)
except:
print "ERROR:Can't encode tags of "+songs
charsetListStr += " '"+charsets[wrongCharset]+charsets[tagCharset]+"' - "+tagCharset+' converted to '+wrongCharset+'\n'
print '\n',"Select charset:\n",charsetListStr, "'s' - skip this file(s)"
if len(tags) >1:
print "'m' - manual for every file"
while 1:
#get user choise end update the tags
choise = raw_input()
if not restoreMode : #normal mode
if choise in charsets.values():
charset = charsets.keys()[charsets.values().index(choise)]
updateTags(tags,titles,artists,albums,charset)
break
else: #restore mode
if (len (choise) == 2) and \
(choise[0] in charsets.values()) and \
(choise[1] in charsets.values()):
charset = charsets.keys()[charsets.values().index(choise[1])]
wrongCharset = charsets.keys()[charsets.values().index(choise[0])]
updateTags(tags,titles,artists,albums,charset,wrongCharset)
break
if choise == 's':
return
if choise == 'm' and len(tags) >1:
for i in range(len(tags)):
askUser ([tags],[songs],[titles],[artists],[albums])
return
#will be executed if no break or return before
print 'What?'
argsFailed = False ;
rootdirs = []
argv = copy.copy ( sys.argv)
argv.__delitem__(0)
for arg in argv:
if ( arg == '--usage' or
arg == '--help' or
arg == '--version'):
print helptext
sys.exit()
elif arg == '--restore':
restoreMode = True
elif os.path.isdir(arg):
rootdirs.append(arg)
#this need because paths may have start in the working dir or in the root dir
elif os.path.isdir (os.path.join (os.getcwd(),arg)):
rootdirs.append (os.path.join (os.getcwd(),arg))
else:
print "Not right argument '",sys.argv,"' It's not a directory.\n Try ",sys.argv[0], " --usage"
argsFailed = True;
if argsFailed:
sys.exit()
if rootdirs == []:
rootdirs = [os.getcwd()]
print 'Starting search in the ',os.getcwd()
for rootdir in rootdirs:
for root, dirs, files in os.walk(rootdir):
passDir (root)
Тема: UTF-8 to CP1251 (Прочитано 1478 раз)
