sorting/main.py at main · freshdeadboy/sorting · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
from pathlib import Path
import shutil
import sys
import file_parser
from normalize import normalize

def handle_media(file_name: Path, target_folder: Path):
    target_folder.mkdir(exist_ok=True, parents=True)
    file_name.replace(target_folder / normalize(file_name.name))

def handle_archive(file_name: Path, target_folder: Path):
    target_folder.mkdir(exist_ok=True, parents=True)
    folder_for_file = target_folder / normalize(file_name.name.replace(file_name.suffix, ''))
    folder_for_file.mkdir(exist_ok=True, parents=True)
    try:
        shutil.unpack_archive(str(file_name.absolute()), str(folder_for_file.absolute()))
    except shutil.ReadError:
        folder_for_file.rmdir()
        return
    file_name.unlink()


def main(folder: Path):
    file_parser.scan(folder)
    for file in file_parser.JPEG_IMAGES:
        handle_media(file, folder / 'images' / 'JPEG')
    for file in file_parser.JPG_IMAGES:
        handle_media(file, folder / 'images' / 'JPG')
    for file in file_parser.PNG_IMAGES:
        handle_media(file, folder / 'images' / 'PNG')
    for file in file_parser.SVG_IMAGES:
        handle_media(file, folder / 'images' / 'SVG')
    for file in file_parser.AVI_VIDEO:
        handle_media(file, folder / 'videos' / 'AVI')
    for file in file_parser.MP4_VIDEO:
        handle_media(file, folder / 'videos' / 'MP4')
    for file in file_parser.MOV_VIDEO:
        handle_media(file, folder / 'videos' / 'MOV')
    for file in file_parser.MKV_VIDEO:
        handle_media(file, folder / 'videos' / 'MKV')
    for file in file_parser.DOC_DOCUMENT:
        handle_media(file, folder / 'documents' / 'DOC')
    for file in file_parser.DOCX_DOCUMENT:
        handle_media(file, folder / 'documents' / 'DOCX')
    for file in file_parser.TXT_DOCUMENT:
        handle_media(file, folder / 'documents' / 'TXT')
    for file in file_parser.PDF_DOCUMENT:
        handle_media(file, folder / 'documents' / 'PDF')
    for file in file_parser.XLSX_DOCUMENT:
        handle_media(file, folder / 'documents' / 'XLSX')
    for file in file_parser.PPTX_DOCUMENT:
        handle_media(file, folder / 'documents' / 'PPTX')
    for file in file_parser.MP3_AUDIO:
        handle_media(file, folder / 'music' / 'MP3')
    for file in file_parser.OGG_AUDIO:
        handle_media(file, folder / 'music' / 'OGG')
    for file in file_parser.WAV_AUDIO:
        handle_media(file, folder / 'music' / 'WAV')
    for file in file_parser.AMR_AUDIO:
        handle_media(file, folder / 'music' / 'AMR')
    for file in file_parser.ZIP_ARCHIVES:
        handle_archive(file, folder / 'archives' / 'ZIP')
    for file in file_parser.GZ_ARCHIVES:
        handle_archive(file, folder / 'archives' / 'GZ')
    for file in file_parser.TAR_ARCHIVES:
        handle_archive(file, folder / 'archives' / 'TAR')
    for file in file_parser.MY_OTHER:
        handle_media(file, folder / 'MY_OTHER')
    for folder in file_parser.FOLDERS[::-1]:

        # Видаляємо пусті папки після сортування
        try:
            folder.rmdir()
        except OSError:
            print(f'Error during remove folder {folder}')
    for file in file_parser.MY_OTHER:
        handle_media(file, folder / 'MY_OTHER')