#!/usr/bin/env python3 import os import json import time import sys from pathlib import Path from html import escape def format_size(num_bytes): if num_bytes is None: return "-" units = ["B", "KB", "MB", "GB", "TB"] size = float(num_bytes) for unit in units: if size < 1024: return f"{size:.1f} {unit}" size /= 1024 return f"{size:.1f} PB" def format_mtime(ts): try: return time.strftime("%Y-%m-%d %H:%M", time.localtime(ts)) except: return "-" def load_icons(icon_file="icon.json"): try: with open(icon_file, "r", encoding="utf-8") as f: return json.load(f) except FileNotFoundError: print("⚠ icon.json not found, using default icons") return {} def count_visible_items(path): try: return len([f for f in os.listdir(path) if not f.startswith(".")]) except Exception: return 0 def get_folder_size(path): total = 0 file_count = 0 for root, dirs, files in os.walk(path): dirs[:] = [d for d in dirs if not d.startswith(".")] for f in files: if not f.startswith("."): try: total += os.path.getsize(os.path.join(root, f)) file_count += 1 except Exception: pass return total, file_count def build_structure(path, base_dir, label=None): rel = os.path.relpath(path, base_dir).replace("\\", "/") struct = { "label": label or os.path.basename(path) or "My Drive", "files": [], "subdirs": {}, "size": 0, "file_count": 0, "path": "" if rel in (".", "") else rel, "modified": format_mtime(os.path.getmtime(path)) } try: entries = os.listdir(path) except Exception as e: print(f"⚠️ Error reading directory {path}: {e}") return struct for entry in entries: if entry.startswith("."): continue full = os.path.join(path, entry) relp = os.path.relpath(full, base_dir).replace("\\", "/") if os.path.isfile(full): size = os.path.getsize(full) ext = Path(entry).suffix.lower().lstrip(".") or Path(entry).name.lower() struct["files"].append({ "name": entry, "path": relp, "size": size, "modified": format_mtime(os.path.getmtime(full)), "ext": ext }) struct["size"] += size struct["file_count"] += 1 elif os.path.isdir(full): sub = build_structure(full, base_dir, entry) struct["subdirs"][entry] = sub struct["size"] += sub["size"] struct["file_count"] += sub["file_count"] return struct def write_structure_json(base_dir): struct = build_structure(base_dir, base_dir, "My Drive") root = { "label": "My Drive", "size": struct["size"], "file_count": struct["file_count"], "modified": struct["modified"], "structure": struct } with open(os.path.join(base_dir, "structure.json"), "w", encoding="utf-8") as f: json.dump({"root": root}, f, indent=2) def generate_breadcrumbs(rel, root_rel): if rel in (".", ""): return 'home' parts = rel.split("/") crumbs = [f'home'] path = "" for i, part in enumerate(parts[:-1]): path += part + "/" link_path = f"{root_rel}/{path}index.html" crumbs.append(f'{escape(part)}') crumbs.append(f'{escape(parts[-1])}') return '⟩'.join(crumbs) def get_icon(name, is_dir, icons): if is_dir: return 'folder' ext = Path(name).suffix.lower().lstrip(".") or Path(name).name.lower() icon_name = icons.get(ext, "insert_drive_file") return f'{icon_name}' def build_tree_html(struct, current_rel="", root_rel=""): html = '" return html def generate_folder_tree(folder_path, base_dir): rel = os.path.relpath(folder_path, base_dir).replace("\\", "/") with open(os.path.join(base_dir, "structure.json"), "r", encoding="utf-8") as f: data = json.load(f) return build_tree_html(data["root"]["structure"], current_rel=rel) def generate_index(folder_path, base_dir): rel = os.path.relpath(folder_path, base_dir).replace("\\", "/") title = "My Drive" if rel in (".", "") else f"Index of {rel}" dirs, files = [], [] try: for e in os.listdir(folder_path): if e.startswith("."): continue full = os.path.join(folder_path, e) (dirs if os.path.isdir(full) else files).append(e) except Exception as e: print(f"⚠️ Error reading directory {folder_path}: {e}") dirs.sort(key=str.lower) files.sort(key=str.lower) root_rel = os.path.relpath(base_dir, folder_path).replace("\\", "/") breadcrumbs = generate_breadcrumbs(rel, root_rel) icons = load_icons() # Load structure.json to get root file_count try: with open(os.path.join(base_dir, "structure.json"), "r", encoding="utf-8") as f: data = json.load(f) file_count = data["root"]["file_count"] except Exception as e: print(f"⚠️ Error reading structure.json: {e}") file_count = 0 # Calculate percentage for quota card max_files = 20000 percentage = (file_count / max_files) * 100 if max_files > 0 else 0 # Current directory info total_size, current_dir_file_count = get_folder_size(folder_path) current_dir_name = os.path.basename(folder_path) if folder_path != base_dir else "My Drive" # Storage card HTML storage_html = f'''

folder_open{escape(current_dir_name)}

{format_size(total_size)} ({current_dir_file_count} file{'s' if current_dir_file_count != 1 else ''})

''' # Quota card HTML quota_html = f'''
File Limit {int(percentage)}% used
{file_count} files of 20,000
''' folder_tree = generate_folder_tree(folder_path, base_dir) html = f""" {escape(title)} — Nimbus Drive
cloud Nimbus Drive
search Ctrl + /
""" with open(os.path.join(folder_path, "index.html"), "w", encoding="utf-8") as f: f.write(html) def build_site(base_dir="."): # Count total directories to process total_dirs = sum(1 for root, dirs, _ in os.walk(base_dir) if not root.startswith(os.path.join(base_dir, "."))) processed_dirs = 0 bar_width = 20 # Width of the progress bar in characters # Write icon.json icons = load_icons() with open(os.path.join(base_dir, "icon.json"), "w", encoding="utf-8") as f: json.dump(icons, f, indent=2) # Write structure.json print("❯ Generating structure.json...") write_structure_json(base_dir) print("✔ structure.json generated") # Generate index.html for each directory with progress bar print("❯ Generating index.html files...") for root, dirs, files in os.walk(base_dir): dirs[:] = [d for d in dirs if not d.startswith(".")] generate_index(root, base_dir) processed_dirs += 1 # Calculate progress progress = processed_dirs / total_dirs if total_dirs > 0 else 1 filled = int(bar_width * progress) bar = "=" * filled + "-" * (bar_width - filled) percentage = int(progress * 100) # Update progress bar in place sys.stdout.write(f"\r❯ [{bar}] {percentage}% ({processed_dirs}/{total_dirs})") sys.stdout.flush() # Clear the progress bar line and print completion message sys.stdout.write("\r" + " " * (bar_width + 20)) # Clear the line sys.stdout.flush() print(f"\r✔ Generated {processed_dirs} index.html files and structure.json") print(f"\r✔✔ Site Generated Successfully : To Explore start a Server") print(f"\n You Can Start The Server @ python3 -m http.server 2104\n") if __name__ == "__main__": build_site()