python爬蟲教程:編寫Python腳本來獲取mp3文件tag信息的教程

這篇文章主要介紹了編寫Python腳本來獲取mp3文件tag信息的教程,代碼基於Python2.x,文中的註釋很詳細,需要的朋友可以參考下
下面利用一個python的實例程序,來學習python。這個程序的目的就是分析出所有MP3文件的Tag信息並輸出。
import os # 導入os模塊,提供文件路徑,列出文件等方法
import sys # 導入sys模塊,使用sys.modules獲取模塊中的所有內容,類似反射的功能
from UserDict import UserDict # 這個表示從UserDict類中導入UserDict,類似於Java中的 import UserDict.UserDict
def stripnulls(data):

  "一個空字符串的處理函數將所有00字節的內容替換爲空字符,病將前後的空字符串去掉"
  # Python中的strip用於去除字符串的首尾字符,同理,lstrip用於去除左邊的字符,rstrip用於去除右邊的字符。
  return data.replace("\00", "").strip()
  
class FileInfo(UserDict):
  '''文件基類,存儲文件的文件名,繼承自UserDict(存儲key-value的一個類,可以重寫__setitem__,__getitem__方法,
  就可以使用[])'''
  # self是定義時使用,使用時不需要,如果沒有參數,則filename默認None,如果有一個參數的話,參數即爲filename
  def __init__(self, filename=None):
    UserDict.__init__(self) # 初始化父類
    self["name"] = filename # 設置name爲 filaname
  
class MP3FileInfo(FileInfo):
  "MP3文件的信息類,用於分析MP3文件和存儲信息"
  # tagDataMap 用於存儲MP3的Tag信息分別所在位置,( key : 開始位置,結束位置, 處理函數),
  # stripnulls表示最開始定義的函數
  tagDataMap = {"title" : ( 3, 33, stripnulls),
  "artist" : ( 33, 63, stripnulls),
  "album" : ( 63, 93, stripnulls),
  "year" : ( 93, 97, stripnulls),
  "comment" : ( 97, 126, stripnulls),
  "genre" : (127, 128, ord)}
    
  def __parse(self, filename): # 解析MP3文件
    self.clear()
    try:
      fsock = open(filename, "rb", 0) # 打開文件
      try:
        # 設置文件讀取的指針位置, seek第二個參數,2表示從文件結尾作爲參考點,
        # -128表示還有128字節結尾的點,0表示文件開頭做參考點,1表示當前位置做參考點
        fsock.seek(-128, 2)
        tagdata = fsock.read(128) # 讀取128字節的數據
      finally:
        fsock.close() # 關閉文件,注意在finally中,出錯也需要關閉文件句柄
      if tagdata[:3] == "TAG": # 判斷是否是有效的含Tag的MP3文件
        # 循環取出Tag信息位置信息, 如3, 33, stripnulls,並依次賦給start, end, parseFunc
        for tag, (start, end, parseFunc) in self.tagDataMap.items():
          # tagdata[start:end]讀出start到end的字節,使用parseFunc處理這些內容
          self[tag] = parseFunc(tagdata[start:end])
    except IOError: # 如果出現IOError,則跳過繼續
      pass
    
  # 重寫__setitem__方法,上面的self[tag] = parseFunc(tagdata[start:end])就會使用這個方法,
  # key爲tag,itme爲parseFunc(tagdata[start:end])
  def __setitem__(self, key, item):
    if key == "name" and item: # 如果key是 name,並且 item不爲空
      self.__parse(item) # 解析MP3文件
      # problem here,should out of the if
      # FileInfo.__setitem__(self, key, item) 如果使用這個縮進就會出現錯誤
    # 之前的錯誤點,注意這兒的縮進,無論如何都會存儲key-value,使用FileInfo.__setitem__父類的方法來存儲
    FileInfo.__setitem__(self, key, item)
        
def listDirectory(directory, fileExtList):
  "獲取directory目錄下的所有fileExtList格式的文件,fileExtList是一個列表,可以有多種格式"
  fileList = [os.path.normcase(f)
    for f in os.listdir(directory)] # 列出所有 directory的文件
  fileList = [os.path.join(directory, f)
    for f in fileList
    # 過濾文件,滿足fileExtList內的一種格式。os.path.splitext將文件分成文件名和擴展名
    if os.path.splitext(f)[1] in fileExtList]
      
  # sys.modules[FileInfo.__module__] 獲取FileInfo.__module__模塊,其中FileInfo.__module__在此會是 main,
  # 如果被別的模塊調用的話就不是了,這是爲什麼不直接用“main”
  def getFileInfoClass(filename, module=sys.modules[FileInfo.__module__]):
    "定義一個函數,獲取文件的信息"
     # 獲取需要用來解析的類,如果是mp3文件結果爲MP3FileInfo,其他爲FileInfo
    subclass = "%sFileInfo" % os.path.splitext(filename)[1].upper()[1:]
    # 返回一個類,注意,返回的是一個“類”。使用getattr獲取moudle模塊中的subclass類
    return hasattr(module, subclass) and getattr(module, subclass) or FileInfo
  # 注意,這句話可能比較難理解, getFileInfoClass(f) (f)爲什麼會有兩個(f)呢,上面已經說過getFileInfoClass(f)
  # 根據文件名返回一個解析類,這兒是返回就是MP3FileInfo,而第二個(f)就表示對這個類以f初始化MP3FileInfo(f)
  return [getFileInfoClass(f) (f) for f in fileList]
  
if __name__ == "__main__": # main函數,在別的模塊中不會允許這裏面的代碼了
  for info in listDirectory("E:\\Music", [".mp3"]): # 循環獲取E:\\Music文件夾中所有的mp3文件的信息
    # 由於MP3FileInfo繼承自FileInfo,FileInfo繼承自UserDict,這個的items()就是獲取key-value集合。
    # 使用"%s=%s"格式化輸出,使用"\n".join將所有信息以換行連接。
    print "\n".join(["%s=%s" % (k, v) for k, v in info.items()])
    print # 每一個文件之後,輸出一個空行

結果爲:

album=What Are Words - Single
 
comment=pythontab
 
name=E:\Music\chris medina - what_are_words.mp3
 
title=What Are Words
 
artist=Chris Medina
 
year=2011
 
genre=13
 
  
 
album=After the Wedding
 
comment=pythontab
 
name=E:\Music\two fathers.mp3
 
title=Two Fathers
 
artist=pythontab
 
year=2010
 
genre=255

注意:邏輯比較多,代碼不算少,不懂的多看註釋
最後給大家推薦一個口碑不錯的python聚集地【點擊進入】,這裏有很多的老前輩學習技巧,學習心得,面試技巧,職場經歷等分享,更爲大家精心準備了零基礎入門資料,實戰項目資料,每天都有程序員定時講解Python技術,分享一些學習的方法和需要留意的小細節

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章