import os import sys from flask import Flask, render_template_string, send_from_directory, abort, jsonify, request, redirect, url_for # -------------------------------------------------------------------------------------- # 1. SETUP & CONFIGURATION (კონფიგურაცია) # -------------------------------------------------------------------------------------- # BASE_DIR: დირექტორია, რომლის დათვალიერებაც გვსურს. # ის დაყენებულია იმ ფოლდერზე, სადაც app.py გაშვებულია. BASE_DIR = os.path.abspath(os.path.dirname(__file__)) # Flask აპლიკაციის ინიციალიზაცია app = Flask(__name__) # -------------------------------------------------------------------------------------- # 2. EMBEDDED HTML/JINJA2 TEMPLATE (UI/UX) # -------------------------------------------------------------------------------------- # ვიყენებთ ერთ ჩაშენებულ Jinja2 შაბლონს. # Tailwind CSS CDN გამოიყენება ლამაზი, მინიმალისტური და რესპონსული დიზაინისთვის. HTML_TEMPLATE = """ ფაილების მინიმალისტური ბრაუზერი
მიმდინარე ბილიკი: {{ base_dir_name }}/{{ current_path or '' }}
{% if error %} {% endif %}
{% macro render_list(items, is_search=False) %} {% if items %} {% else %}

ფაილები არ მოიძებნა

დირექტორია {{ current_path or base_dir_name }} ცარიელია ან მიუწვდომელია.

{% endif %} {% endmacro %}
{{ render_list(files_and_folders) }}
""" # -------------------------------------------------------------------------------------- # 3. HELPER FUNCTION (დამხმარე ფუნქცია) # -------------------------------------------------------------------------------------- def get_current_directory_content(path=''): """ ჩამოთვლის მიმდინარე დირექტორიის შიგთავსს BASE_DIR-ის მიმართ. """ # უსაფრთხო ბილიკის შექმნა full_path = os.path.join(BASE_DIR, path) # უსაფრთხოების შემოწმება: დარწმუნდით, რომ ბილიკი არ გადის BASE_DIR-ის ფარგლებს if not os.path.abspath(full_path).startswith(BASE_DIR): # 403 Forbidden return [], "უსაფრთხოების შეცდომა: ბილიკი არ არის დაშვებული.", "" # FIX: თუ ბილიკი ფაილზე მიუთითებს, არ დააბრუნოთ შეცდომა, არამედ მიეცით ცარიელი სია, # რადგან index() ფუნქცია ახლა აკეთებს გადამისამართებას. if not os.path.exists(full_path) or not os.path.isdir(full_path): # ეს შეცდომა გამოჩნდება, თუ ფოლდერი არ არსებობს return [], "დირექტორია ვერ მოიძებნა.", "" items = [] try: # მიმდინარე დირექტორიის შიგთავსის ჩამოთვლა for entry in os.listdir(full_path): # გამოტოვება: ფარული ფაილები და სისტემური ფოლდერები if entry.startswith('.') or entry in ['__pycache__', 'venv']: continue # ბილიკების ფორმატირება URL-ებისთვის (slash-ის გამოყენებით) entry_path = os.path.join(path, entry).replace('\\', '/') full_entry_path = os.path.join(BASE_DIR, entry_path) item = { 'name': entry, 'path': entry_path, 'is_dir': os.path.isdir(full_entry_path) } if not item['is_dir']: # ფაილის ზომის დამატება item['size'] = os.path.getsize(full_entry_path) items.append(item) # დახარისხება: ჯერ ფოლდერები, შემდეგ ფაილები items.sort(key=lambda x: (not x['is_dir'], x['name'].lower())) except Exception as e: # წვდომის შეცდომის დამუშავება return [], f"დირექტორიაზე წვდომის შეცდომა: {e}", "" # მშობელი ბილიკის გამოთვლა (სწორი ნავიგაციისთვის) parent_path = '/'.join(path.split('/')[:-1]) if path else '' return items, None, parent_path def recursive_search_directory(search_term=""): """ რეკურსიულად ეძებს მთლიან BASE_DIR-ში ფაილებსა და ფოლდერებს, რომლებიც ემთხვევა საძიებო ტერმინს. """ search_term = search_term.lower() results = [] # os.walk არის ყველაზე ეფექტური მეთოდი რეკურსიული ძიებისთვის for root, dirs, files in os.walk(BASE_DIR): # გამოტოვება: ფარული ფაილები და სისტემური ფოლდერები dirs[:] = [d for d in dirs if not d.startswith('.') and d not in ['__pycache__', 'venv']] # გამოთვლა: ბილიკი BASE_DIR-თან მიმართებაში relative_root = os.path.relpath(root, BASE_DIR).replace('\\', '/') if relative_root == '.': relative_root = "" # Check folders (dirs) and files for name in dirs + files: if search_term in name.lower(): is_dir = name in dirs path = os.path.join(relative_root, name).replace('\\', '/') full_path = os.path.join(root, name) item = { 'name': name, 'path': path, 'is_dir': is_dir, 'size': 0 } if not is_dir and os.path.exists(full_path): # ფაილის ზომის დამატება item['size'] = os.path.getsize(full_path) results.append(item) # დახარისხება: ჯერ ფოლდერები, შემდეგ ფაილები results.sort(key=lambda x: (not x['is_dir'], x['name'].lower())) return results # -------------------------------------------------------------------------------------- # 4. FLASK ROUTES (Flask როუტები) # -------------------------------------------------------------------------------------- @app.route('/', defaults={'req_path': ''}) @app.route('/') def index(req_path): """ მთავარი როუტი დირექტორიების დასათვალიერებლად. """ # URL ბილიკის ნორმალიზება current_path = req_path.replace('\\', '/').strip('/') full_path_check = os.path.join(BASE_DIR, current_path) # FIX: თუ მომხმარებელი შედის ფაილის ბილიკზე, გადამისამართება გადმოწერის როუტზე if os.path.isfile(full_path_check) and os.path.abspath(full_path_check).startswith(BASE_DIR): return redirect(url_for('download_file', filename=current_path)) # მიმდინარე დირექტორიის შიგთავსის მიღება files_and_folders, error, _ = get_current_directory_content(current_path) # საბაზისო დირექტორიის სახელის ჩვენება base_dir_name = os.path.basename(BASE_DIR) or "Root" return render_template_string( HTML_TEMPLATE, files_and_folders=files_and_folders, current_path=current_path, error=error, base_dir_name=base_dir_name ) # --- API Route for Dynamic Tree View --- @app.route('/api/structure/', defaults={'req_path': ''}) @app.route('/api/structure/') def api_structure(req_path): """ აბრუნებს დირექტორიის შიგთავსს JSON ფორმატში Tree View-სთვის. """ current_path = req_path.replace('\\', '/').strip('/') files_and_folders, error, _ = get_current_directory_content(current_path) if error: return jsonify({'error': error}), 400 tree_nodes = [ {'name': item['name'], 'path': item['path'], 'is_dir': item['is_dir']} for item in files_and_folders ] return jsonify(tree_nodes) # --- Global Search API Route (გლობალური ძიების API) --- @app.route('/api/search') def api_search(): """ ასრულებს რეკურსიულ ძიებას მთელ BASE_DIR-ში და აბრუნებს შედეგებს JSON-ად. """ query = request.args.get('q', '') if not query: return jsonify([]) # უსაფრთხოების შემოწმება: ძებნის ტერმინის ნორმალიზება query = query.strip() results = recursive_search_directory(query) return jsonify(results) # ------------------------------------------- @app.route('/download/') def download_file(filename): """ როუტი ფაილის უსაფრთხოდ გადმოსაწერად. """ # ფაილის ბილიკის ნორმალიზება safe_filename = filename.replace('\\', '/').strip('/') full_path = os.path.join(BASE_DIR, safe_filename) # უსაფრთხოების შემოწმება (Directory Traversal Prevention) if not os.path.abspath(full_path).startswith(BASE_DIR): abort(403) # აკრძალული # დარწმუნდით, რომ ეს არის ფაილი if not os.path.isfile(full_path): # თუ ფაილი არ არის, შესაძლოა ეს იყო ფოლდერი ან არარსებული ბილიკი abort(404) # ვერ მოიძებნა try: # send_from_directory უსაფრთხოდ ემსახურება ფაილებს BASE_DIR-დან return send_from_directory( directory=BASE_DIR, path=safe_filename, as_attachment=True ) except FileNotFoundError: abort(404) except Exception: abort(500) # შიდა სერვერის შეცდომა # -------------------------------------------------------------------------------------- # 5. RUN APP (აპლიკაციის გაშვება) # -------------------------------------------------------------------------------------- if __name__ == '__main__': # Flask-ის გაშვება # დარწმუნდით, რომ გაქვთ დაყენებული Flask: pip install Flask app.run(debug=True)