|
|
from flask import Flask, render_template, request |
|
|
import requests |
|
|
import json |
|
|
|
|
|
app = Flask(__name__) |
|
|
|
|
|
def get_folder_json(folder_id, headers): |
|
|
url = f"https://storage.yoghourt.space/content/list?parent_id={folder_id}&order=2&page=1&size=100" |
|
|
response = requests.get(url, headers=headers) |
|
|
return response.text |
|
|
|
|
|
def process_content(json_data): |
|
|
try: |
|
|
data = json.loads(json_data) |
|
|
except json.JSONDecodeError: |
|
|
return [], [] |
|
|
|
|
|
if "data" not in data or data["data"] is None: |
|
|
return [], [] |
|
|
|
|
|
folders = [] |
|
|
files = [] |
|
|
|
|
|
content_list = data["data"].get("content_list") |
|
|
|
|
|
if not content_list: |
|
|
return folders, files |
|
|
|
|
|
for item in content_list: |
|
|
if "content_type" not in item or "content" not in item: |
|
|
continue |
|
|
content_type = item["content_type"] |
|
|
content = item["content"] |
|
|
|
|
|
if content_type == "folder": |
|
|
if "id" in content and "name" in content: |
|
|
folders.append({"id": content["id"], "name": content["name"]}) |
|
|
elif content_type == "file": |
|
|
if "cid" in content and "name" in content: |
|
|
files.append({"cid": content["cid"], "name": content["name"]}) |
|
|
|
|
|
return folders, files |
|
|
|
|
|
def generate_tree(folder_id, headers, gateway, indent=0): |
|
|
result = [] |
|
|
folder_json = get_folder_json(folder_id, headers) |
|
|
folders, files = process_content(folder_json) |
|
|
|
|
|
for folder in folders: |
|
|
result.append(" " * indent + f"{folder['name']}:") |
|
|
result.extend(generate_tree(folder["id"], headers, gateway, indent + 1)) |
|
|
|
|
|
for file in files: |
|
|
result.append(" " * indent + f"{file['name']}:https://{gateway}/ipfs/{file['cid']}?filename={file['name']}") |
|
|
|
|
|
return result |
|
|
|
|
|
@app.route('/', methods=['GET', 'POST']) |
|
|
def index(): |
|
|
if request.method == 'POST': |
|
|
auth = request.form['auth'] |
|
|
root_id = request.form['root_id'] |
|
|
gateway = request.form['gateway'] |
|
|
headers = {"Authorization": auth} |
|
|
|
|
|
try: |
|
|
output_lines = generate_tree(root_id, headers, gateway) |
|
|
output = '\n'.join(output_lines) |
|
|
return render_template('result.html', output=output) |
|
|
except Exception as e: |
|
|
return f"Error: {str(e)}" |
|
|
|
|
|
return render_template('index.html') |
|
|
|
|
|
if __name__ == '__main__': |
|
|
app.run(host='0.0.0.0', port=7860) |
|
|
|
