start

main.py

@@ -0,0 +1,154 @@
+import os
+import csv
+import time
+import pathlib
+from datetime import datetime
+
+# Пути для поиска файлов
+SEARCH_DIRS = ["Desktop", "Downloads", "Documents"]
+EXTENSIONS = (".mov", ".avi", ".mp3", ".mp4", ".mkv", ".wav", ".flac")
+
+# Файлы с хостами и результатами
+HOSTS_FILE = "hosts.txt"
+OUTPUT_CSV = "found_files.csv"
+FAILED_CSV = "failed_hosts.csv"
+
+# Системные папки, которые не нужно сканировать
+EXCLUDE_FOLDERS = {"Administrator", "DefaultAccount", "Public", "All Users", "Default", "WDAGUtilityAccount"}
+
+# Список недоступных ПК
+failed_hosts = []
+
+def format_size(size_bytes):
+    """ Форматирование размера файла в MB/GB """
+    if size_bytes >= 1_073_741_824:
+        return f"{size_bytes / 1_073_741_824:.2f} GB"
+    elif size_bytes >= 1_048_576:
+        return f"{size_bytes / 1_048_576:.2f} MB"
+    return f"{size_bytes} bytes"
+
+def get_users_from_folders(host):
+    """ Получает список пользователей, проверяя папки в \\host\c$\Users """
+    users_path = f"\\\\{host}\\c$\\Users"
+    
+    try:
+        if not os.path.exists(users_path):
+            print(f"[Ошибка] ПК {host} недоступен!")
+            failed_hosts.append(host)
+            return []
+        
+        users = [u for u in os.listdir(users_path) if u not in EXCLUDE_FOLDERS]
+        print(f"[{host}] Найдено {len(users)} пользователей: {users}")
+        return users
+    except Exception as e:
+        print(f"[Ошибка] Не удалось получить пользователей с {host}: {e}")
+        failed_hosts.append(host)
+        return []
+
+def get_file_info(file_path):
+    """ Получает информацию о файле """
+    try:
+        file_stat = os.stat(file_path)
+        return {
+            "size_bytes": file_stat.st_size,
+            "size_formatted": format_size(file_stat.st_size),
+            "created": datetime.fromtimestamp(file_stat.st_ctime).strftime("%Y-%m-%d %H:%M:%S"),
+            "last_accessed": datetime.fromtimestamp(file_stat.st_atime).strftime("%Y-%m-%d %H:%M:%S"),
+        }
+    except Exception as e:
+        print(f"[Ошибка] Не удалось получить данные о файле {file_path}: {e}")
+        return None
+
+def wait_for_folder_access(path, timeout=10):
+    """ Открывает проводник для доступа к скрытым файлам и ждет индексации """
+    try:
+        print(f"[Открытие проводника] {path}")
+        os.startfile(path)
+        time.sleep(3)
+
+        start_time = time.time()
+        while time.time() - start_time < timeout:
+            if os.path.exists(path) and os.listdir(path):
+                print(f"[Готово] Папка доступна: {path}")
+                return True
+            print(f"[Ожидание] Файлы еще не загружены... ({int(time.time() - start_time)}/{timeout} сек)")
+            time.sleep(2)
+
+        print(f"[Ошибка] Не удалось получить доступ к {path}")
+        return False
+    except Exception as e:
+        print(f"[Ошибка] Не удалось открыть {path}: {e}")
+        return False
+
+def scan_host(host):
+    """ Сканирует ПК и находит файлы с нужными расширениями """
+    results = []
+    users = get_users_from_folders(host)
+
+    if not users:
+        return results
+
+    for user in users:
+        base_path = f"\\\\{host}\\c$\\Users\\{user}"
+        if not os.path.exists(base_path):
+            print(f"[Пропуск] Папка {base_path} не найдена")
+            continue
+
+        if not wait_for_folder_access(base_path):
+            continue
+
+        for folder in SEARCH_DIRS:
+            search_path = os.path.join(base_path, folder)
+            if not os.path.exists(search_path):
+                print(f"[Пропуск] Нет папки {search_path}")
+                continue
+
+            print(f"[Поиск] {search_path}")
+            for root, _, files in os.walk(search_path):
+                for file in files:
+                    if file.lower().endswith(EXTENSIONS):
+                        file_path = os.path.join(root, file)
+                        file_info = get_file_info(file_path)
+                        if file_info:
+                            results.append([
+                                host, user, pathlib.Path(file).suffix, 
+                                file_info["size_bytes"], file_info["size_formatted"],
+                                file_info["created"], file_info["last_accessed"], file_path
+                            ])
+                            print(f"[Найден файл] {file_path} ({file_info['size_formatted']})")
+
+    return results
+
+def main():
+    with open(HOSTS_FILE, "r") as f:
+        hosts = [line.strip() for line in f.readlines() if line.strip()]
+
+    all_results = []
+    for host in hosts:
+        print(f"\n🔍 [Сканирование] ПК: {host}")
+        results = scan_host(host)
+        all_results.extend(results)
+
+    # Сохранение найденных файлов в CSV
+    with open(OUTPUT_CSV, "w", newline="", encoding="utf-8") as csv_file:
+        writer = csv.writer(csv_file)
+        writer.writerow(["PC Name", "User", "File Type", "Size (bytes)", "Size (Formatted)", "Created Date", "Last Accessed", "File Path"])
+        writer.writerows(all_results)
+
+    print(f"\n✅ Поиск завершен! Результаты сохранены в {OUTPUT_CSV}")
+
+    # Сохранение списка недоступных ПК
+    if failed_hosts:
+        with open(FAILED_CSV, "w", newline="", encoding="utf-8") as csv_file:
+            writer = csv.writer(csv_file)
+            writer.writerow(["PC Name"])
+            writer.writerows([[host] for host in failed_hosts])
+
+        print("\n❌ [Ошибка] Не удалось подключиться к следующим ПК:")
+        for host in failed_hosts:
+            print(f"   - {host}")
+
+        print(f"📄 Список недоступных ПК сохранен в {FAILED_CSV}")
+
+if __name__ == "__main__":
+    main()