134 lines
4.5 KiB
Python
134 lines
4.5 KiB
Python
from flask import Blueprint, flash, redirect, render_template, request, url_for
|
|
from flask_login import login_required
|
|
|
|
from app.extensions import db
|
|
from app.guides import GUIDES
|
|
from app.models import Mirror, Profile, ProfileItem, TestResult
|
|
from app.services.tester import test_all_enabled, test_mirror
|
|
|
|
mirrors_bp = Blueprint("mirrors", __name__, url_prefix="/mirrors")
|
|
|
|
|
|
@mirrors_bp.route("/")
|
|
@login_required
|
|
def list_mirrors():
|
|
category = request.args.get("category")
|
|
query = Mirror.query.order_by(Mirror.category, Mirror.priority)
|
|
if category:
|
|
query = query.filter_by(category=category)
|
|
mirrors = query.all()
|
|
categories = Mirror.CATEGORIES
|
|
return render_template(
|
|
"mirrors/list.html",
|
|
mirrors=mirrors,
|
|
categories=categories,
|
|
category_labels=Mirror.CATEGORY_LABELS,
|
|
current_category=category,
|
|
guides=GUIDES,
|
|
)
|
|
|
|
|
|
@mirrors_bp.route("/create", methods=["GET", "POST"])
|
|
@login_required
|
|
def create():
|
|
if request.method == "POST":
|
|
mirror = Mirror(
|
|
category=request.form["category"],
|
|
name=request.form["name"].strip(),
|
|
url=request.form.get("url", "").strip() or None,
|
|
ips=request.form.get("ips", "").strip() or None,
|
|
priority=int(request.form.get("priority", 100)),
|
|
enabled=request.form.get("enabled") == "on",
|
|
notes=request.form.get("notes", "").strip() or None,
|
|
)
|
|
meta_prefix = request.form.get("instead_prefix", "").strip()
|
|
if meta_prefix:
|
|
mirror.set_meta({"instead_prefix": meta_prefix})
|
|
db.session.add(mirror)
|
|
db.session.commit()
|
|
flash("میرور با موفقیت ایجاد شد.", "success")
|
|
return redirect(url_for("mirrors.list_mirrors"))
|
|
|
|
return render_template(
|
|
"mirrors/form.html",
|
|
mirror=None,
|
|
categories=Mirror.CATEGORIES,
|
|
category_labels=Mirror.CATEGORY_LABELS,
|
|
)
|
|
|
|
|
|
@mirrors_bp.route("/<int:mirror_id>/edit", methods=["GET", "POST"])
|
|
@login_required
|
|
def edit(mirror_id):
|
|
mirror = Mirror.query.get_or_404(mirror_id)
|
|
if request.method == "POST":
|
|
mirror.category = request.form["category"]
|
|
mirror.name = request.form["name"].strip()
|
|
mirror.url = request.form.get("url", "").strip() or None
|
|
mirror.ips = request.form.get("ips", "").strip() or None
|
|
mirror.priority = int(request.form.get("priority", 100))
|
|
mirror.enabled = request.form.get("enabled") == "on"
|
|
mirror.notes = request.form.get("notes", "").strip() or None
|
|
meta_prefix = request.form.get("instead_prefix", "").strip()
|
|
if meta_prefix:
|
|
mirror.set_meta({"instead_prefix": meta_prefix})
|
|
db.session.commit()
|
|
flash("میرور بهروزرسانی شد.", "success")
|
|
return redirect(url_for("mirrors.list_mirrors"))
|
|
|
|
return render_template(
|
|
"mirrors/form.html",
|
|
mirror=mirror,
|
|
categories=Mirror.CATEGORIES,
|
|
category_labels=Mirror.CATEGORY_LABELS,
|
|
)
|
|
|
|
|
|
@mirrors_bp.route("/<int:mirror_id>/delete", methods=["POST"])
|
|
@login_required
|
|
def delete(mirror_id):
|
|
mirror = Mirror.query.get_or_404(mirror_id)
|
|
db.session.delete(mirror)
|
|
db.session.commit()
|
|
flash("میرور حذف شد.", "warning")
|
|
return redirect(url_for("mirrors.list_mirrors"))
|
|
|
|
|
|
@mirrors_bp.route("/<int:mirror_id>/test", methods=["POST"])
|
|
@login_required
|
|
def test_one(mirror_id):
|
|
mirror = Mirror.query.get_or_404(mirror_id)
|
|
results = test_mirror(mirror)
|
|
ok = all(r.success for r in results)
|
|
flash(
|
|
f"تست {mirror.name}: {'موفق' if ok else 'ناموفق'}",
|
|
"success" if ok else "danger",
|
|
)
|
|
return redirect(url_for("mirrors.list_mirrors", category=mirror.category))
|
|
|
|
|
|
@mirrors_bp.route("/test-all", methods=["POST"])
|
|
@login_required
|
|
def test_all():
|
|
category = request.form.get("category") or None
|
|
test_all_enabled(category)
|
|
flash("تست همه میرورها انجام شد.", "info")
|
|
return redirect(url_for("mirrors.list_mirrors", category=category))
|
|
|
|
|
|
@mirrors_bp.route("/results")
|
|
@login_required
|
|
def results():
|
|
category = request.args.get("category")
|
|
query = TestResult.query.join(Mirror).order_by(TestResult.tested_at.desc())
|
|
if category:
|
|
query = query.filter(Mirror.category == category)
|
|
results_list = query.limit(100).all()
|
|
return render_template(
|
|
"mirrors/results.html",
|
|
results=results_list,
|
|
category_labels=Mirror.CATEGORY_LABELS,
|
|
current_category=category,
|
|
categories=Mirror.CATEGORIES,
|
|
)
|