読者です 読者をやめる 読者になる 読者になる

久々にPythonでプログラム書いたのでメモ.

プログラム

仕事でテストデータ作るのが面倒だったのでw ツールを作成してみました.
何がしたいかというと,.mp3ファイルがたくさん欲しかったので,IDv3タグを変更してIdv3タグ的に別のアーティスト・アルバムにしたかったのです.

IDv3タグをいじれるライブラリを探したらmutagenというのを発見したので,
早速ダウンロードして使ってみました.

作ったソースはこんな感じです. Python慣れてないから,変なことしてるかも知れないけど・・・
#エラー処理とかはちゃんとやってません.自分で使うのがメインだし.

#!/usr/bin/python

import os, sys, time
import threading

sys.path.append('mutagen')
from mutagen.mp3 import MP3
from mutagen.easyid3 import EasyID3
import mutagen.id3

class ExcuteWriting(threading.Thread):
	def __init__(self, tagData):
		threading.Thread.__init__(self)
		self.tagData = tagData
		
	def run(self):
		
		target_dir =  self.tagData.result_dir + "/" + "artist_" + str(self.tagData.artist)
		create_dir(target_dir)

		for album in range(int(self.tagData.album_max)):
			for track in range(int(self.tagData.track_max)):
				filename = target_dir + "/" + "album_" + str(album) + "_" + str(track + 1) + ".mp3"
				print "Creating %s...\n" % filename
				self.copy_file(self.tagData.base_file, filename)
				self.write_mp3_tag(filename, album)
	
	def write_mp3_tag(self, filename, album):
		m = MP3(filename, ID3=EasyID3)

		m.add_tags(ID3=EasyID3)

		m['title'] = "Title" + str(self.tagData.artist)
		m['artist'] = str(self.tagData.artist)
		m['album'] = str(album)
		m.save()
		m.pprint()
#		print(m.pprint())


	# Read base file and copy it.
	def copy_file(self, old_name, new_name):
		try:
			fr = open(old_name, "rb")
			fw = open(new_name, "wb")
			
			fw.write(fr.read())
			
			fr.close()
			fw.close()
			
		except :
			print "file read/write error occured\n"


class TagData:
	def __init__(self, base_file, artist, album_max, track_max, result_dir):
		self.base_file = base_file
		self.artist = artist
		self.album_max = album_max
		self.track_max = track_max
		self.result_dir = result_dir



def create_dir(ret_dir):
	# if we do have result dir, we won't create it.
	if (os.access(ret_dir, os.F_OK) == False):
		os.mkdir(ret_dir)
	
if __name__ == "__main__":

	# At first, we have to check command line arguments.
	if (len(sys.argv) != 6):
		print "usage: id3writer [base file] [output directory] [n artist] [n album] [n tracks]\n"
		exit (0)

	base_file = sys.argv[1]
	ret_dir = sys.argv[2]
	artist_max = sys.argv[3]
	album_max = sys.argv[4]
	track_max = sys.argv[5]

	# Do we have result dir?
	create_dir(ret_dir)
	
	for artist_num in range (int(artist_max)):
		# Max thread is 10.
		if (threading.activeCount() > 10):
			while threading.activeCount() > 10:
				time.sleep(0.5) 

		tagData = TagData(base_file, artist_num, album_max, track_max, ret_dir)
		worker = ExcuteWriting(tagData)
		worker.start()

大量にデータ作るのにシーケンシャルにやってたら時間かかりそうなので,スレッドも初めて使ってみました.
このプログラムが行うのは,1アルバム辺りNトラックあるアルバムをM個,Kアーティスト分作成するという内容です.

例えば,1アルバムに2トラック・1アーティストに2アルバムというのを2アーティスト分作るとこんな感じです.

[masami@moonlight:~/experiment/id3writer]% ./id3write.py base.mp3 result 2 2 2          
Creating result/artist_0/album_0_1.mp3...

Creating result/artist_1/album_0_1.mp3...

Creating result/artist_0/album_0_2.mp3...

Creating result/artist_1/album_0_2.mp3...

Creating result/artist_0/album_1_1.mp3...

Creating result/artist_1/album_1_1.mp3...

Creating result/artist_0/album_1_2.mp3...

Creating result/artist_1/album_1_2.mp3...

[masami@moonlight:~/experiment/id3writer]% ls -ls result/*
result/artist_0:
合計 2360
  4 drwxr-xr-x 2 masami masami   4096 Jan  9 00:19 ./
  4 drwxr-xr-x 4 masami masami   4096 Jan  9 00:19 ../
588 -rw-r--r-- 1 masami masami 598595 Jan  9 00:19 album_0_1.mp3
588 -rw-r--r-- 1 masami masami 598595 Jan  9 00:19 album_0_2.mp3
588 -rw-r--r-- 1 masami masami 598595 Jan  9 00:19 album_1_1.mp3
588 -rw-r--r-- 1 masami masami 598595 Jan  9 00:19 album_1_2.mp3

result/artist_1:
合計 2360
  4 drwxr-xr-x 2 masami masami   4096 Jan  9 00:19 ./
  4 drwxr-xr-x 4 masami masami   4096 Jan  9 00:19 ../
588 -rw-r--r-- 1 masami masami 598595 Jan  9 00:19 album_0_1.mp3
588 -rw-r--r-- 1 masami masami 598595 Jan  9 00:19 album_0_2.mp3
588 -rw-r--r-- 1 masami masami 598595 Jan  9 00:19 album_1_1.mp3
588 -rw-r--r-- 1 masami masami 598595 Jan  9 00:19 album_1_2.mp3
[masami@moonlight:~/experiment/id3writer]% 

スレッドを使うときは"threading.Thread"を継承する.継承の仕方はクラス名の後に(threading.Thread)を付けると.

class ExcuteWriting(threading.Thread):

コンストラクタからthreading.Threadの__init__を呼んであげる必要があります.

	def __init__(self, tagData):
		threading.Thread.__init__(self)

そうしたら,スレッド動かすだけなので,
クラスのインスタンスを生成して,startメソッドを呼べばOKです.

worker = ExcuteWriting(tagData)
		worker.start()

start()からExcuteWritingのrunメソッドが呼ばれるので,runメソッドは必須です.
今回は使ってないですけど、class localってTLSみたいなもんですかね?
リファレンスによるとクラスに対して使う感じのようです.

 foo = threading.local()
 foo.x = 10

あと,無制限にスレッド作るのもなんなので,threading.activeCount()というメソッドを使って,
動いているスレッド数は最大10個までという感じにしました.

		if (threading.activeCount() > 10):
			while threading.activeCount() > 10:
				time.sleep(0.5) 

スレッド周りは大体こんな感じです.

実際にIDv3タグを弄るところはこんな感じになってます.
MP3クラスのインスタンスを作って・・・

		m = MP3(filename, ID3=EasyID3)

		m.add_tags(ID3=EasyID3)

タグに対して値をセットします.
trackを試したら"無い"と怒られました(つд⊂)エーン

		m['title'] = "Title" + str(self.tagData.artist)
		m['artist'] = str(self.tagData.artist)
		m['album'] = str(album)

最後にセーブすれば完了です.

		m.save()

スレッド使ったり,IDv3タグ弄るライブラリ使っても,たったこれだけのコードで必要な処理ができるので
Python便利です!