مدیریت فایل‌ها

fm = FileManager()

# ساختار اولیه
fm.create_dir('test_fm', '.')
fm.create_dir('test1', 'test_fm')
fm.create_dir('test2', 'test_fm/test1')

# ایجاد فایل‌ها
fm.create_file('document.txt', 'test_fm')
fm.create_file('document.txt', 'test_fm/test1')
fm.create_file('document.txt', 'test_fm/test1/test2')
fm.create_file('another.txt', 'test_fm')

test_fm/
    ├── test1/
        │   ├── test2/
            │   │   └── document.txt
        │   └── document.txt
    ├── document.txt
    └── another.txt

print(fm.find('document.txt', 'test_fm'))

['test_fm/document.txt', 'test_fm/test1/document.txt', 'test_fm/test1/test2/document.txt']

fm.delete('document.txt', 'test_fm')
fm.delete('document.txt', 'test_fm/test1/')
fm.delete('document.txt', 'test_fm/test1/test2')

test_fm/
    ├── test1/
        │   └── test2/
    └── another.txt

fm.restore('document.txt')
fm.restore('document.txt')

test_fm/
    ├── test1/
        │   ├── test2/
            │   │   └── document.txt
        │   └── document.txt
    └── another.txt

print(fm.find('document.txt', 'test_fm'))

['test_fm/test1/document.txt', 'test_fm/test1/test2/document.txt']

import os
import shutil
import tempfile

class FileManager:
    def __init__(self):
        # دیکشنری برای ذخیره‌ی فایل‌های حذف‌شده به صورت LIFO بر اساس نام
        self.trash = {}
        # پوشه‌ی موقت برای ذخیره‌ی فایل‌های حذف‌شده
        self.trash_dir = tempfile.mkdtemp()

    def _get_full_path(self, address, name):
        if address == '.':
            return name
        return os.path.join(address, name)

    def create_dir(self, name, address):
        path = self._get_full_path(address, name)
        if not os.path.exists(path):
            os.makedirs(path, exist_ok=True)

    def create_file(self, name, address):
        path = self._get_full_path(address, name)
        if not os.path.exists(path):
            open(path, 'w').close()

    def delete(self, name, address):
        path = self._get_full_path(address, name)
        if os.path.exists(path) and os.path.isfile(path):
            # ذخیره‌سازی در سطل زباله
            if name not in self.trash:
                self.trash[name] = []

            # ایجاد یک نام یکتا برای فایل در سطل زباله
            trash_path = os.path.join(self.trash_dir, f"{name}_{len(self.trash[name])}")
            # کپی کردن فایل به سطل زباله (با حفظ محتوا)
            shutil.copy2(path, trash_path)
            self.trash[name].append({'trash_path': trash_path, 'original_path': path})
            # حذف فایل اصلی
            os.remove(path)

    def find(self, name, address):
        result = []
        for root, dirs, files in os.walk(address):
            if name in files:
                result.append(os.path.join(root, name))
        return result

    def restore(self, name):
        if name in self.trash and self.trash[name]:
            # LIFO: آخرین فایل اضافه‌شده را بازیابی می‌کند
            entry = self.trash[name].pop()
            trash_path = entry['trash_path']
            original_path = entry['original_path']

            # اطمینان از وجود پوشه‌ی مقصد
            os.path.dirname(original_path)
            os.makedirs(os.path.dirname(original_path), exist_ok=True)

            # بازیابی فایل به مسیر اصلی
            shutil.copy2(trash_path, original_path)

            # حذف فایل از سطل زباله
            os.remove(trash_path)

            # اگر لیست خالی شد، کلید را حذف می‌کنیم
            if not self.trash[name]:
                del self.trash[name]