main.py

# -*- coding: utf-8 -*-
import os
import sys
import csv
import logging
from logging.handlers import RotatingFileHandler
from xml.etree.ElementTree import Element, SubElement, ElementTree

from PyQt5 import QtCore, QtGui, QtWidgets


APP_NAME = "ساختار درختی فولدر - خروجی XML/CSV"
LOG_FILENAME = "folder_tree_app.log"
VAZIR_FONT_FILE = "Vazirmatn-Bold.ttf"  # در صورت وجود کنار فایل، اعمال می‌شود

# -----------------------------
# لاگ‌نویسی پیشرفته
# -----------------------------
logger = logging.getLogger("folder_tree")
logger.setLevel(logging.INFO)

# هندلر چرخشی فایل لاگ
file_handler = RotatingFileHandler(LOG_FILENAME, maxBytes=1_000_000, backupCount=3, encoding="utf-8")
file_handler.setFormatter(logging.Formatter("%(asctime)s | %(levelname)s | %(message)s"))
logger.addHandler(file_handler)

# هندلر استریم برای کنسول (در صورت اجرای از ترمینال)
stream_handler = logging.StreamHandler()
stream_handler.setFormatter(logging.Formatter("%(levelname)s | %(message)s"))
logger.addHandler(stream_handler)


def human_size(num_bytes: int) -> str:
    # تبدیل اندازه به قالب خوانا
    for unit in ["B", "KB", "MB", "GB", "TB"]:
        if num_bytes < 1024.0:
            return f"{num_bytes:.1f} {unit}"
        num_bytes /= 1024.0
    return f"{num_bytes:.1f} PB"


def walk_directory(root_path: str):
    """
    تولید ساختار درختی به صورت دیکشنری‌های تو در تو:
    {
      'name': 'Folder',
      'path': '/abs/path',
      'type': 'dir|file',
      'size': int or None,
      'children': [...]
    }
    """
    root_node = {
        "name": os.path.basename(root_path) or root_path,
        "path": os.path.abspath(root_path),
        "type": "dir",
        "size": None,
        "children": []
    }

    for dirpath, dirnames, filenames in os.walk(root_path):
        # مرتب‌سازی برای خروجی ثابت
        dirnames.sort()
        filenames.sort()

        # گره جاری را پیدا/بسازید درختی
        # نگاشت مسیر => گره
        # برای کارایی، یک دیکشنری کمکی استفاده می‌کنیم
        # اما ساده‌تر: گره‌ها را با مسیرشان جستجو می‌کنیم (کافی است چون os.walk مسیر سلسله‌مراتبی می‌دهد)
        def find_or_create_node(current_path, parent):
            if os.path.abspath(current_path) == parent["path"]:
                return parent
            # پیمایش عمقی
            for ch in parent["children"]:
                if ch["type"] == "dir":
                    found = find_or_create_node(current_path, ch)
                    if found:
                        return found
            return None

        parent_node = find_or_create_node(dirpath, root_node)
        if parent_node is None:
            # در شرایط نادر، مسیر ممکن است هم‌خوانی نداشته باشد
            logger.warning(f"مسیر والد یافت نشد: {dirpath}")
            continue

        # پوشه‌ها
        for d in dirnames:
            full = os.path.join(dirpath, d)
            parent_node["children"].append({
                "name": d,
                "path": os.path.abspath(full),
                "type": "dir",
                "size": None,
                "children": []
            })

        # فایل‌ها
        for f in filenames:
            full = os.path.join(dirpath, f)
            try:
                size = os.path.getsize(full)
            except Exception:
                size = 0
            parent_node["children"].append({
                "name": f,
                "path": os.path.abspath(full),
                "type": "file",
                "size": size,
                "children": []
            })

    return root_node


def tree_to_xml(root_node):
    """
    تبدیل ساختار دیکشنری به XML ElementTree
    """
    root_el = Element("folder", {
        "name": root_node["name"],
        "path": root_node["path"]
    })

    def add_children(parent_el, node):
        for ch in node.get("children", []):
            if ch["type"] == "dir":
                el = SubElement(parent_el, "folder", {"name": ch["name"], "path": ch["path"]})
                add_children(el, ch)
            else:
                attrs = {"name": ch["name"], "path": ch["path"], "size": str(ch["size"] or 0)}
                SubElement(parent_el, "file", attrs)

    add_children(root_el, root_node)
    return ElementTree(root_el)


def tree_to_csv(root_node, csv_path):
    """
    تولید CSV با ستون‌ها: نوع، نام، مسیر، اندازه (خوانا)، اندازه (بایت)
    """
    with open(csv_path, "w", newline="", encoding="utf-8") as f:
        writer = csv.writer(f)
        writer.writerow(["نوع", "نام", "مسیر", "اندازه خوانا", "اندازه (بایت)"])

        def write_node(node):
            if node["type"] == "file":
                writer.writerow([
                    "فایل",
                    node["name"],
                    node["path"],
                    human_size(node["size"] or 0),
                    node["size"] or 0
                ])
            elif node["type"] == "dir":
                # برای پوشه اندازه ثبت نمی‌کنیم
                writer.writerow([
                    "پوشه",
                    node["name"],
                    node["path"],
                    "",
                    ""
                ])
                for ch in node.get("children", []):
                    write_node(ch)

        write_node(root_node)


class LogTextHandler(logging.Handler):
    """
    هندلر سفارشی برای ارسال لاگ‌ها به QTextEdit در رابط
    """
    def __init__(self, widget):
        super().__init__()
        self.widget = widget

    def emit(self, record):
        msg = self.format(record)
        # به انتهای متن اضافه شود (راست‌چین و فارسی حفظ می‌شود)
        self.widget.append(msg)


class FolderTreeApp(QtWidgets.QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle(APP_NAME)
        self.setMinimumSize(900, 600)

        # راست‌چین سراسری
        self.setLayoutDirection(QtCore.Qt.RightToLeft)

        # اعمال فونت وزیرمتن بولد در صورت وجود
        self.apply_vazir_font()

        # ویجت مرکزی و لایه‌ها
        central = QtWidgets.QWidget()
        self.setCentralWidget(central)
        main_layout = QtWidgets.QVBoxLayout(central)

        # نوار ابزار بالا
        top_bar = QtWidgets.QHBoxLayout()

        self.path_edit = QtWidgets.QLineEdit()
        self.path_edit.setPlaceholderText("مسیر فولدر را وارد کنید یا با درگ‌ و دراپ رها کنید...")
        self.path_edit.setAlignment(QtCore.Qt.AlignRight)

        browse_btn = QtWidgets.QPushButton("انتخاب فولدر")
        browse_btn.clicked.connect(self.browse_folder)

        build_btn = QtWidgets.QPushButton("ساخت درخت")
        build_btn.clicked.connect(self.build_tree)

        export_xml_btn = QtWidgets.QPushButton("خروجی XML")
        export_xml_btn.clicked.connect(self.export_xml)

        export_csv_btn = QtWidgets.QPushButton("خروجی CSV")
        export_csv_btn.clicked.connect(self.export_csv)

        top_bar.addWidget(self.path_edit, stretch=1)
        top_bar.addWidget(browse_btn)
        top_bar.addWidget(build_btn)
        top_bar.addWidget(export_xml_btn)
        top_bar.addWidget(export_csv_btn)

        main_layout.addLayout(top_bar)

        # ویوی درخت
        self.tree_view = QtWidgets.QTreeView()
        self.tree_view.setEditTriggers(QtWidgets.QAbstractItemView.NoEditTriggers)
        self.tree_view.setAlternatingRowColors(True)
        self.tree_view.setAcceptDrops(True)
        self.tree_view.setDragDropMode(QtWidgets.QAbstractItemView.DropOnly)
        self.tree_view.viewport().setAcceptDrops(True)

        # مدل درخت
        self.model = QtGui.QStandardItemModel()
        self.model.setHorizontalHeaderLabels(["نام", "نوع", "اندازه"])
        self.tree_view.setModel(self.model)
        self.tree_view.setHeaderHidden(False)
        self.tree_view.setLayoutDirection(QtCore.Qt.RightToLeft)

        main_layout.addWidget(self.tree_view, stretch=1)

        # ناحیه وضعیت و لاگ
        status_layout = QtWidgets.QHBoxLayout()
        self.status_label = QtWidgets.QLabel("آماده")
        self.status_label.setAlignment(QtCore.Qt.AlignRight)
        status_layout.addWidget(self.status_label)

        main_layout.addLayout(status_layout)

        log_label = QtWidgets.QLabel("لاگ برنامه:")
        self.log_text = QtWidgets.QTextEdit()
        self.log_text.setReadOnly(True)
        self.log_text.setLayoutDirection(QtCore.Qt.RightToLeft)

        main_layout.addWidget(log_label)
        main_layout.addWidget(self.log_text, stretch=0)

        # اتصال لاگ‌ها به رابط
        ui_handler = LogTextHandler(self.log_text)
        ui_handler.setLevel(logging.INFO)
        ui_handler.setFormatter(logging.Formatter("%(asctime)s | %(levelname)s | %(message)s"))
        logger.addHandler(ui_handler)

        # پشتیبانی درگ‌ و دراپ در سطح پنجره
        self.setAcceptDrops(True)

        self.current_tree = None  # ساختار درختی داده‌ها

    # -----------------------------
    # فونت وزیرمتن
    # -----------------------------
    def apply_vazir_font(self):
        try:
            if os.path.exists(VAZIR_FONT_FILE):
                font_id = QtGui.QFontDatabase.addApplicationFont(VAZIR_FONT_FILE)
                if font_id != -1:
                    families = QtGui.QFontDatabase.applicationFontFamilies(font_id)
                    if families:
                        font = QtGui.QFont(families[0])
                        font.setBold(True)
                        self.setFont(font)
                        logger.info(f"فونت وزیرمتن اعمال شد: {families[0]}")
                        return
            # fallback: فونت سیستم با بولد
            font = QtGui.QFont()
            font.setBold(True)
            self.setFont(font)
            logger.info("فونت سیستم (بولد) اعمال شد؛ فایل Vazirmatn-Bold.ttf یافت نشد.")
        except Exception as e:
            logger.warning(f"اعمال فونت با خطا مواجه شد: {e}")

    # -----------------------------
    # درگ‌ و دراپ
    # -----------------------------
    def dragEnterEvent(self, event: QtGui.QDragEnterEvent):
        if event.mimeData().hasUrls():
            event.acceptProposedAction()
        else:
            event.ignore()

    def dropEvent(self, event: QtGui.QDropEvent):
        urls = event.mimeData().urls()
        if urls:
            local_path = urls[0].toLocalFile()
            if os.path.isdir(local_path):
                self.path_edit.setText(local_path)
                self.build_tree()
            else:
                logger.warning("فقط فولدر قابل دراپ است.")
        event.acceptProposedAction()

    # -----------------------------
    # انتخاب فولدر
    # -----------------------------
    def browse_folder(self):
        dir_path = QtWidgets.QFileDialog.getExistingDirectory(self, "انتخاب فولدر", os.path.expanduser("~"))
        if dir_path:
            self.path_edit.setText(dir_path)
            self.build_tree()

    # -----------------------------
    # ساخت درخت
    # -----------------------------
    def build_tree(self):
        root_path = self.path_edit.text().strip()
        if not root_path or not os.path.isdir(root_path):
            self.status_label.setText("مسیر معتبر نیست.")
            logger.error("مسیر معتبر نیست یا فولدر وجود ندارد.")
            return

        self.status_label.setText("در حال ساخت درخت...")
        QtWidgets.QApplication.setOverrideCursor(QtCore.Qt.WaitCursor)
        try:
            logger.info(f"شروع ساخت درخت از: {root_path}")
            self.current_tree = walk_directory(root_path)
            self.populate_tree_view(self.current_tree)
            self.status_label.setText("درخت ساخته شد.")
            logger.info("درخت با موفقیت ساخته شد.")
        except Exception as e:
            self.status_label.setText("خطا در ساخت درخت.")
            logger.exception(f"خطا در ساخت درخت: {e}")
        finally:
            QtWidgets.QApplication.restoreOverrideCursor()

    def populate_tree_view(self, data_tree):
        self.model.removeRows(0, self.model.rowCount())

        def add_items(parent_item, node):
            name_item = QtGui.QStandardItem(node["name"])
            type_item = QtGui.QStandardItem("پوشه" if node["type"] == "dir" else "فایل")
            size_item = QtGui.QStandardItem("" if node["type"] == "dir" else human_size(node["size"] or 0))

            # راست‌چین هر ستون
            for it in (name_item, type_item, size_item):
                it.setTextAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter)

            parent_item.appendRow([name_item, type_item, size_item])

            for ch in node.get("children", []):
                add_items(name_item, ch)

        # ریشه را اضافه کنید
        root_item = self.model.invisibleRootItem()
        add_items(root_item, data_tree)
        self.tree_view.expandAll()
        self.tree_view.resizeColumnToContents(0)

    # -----------------------------
    # خروجی‌ها
    # -----------------------------
    def export_xml(self):
        if not self.current_tree:
            self.status_label.setText("ابتدا درخت را بسازید.")
            logger.warning("خروجی XML ممکن نیست؛ درخت ساخته نشده.")
            return

        default_name = f"{self.current_tree['name']}_tree.xml"
        out_path, _ = QtWidgets.QFileDialog.getSaveFileName(self, "ذخیره XML", default_name, "XML Files (*.xml)")
        if out_path:
            try:
                tree = tree_to_xml(self.current_tree)
                tree.write(out_path, encoding="utf-8", xml_declaration=True)
                self.status_label.setText("XML ذخیره شد.")
                logger.info(f"XML ذخیره شد: {out_path}")
            except Exception as e:
                self.status_label.setText("خطا در ذخیره XML.")
                logger.exception(f"خطا در ذخیره XML: {e}")

    def export_csv(self):
        if not self.current_tree:
            self.status_label.setText("ابتدا درخت را بسازید.")
            logger.warning("خروجی CSV ممکن نیست؛ درخت ساخته نشده.")
            return

        default_name = f"{self.current_tree['name']}_tree.csv"
        out_path, _ = QtWidgets.QFileDialog.getSaveFileName(self, "ذخیره CSV", default_name, "CSV Files (*.csv)")
        if out_path:
            try:
                tree_to_csv(self.current_tree, out_path)
                self.status_label.setText("CSV ذخیره شد.")
                logger.info(f"CSV ذخیره شد: {out_path}")
            except Exception as e:
                self.status_label.setText("خطا در ذخیره CSV.")
                logger.exception(f"خطا در ذخیره CSV: {e}")


def main():
    # رفع مشکلات نمایش فارسی روی برخی سیستم‌ها
    os.environ.setdefault("QT_ENABLE_HIGHDPI_SCALING", "1")
    os.environ.setdefault("QT_SCALE_FACTOR", "1")

    app = QtWidgets.QApplication(sys.argv)

    # آیکن برنامه (اختیاری)
    # icon = QtGui.QIcon("app.png")
    # app.setWindowIcon(icon)

    win = FolderTreeApp()
    win.show()

    logger.info("برنامه اجرا شد.")
    sys.exit(app.exec_())


if __name__ == "__main__":
    main()