Update

AlgorithmAlchemy · AlgorithmAlchemy · commit 09043ae23735 · 2025-04-30T20:37:07.000+03:00
diff --git a/README.md b/README.md
@@ -18,6 +18,10 @@
 Программа предоставляет функциональность для визуализации данных в виде гистограмм и круговых диаграмм, отображая количество различных импортируемых библиотек и их использование в проекте. Это позволяет разработчикам увидеть, какие библиотеки они активно используют, а также улучшить управление зависимостями.
 
 
+## Принцип работы
+<img src="https://github.com/user-attachments/assets/2a8bd464-d3ed-41d0-88bc-ce51682ba76e" style="width: 50%;" />
+
+
 
 ## Функциональность
 
diff --git a/main.py b/main.py
@@ -1,6 +1,9 @@
+# main.py
+
 import os
 import re
 import threading
+from threading import Event
 from collections import Counter
 import matplotlib.pyplot as plt
 from tkinter import Tk, filedialog, Label, Button, Text, Scrollbar, Frame, Menu
@@ -10,36 +13,25 @@
 import queue
 from concurrent.futures import ThreadPoolExecutor, as_completed
 import ast
+from stats_window import open_stats_window, analyze_project_structure, parse_python_files
+
+from utils import read_gitignore, is_ignored
+
 init(autoreset=True)
+stop_event = Event()
 
 # Очередь для передачи данных между потоками
 imports_count = {}
 task_queue = queue.Queue()
 
 
+
+
 # =========================
 # Функции для анализа файлов
 # =========================
 
-def read_gitignore(directory):
-    gitignore_path = os.path.join(directory, '.gitignore')
-    ignored_paths = set()
 
-    if os.path.exists(gitignore_path):
-        with open(gitignore_path, 'r') as f:
-            for line in f:
-                line = line.strip()
-                if line and not line.startswith('#'):
-                    ignored_paths.add(line)
-    return ignored_paths
-
-
-def is_ignored(file_path, ignored_paths):
-    relative_path = os.path.relpath(file_path)
-    for ignored_path in ignored_paths:
-        if relative_path.startswith(ignored_path):
-            return True
-    return False
 
 def get_gitignore_excluded_dirs(gitignore_path='.gitignore'):
     excluded_dirs = []
@@ -93,7 +85,7 @@ def find_imports_in_file(file_path):
 
 
 
-def scan_directory_for_imports_parallel(directory, progress_label, output_text):
+def scan_directory_for_imports_parallel(directory, progress_label, output_text,task_queue, stop_event):
     global imports_count, total_imports
 
     ignored_paths = read_gitignore(directory)
@@ -120,6 +112,8 @@ def scan_directory_for_imports_parallel(directory, progress_label, output_text):
     imports_list = []
 
     def process_file(file_path):
+        if stop_event.is_set():
+            return []
         return find_imports_in_file(file_path)
 
     with ThreadPoolExecutor(max_workers=20) as executor:
@@ -139,6 +133,11 @@ def process_file(file_path):
 
     task_queue.put(('stats', imports_count, total_imports))
 
+    progress_label.config(text="Анализ структуры проекта...")
+    progress_label.update()
+    # После анализа импортов
+    analyze_project_structure(directory, task_queue)
+
 
 
 
@@ -151,7 +150,7 @@ def browse_directory():
     if directory:
         excluded_dirs = get_gitignore_excluded_dirs()  # Получаем исключенные директории из .gitignore
         threading.Thread(target=scan_directory_for_imports_parallel,
-                         args=(directory, progress_label, output_text),  # Передаем только 3 аргумента
+                         args=(directory, progress_label, output_text, task_queue, stop_event),  # Передаем только 3 аргумента
                          daemon=True).start()
 
 
@@ -274,27 +273,46 @@ def on_copy(event=None):
 def update_gui():
     try:
         message = task_queue.get_nowait()
+
         if isinstance(message, str):
             progress_label.config(text=message)
+
         elif isinstance(message, tuple) and message[0] == 'stats':
             imports_count, total_imports = message[1], message[2]
             print_import_statistics(imports_count, total_imports, output_text)
             plot_import_statistics(imports_count, total_imports)
+
+        elif isinstance(message, tuple) and message[0] == 'project_stats':
+            structure = message[1]
+            output_text.insert("insert", "\n--- Статистика проекта ---\n")
+            output_text.insert("insert", f"Всего файлов: {structure['total_files']}\n")
+            output_text.insert("insert", f"Всего директорий: {structure['total_dirs']}\n")
+            output_text.insert("insert", f"Python файлов: {structure['py_files']} (моего кода)\n")
+            output_text.insert("insert", f"Python файлов в виртуальных/служебных папках: {structure['py_files_venv']}\n")
+            output_text.insert("insert", f"Прочих файлов: {structure['other_files']}\n")
+            output_text.insert("insert", f"\nСписок директорий:\n")
+            for folder in structure['folders']:
+                output_text.insert("insert", f" - {folder}\n")
+
     except queue.Empty:
         pass
-    window.after(100, update_gui)  # Обновляем GUI каждую сотую долю секунды
 
+    window.after(100, update_gui)
 
 # =========================
 # GUI
 # =========================
 
+def on_closing():
+    stop_event.set()  # Сигнал остановки для потоков
+    window.destroy()  # Закрытие окна
 
 
 
 window = Tk()
 window.title("Статистика импортов в проектах")
 window.geometry("1200x800")
+window.protocol("WM_DELETE_WINDOW", on_closing)
 
 frame = Frame(window)
 frame.pack(pady=20)
@@ -333,6 +351,16 @@ def update_gui():
 btn_others = Button(lib_frame, text="Показать прочие библиотеки", command=lambda: show_others(imports_count, total_imports, output_text))
 btn_others.pack(side="left", padx=10)
 
+# Получаем данные о проектах для анализа
+default_project_path = "E:/Code/PYTHON/projects"  # <-- путь к проектам
+project_data = list(parse_python_files(default_project_path).values())
+
+
+# Новая кнопка: временной анализ проектов
+btn_stats_by_date = Button(lib_frame, text="Анализ проектов по дате", command=lambda: open_stats_window(window, project_data) )
+btn_stats_by_date.pack(side="left", padx=10)
+
+
 # Добавляем контекстное меню для копирования
 context_menu = Menu(window, tearoff=0)
 context_menu.add_command(label="Копировать", command=on_copy)
@@ -349,7 +377,10 @@ def show_context_menu(event):
 progress_label.pack(pady=10)
 
 # Запуск функции обновления GUI
-window.after(100, update_gui)
-
-window.mainloop()
+def periodic_check():
+    update_gui()
+    window.after(100, periodic_check)  # каждые 100 мс проверяем очередь
 
+# Запуск Tkinter
+periodic_check()
+window.mainloop()
diff --git a/stats_window.py b/stats_window.py
@@ -0,0 +1,148 @@
+# stats_window.py
+
+import tkinter as tk
+import pandas as pd
+import matplotlib.pyplot as plt
+from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
+import os
+import ast
+import datetime
+
+from utils import read_gitignore, is_ignored
+
+def analyze_project_structure(directory, task_queue):
+    ignored_paths = read_gitignore(directory)
+
+    structure = {
+        'total_files': 0,
+        'total_dirs': 0,
+        'py_files': 0,
+        'py_files_venv': 0,
+        'other_files': 0,
+        'folders': []
+    }
+
+    venv_like = ('venv', '.venv', 'env', '.env', '__pycache__', '.git', '.idea', '.vscode', '.mypy_cache')
+
+    for root, dirs, files in os.walk(directory):
+        # исключаем сразу ненужные папки
+        dirs[:] = [d for d in dirs if d not in venv_like and not is_ignored(os.path.join(root, d), ignored_paths)]
+
+        structure['total_dirs'] += len(dirs)
+        structure['total_files'] += len(files)
+
+        for file in files:
+            file_path = os.path.join(root, file)
+            if file.endswith('.py'):
+                if any(p in file_path for p in venv_like) or is_ignored(file_path, ignored_paths):
+                    structure['py_files_venv'] += 1
+                else:
+                    structure['py_files'] += 1
+            else:
+                structure['other_files'] += 1
+
+        relative_root = os.path.relpath(root, directory)
+        structure['folders'].append(relative_root)
+
+    task_queue.put(('project_stats', structure))
+
+
+
+"""
+def parse_python_files(projects_dir):
+    project_stats = {}
+
+    for root, dirs, files in os.walk(projects_dir):
+        project_name = os.path.relpath(root, projects_dir).split(os.sep)[0]
+
+        if project_name not in project_stats:
+            project_stats[project_name] = {
+                "py_count": 0,
+                "libs": set(),
+                "created": None,
+                "dirs": set()
+            }
+
+        for file in files:
+            if file.endswith(".py"):
+                file_path = os.path.join(root, file)
+
+                # Обновляем счётчик Python файлов
+                project_stats[project_name]["py_count"] += 1
+
+                # Обновляем дату создания проекта
+                creation_time = os.path.getctime(file_path)
+                creation_date = datetime.datetime.fromtimestamp(creation_time)
+
+                current_created = project_stats[project_name]["created"]
+                if current_created is None or creation_date < current_created:
+                    project_stats[project_name]["created"] = creation_date
+
+                # Сбор директорий
+                rel_dir = os.path.relpath(root, os.path.join(projects_dir, project_name))
+                if rel_dir != ".":
+                    project_stats[project_name]["dirs"].add(rel_dir)
+
+                # Парсим файл для библиотек
+                try:
+                    with open(file_path, "r", encoding="utf-8") as f:
+                        node = ast.parse(f.read(), filename=file_path)
+
+                    for sub_node in ast.walk(node):
+                        if isinstance(sub_node, ast.Import):
+                            for alias in sub_node.names:
+                                project_stats[project_name]["libs"].add(alias.name.split('.')[0])
+                        elif isinstance(sub_node, ast.ImportFrom):
+                            if sub_node.module:
+                                project_stats[project_name]["libs"].add(sub_node.module.split('.')[0])
+                except Exception:
+                    continue
+
+    # Приведение к нужному формату
+    for proj in project_stats:
+        project_stats[proj]["libs"] = sorted(project_stats[proj]["libs"])
+        project_stats[proj]["dirs"] = sorted(project_stats[proj]["dirs"])
+        if project_stats[proj]["created"]:
+            project_stats[proj]["created"] = project_stats[proj]["created"].strftime("%Y-%m-%d %H:%M:%S")
+
+    return project_stats
+"""
+
+
+def open_stats_window(root, project_data: list[dict]):
+    stats_win = tk.Toplevel(root)
+    stats_win.title("Статистика проектов")
+    stats_win.geometry("800x600")
+
+    df = pd.DataFrame(project_data)
+    if df.empty or 'date' not in df.columns:
+        tk.Label(stats_win, text="Недостаточно данных для отображения статистики.").pack(pady=20)
+        return
+
+    df['date'] = pd.to_datetime(df['date'])
+
+    stats_canvas = tk.Frame(stats_win)
+    stats_canvas.pack(fill="both", expand=True)
+
+    # === График по дате создания ===
+    fig, ax = plt.subplots(figsize=(6, 4))
+    df_by_month = df.groupby(df['date'].dt.to_period('M')).size().sort_index()
+    df_by_month.plot(kind='bar', ax=ax, title='Проекты по месяцам', rot=45)
+    fig.tight_layout()
+
+    canvas = FigureCanvasTkAgg(fig, master=stats_canvas)
+    canvas.draw()
+    canvas.get_tk_widget().pack(pady=10)
+
+    # === Анализ технологий ===
+    if 'stack' in df.columns:
+        all_stacks = sum(df['stack'].tolist(), [])
+        stack_series = pd.Series(all_stacks).value_counts()
+        stack_text = "\n".join(f"{lang}: {count}" for lang, count in stack_series.items())
+    else:
+        stack_text = "Нет данных по стеку технологий."
+
+    tk.Label(stats_canvas, text="Анализ технологий:\n" + stack_text, justify="left", font=("Arial", 12)).pack(pady=5)
+
+    # === Кнопка закрытия ===
+    tk.Button(stats_canvas, text="Закрыть", command=stats_win.destroy).pack(pady=10)
diff --git a/utils.py b/utils.py
@@ -0,0 +1,22 @@
+import os
+
+
+def read_gitignore(directory):
+    gitignore_path = os.path.join(directory, '.gitignore')
+    ignored_paths = set()
+
+    if os.path.exists(gitignore_path):
+        with open(gitignore_path, 'r') as f:
+            for line in f:
+                line = line.strip()
+                if line and not line.startswith('#'):
+                    ignored_paths.add(line)
+    return ignored_paths
+
+
+def is_ignored(file_path, ignored_paths):
+    relative_path = os.path.relpath(file_path)
+    for ignored_path in ignored_paths:
+        if relative_path.startswith(ignored_path):
+            return True
+    return False
