Сортировка изображений по разрешению с помощью Python и PIL

Захотелось зарегистрироваться на Хабре, но так как знаниями особо не блещу, а аудитория тут как раз наоборот, решил попробовать выложить скрипт на питоне, который написал по просьбе знакомого за бутылку 7апа :) Мне скриптик так же помог упорядочить свалку картинок ~15гб.

Скрипт проходит по директории и создает в ней папки вида ШиринахВысота и пихает туда соответствующие по разрешению изображения.

P.S. Ошибок не совершает тот, кто ничего не делает.

Copy Source | Copy HTML

1. #!/usr/bin/env python
2. # -*- coding: UTF-8 -*-
3. """ sorts images by resolution"""
4.  
5. import os,sys
6. from PIL import Image
7.  
8. # задаёт директорию для сортировки
9. dirname = os.path.abspath(sys.argv[1])
10. try:
11.     newdir = os.path.abspath(sys.argv[2])
12. except:
13.     newdir = dirname
14.  
15. def image_sort(dirname, newdir, recur= 0):
16.     if not recur:print 'sorting started ...' # если главная папка
17.     else: print 'sorting started in %s...'%dirname # если подпапка
18.  
19. # собирает все подпапки в список и рекурсивно обходит
20.  
21.     imagelist = []
22.  
23.     if os.path.isdir(dirname):
24.         for x in os.listdir(dirname):
25.             absx = dirname+os.sep+x
26.             if os.path.isfile(absx):imagelist.append(absx)
27.             else:
28.                 #print 'summon subsort in %s'%x
29.                 image_sort(absx, newdir+os.sep+x,recur=1)
30.         # проходит по содержимому папки/подпапки
31.         for name in imagelist:
32.             try:
33.                 resolution = Image.open(name).size #получить разрешение
34.             except IOError:
35.                 print 'seems not image: '+ name, '/n'
36.                 continue
37.             imdir = '%sx%s'%(resolution[ 0],resolution[1])
38.             imdir = os.path.join(newdir,imdir)
39.             #если имя папки такое же как и разрешение картинки
40.             if os.path.split(dirname)[-1] == os.path.split(imdir)[-1]:
41.                 continue
42.             elif not os.path.exists(imdir):
43.                 #print 'making dir %s'%imdir
44.                 os.mkdir(imdir)
45.             try:
46.                 os.system('move "%s" "%s"'%(name,imdir))
47.             except WindowsError:
48.                 print 'error with '+ name, '/n'
49.     if not recur:print 'sorting completed!'
50.  
51. if __name__ == '__main__':
52.     image_sort(dirname, newdir)