Автоматизатор

# -*- coding: utf-8 -*-
__author__="vaulter"
__date__ ="$05.09.2009 22:03:27$"
"""
Это мой первый скрипт на python
Он переименовывает файлы, имена которых не содержат номера
согласно таблице, считанной из указанного cvs файла.

То есть, иными словами, есть таблица
One, 1
Two, 2

И есть файлы
1. One.avi
Two.avi

Скрипт пропустит 1. One.avi и переименует Two.avi -> 2. Two.avi
все :)

зы: Было написано для упорядочивания серий Dora Explorer :)
"""

import os
import os.path
import re
import csv
import difflib

class DoraSeries():
content = ""
series = {}
in_seria = 0

def __init__( self, url ):
self.url = url
listReader = csv.reader( open( self.url ) )

for row in listReader:
self.series[ row[ 0 ].strip().lower() ] = row[ 1 ].strip()

def get_seria( self, name ):
prepared = name.strip().lower()
if prepared in self.series:
return self.series[ prepared ]
else:
return 0

def get_most_close( self, name ):
max = 0
matcher = difflib.SequenceMatcher()
matcher.set_seq2( name )

for k in self.series:
matcher.set_seq1( k )
r = matcher.ratio()
if max < r:
max = r
kmax = k
print name, " ? ", k, " = ", r

if max > 0.73: #threshold
return self.series[ kmax ]

return 0

def rename_by_seria( self, filename ):
parts = os.path.split( filename )
name = os.path.splitext( os.path.basename( filename ) )[ 0 ]
seria = self.get_seria( name )

if not seria:
seria = self.get_most_close( name )

if seria > 0:
newname = parts[ 0 ] + seria + "." + parts[ 1 ]
print filename, " > ", newname
os.rename( filename, newname )

dir = "/home/vaulter/video/Dora Explorer"
wiki_page = DoraSeries( '/home/vaulter/NetBeansProjects/Dora/list.csv' );

if __name__ == "__main__":
names = os.listdir( dir );
os.chdir( dir )

for name in names:
fullname = os.path.join( dir, name ) # получаем полное имя

if os.path.isfile( fullname ): # если это файл...
print name,

if re.match( r"(\d+)\..*", name ):
print "YES"

else:
print "not match"
wiki_page.rename_by_seria( name )