Привет, Хабр! Думаю, всем знакома ситуация: десятки файлов на рабочем столе, сваленных в единую кучу. Скрины, документы, архивы — и всё в одном рабочем пространстве. Наводить порядок не всегда получается, а жить в беспорядке не очень удобно. Недавно, в процессе наведения порядка, я утомился делать это руками и очень захотел написать скрипт на Python, который структурирует и раскидает всё по папкам самостоятельно.
Скрипт в статье:
Автоматически сортирует файлы по 9 категориям
Поддерживает 50+ форматов файлов
Предоставляет гибкие настройки
Введение
Начну с небольшой «демонстрации возможностей» скрипта. Работа со скриптом начинается с настройки конфига, где можно указать не только папку для сортировки, но и некоторые правила, такие как: создание подпапки по дате (полезно для скринов/фото), пропуск скрытых файлов, создание бэкапов, тестовый режим (покажет все действия без реального перемещения), а также возможность добавлять новые расширения и менять структуру сортировки. Всё это подкрепляется логированием.
Сам скрипт:
import os
import shutil
import argparse
from datetime import datetime
def setup_config():
"""Настройки по умолчанию с возможностью переопределения"""
return {
# Основные параметры
'target_folder': os.path.expanduser("~/Desktop"),
'log_file': os.path.expanduser("~/Desktop/sort_files.log"),
'date_format': "%Y-%m-%d %H:%M:%S",
# Правила сортировки (можно расширять, менять и тд.)
'sort_rules': {
"Images": [".jpg", ".jpeg", ".png", ".gif", ".webp", ".bmp", ".svg", ".tiff"],
"Documents": [".pdf", ".docx", ".xlsx", ".pptx", ".txt", ".rtf", ".odt", ".csv"],
"Archives": [".zip", ".rar", ".7z", ".tar", ".gz", ".bz2"],
"Media": [".mp3", ".mp4", ".avi", ".mkv", ".mov", ".wav", ".flac"],
"Code": [".py", ".js", ".html", ".css", ".json", ".xml", ".php", ".sh"],
"Executables": [".exe", ".msi", ".dmg", ".pkg", ".deb"],
"Design": [".psd", ".ai", ".sketch", ".fig"],
"Data": [".sql", ".db", ".sqlite", ".mdb"],
"Other": [] # Для нераспознанных расширений
},
# Дополнительные опции
'create_date_folders': False, # Создавать подпапки по дате
'skip_hidden': True, # Пропускать скрытые файлы
'dry_run': False, # Тестовый режим без реальных действий
'backup_files': False # Создавать копии перед перемещением
}
def create_folders(target_folder, folders, dry_run=False):
"""Создание папки для категорий"""
for folder in folders:
path = os.path.join(target_folder, folder)
if not dry_run:
os.makedirs(path, exist_ok=True)
else:
print(f"[DRY RUN] Создана папка: {path}")
def get_file_category(filename, sort_rules):
"""Определение категории файла по его расширению"""
ext = os.path.splitext(filename)[1].lower()
for category, extensions in sort_rules.items():
if ext in extensions:
return category
return "Other"
def backup_file(src, dest, dry_run=False):
"""Создание резервной копии файла перед перемещением"""
backup_path = dest + ".backup"
if not dry_run:
shutil.copy2(src, backup_path)
return backup_path
def sort_files(config):
"""Основная функция сортировки файлов"""
target_folder = config['target_folder']
sort_rules = config['sort_rules']
folders = list(sort_rules.keys())
# Создание лог-файла
with open(config['log_file'], 'a') as log:
log.write(f"\n=== Сортировка начата: {datetime.now().strftime(config['date_format'])} ===\n")
# Создание необходимых папок
create_folders(target_folder, folders, config['dry_run'])
# Обработка файлов
for item in os.listdir(target_folder):
item_path = os.path.join(target_folder, item)
# Пропуск папок и скрытых файлов (если включено)
if not os.path.isfile(item_path):
continue
if config['skip_hidden'] and item.startswith('.'):
continue
# Определение категории
category = get_file_category(item, sort_rules)
# Формировка пути назначения
dest_folder = os.path.join(target_folder, category)
# Опционально: создание подпапки по дате
if config['create_date_folders']:
today = datetime.now().strftime("%Y-%m-%d")
dest_folder = os.path.join(dest_folder, today)
if not config['dry_run']:
os.makedirs(dest_folder, exist_ok=True)
dest_path = os.path.join(dest_folder, item)
# Логирование
action = f"[+] {item} -> {category}"
if config['create_date_folders']:
action += f"/{today}"
print(action)
log.write(action + "\n")
# Перемещение (или тестовый режим)
if not config['dry_run']:
# Создание резервной копии при необходимости
if config['backup_files']:
backup_path = backup_file(item_path, dest_path)
log.write(f" Создана резервная копия: {backup_path}\n")
# Перемещение файла
try:
shutil.move(item_path, dest_path)
except Exception as e:
error_msg = f"Ошибка при перемещении {item}: {str(e)}"
print(error_msg)
log.write(error_msg + "\n")
def main():
"""Основная функция с обработкой аргументов командной строки"""
parser = argparse.ArgumentParser(
description="Скрипт для сортировки файлов на рабочем столе по категориям",
epilog="Пример использования: python sort_desktop.py --target ~/Desktop --dry-run"
)
# Аргументы командной строки
parser.add_argument('--target', help="Папка для сортировки (по умолчанию: Desktop)")
parser.add_argument('--log', help="Файл для записи логов")
parser.add_argument('--dry-run', action='store_true', help="Тестовый режим без реальных действий")
parser.add_argument('--date-folders', action='store_true', help="Создавать подпапки по дате")
parser.add_argument('--no-skip-hidden', action='store_true', help="Не пропускать скрытые файлы")
parser.add_argument('--backup', action='store_true', help="Создавать резервные копии файлов")
parser.add_argument('--show-config', action='store_true', help="Показать текущие настройки и выйти")
args = parser.parse_args()
# Загрузка конфигурации
config = setup_config()
# Переопределение настройки из аргументов
if args.target:
config['target_folder'] = os.path.expanduser(args.target)
if args.log:
config['log_file'] = os.path.expanduser(args.log)
if args.dry_run:
config['dry_run'] = True
if args.date_folders:
config['create_date_folders'] = True
if args.no_skip_hidden:
config['skip_hidden'] = False
if args.backup:
config['backup_files'] = True
# Показ конфигурации (если нужно)
if args.show_config:
print("Текущие настройки:")
for key, value in config.items():
print(f" {key}: {value}")
return
# Запуск сортировки
print("=== Начало сортировки ===")
print(f"Целевая папка: {config['target_folder']}")
print(f"Режим dry-run: {'да' if config['dry_run'] else 'нет'}")
sort_files(config)
print("\n=== Сортировка завершена ===")
print(f"Подробности в лог-файле: {config['log_file']}")
if __name__ == "__main__":
main()Структура скрипта
Скрипт состоит из нескольких ключевых функций:
setup_config()- инициализация настроек по умолчаниюcreate_folders()- создание папок для категорийget_file_category()- определение категории файла по расширениюbackup_file()- создание резервной копии файлаsort_files()- основная функция сортировкиmain()- обработка аргументов командной строки и запуск процесса
Настройка конфигурации
Функция setup_config() возвращает словарь с настройками:
{
'target_folder': "~/Desktop", # Папка для сортировки
'log_file': "~/Desktop/sort_files.log", # Файл для логов
'date_format': "%Y-%m-%d %H:%M:%S", # Формат даты в логах
# Правила сортировки по расширениям
'sort_rules': {
"Images": [".jpg", ".jpeg", ".png", ...],
"Documents": [".pdf", ".docx", ".xlsx", ...],
},
# Дополнительные опции
'create_date_folders': False, # Создавать подпапки по дате
'skip_hidden': True, # Пропускать скрытые файлы
'dry_run': False, # Тестовый режим
'backup_files': False # Создавать копии файлов
}Алгоритм работы
Инициализация: загрузка конфигурации, обработка аргументов командной строки
Подготовка: создание папок для всех категорий из
sort_rulesОбработка файлов:
Для каждого файла в целевой папке определяется категория
Формируется путь назначения (с учетом подпапок по дате при необходимости)
Создается резервная копия (если включено)
Файл перемещается в соответствующую папку
Логирование: все действия записываются в лог-файл
Обработка аргументов командной строки
Скрипт использует модуль argparse для гибкой настройки:
parser = argparse.ArgumentParser(
description="Скрипт для сортировки файлов на рабочем столе по категориям",
epilog="Пример использования: python sort.py --target ~/Desktop --dry-run"
)
parser.add_argument('--target', help="Папка для сортировки (по умолчанию: Desktop)")
parser.add_argument('--log', help="Файл для записи логов")
parser.add_argument('--dry-run', action='store_true', help="Тестовый режим без реальных действий")Это позволяет легко изменять поведение скрипта без редактирования кода.
Пример и демонстрация использования
Для демонстрации наведу искусственный беспорядок на рабочем столе:
И, командой в cmd (python sort.py - базовая сортировка) запустим скрипт. Результат:
А также, можем заглянуть в файл лога:
Далее, стоит рассмотреть остальные варианты команд. Начнём с тестового запуска для проверки (python sort.py --dry-run):
В данном случае, все наши файлы остались на месте. Данная команда нужна больше для настройки, чтобы ничего случайно не потерять. Далее, рассмотрим сортировку по дате (python sort.py --date-folders):
Папки остались всё те же, но в них появились подпапки с меткой по дате. Выглядит это следующим образом:
Аналогично всем инструментам, есть поддержка продвинутых сценариев, пару примеров которых я приведу:
# Сортировка с резервными копиями и подпапками по дате
python sort.py --backup --date-folders
# Тестовый прогон для папки "Загрузки"
python sort.py --target ~/Downloads --dry-runЗаключение
Представленный в статье скрипт — хорошее решение для тех, кто устал от хаоса на рабочем столе. Он не требует глубоких знаний Python для использования и настройки, но предлагает достаточно широкий функционал по сортировке.
Хочу спросить у сообщества:
Какие функции вы бы добавили в подобный инструмент?
Интересна ли вам версия с графическим интерфейсом?
Стоит ли развивать это решение в полноценный набор инструментов для автоматизации рутины?
Спрашиваю из-за желания изучить создание ПО. В процессе написания скрипта подумал, что добавление GUI/шаблонов/тегов в данном решении смотрелись бы отлично. Да, возможно скрипт не найдет широкого применения и по большей части для многих будет бесполезен, но не поделиться им не мог. Он решил мою вечную проблему свалки.
P.S. Я веду свою группу в Телеграмм, буду рад видеть всех, кому интересен процесс написания скриптов и автоматизация в мире IT.