116 lines
3.0 KiB
Python
116 lines
3.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""Flask Web 应用：展示 IPTV 直播源结果"""
|
|
|
|
import os
|
|
import json
|
|
from flask import Flask, render_template, jsonify, request
|
|
from collector import collect_all
|
|
from checker import check_all
|
|
from deduplicator import deduplicate, group_by_category
|
|
from config import OUTPUT_DIR, OUTPUT_FILE
|
|
|
|
app = Flask(__name__)
|
|
|
|
# 缓存结果
|
|
_cache = {
|
|
"channels": [],
|
|
"groups": {},
|
|
"checked": False,
|
|
}
|
|
|
|
|
|
def load_from_file(filepath):
|
|
"""从 M3U 文件加载频道"""
|
|
channels = []
|
|
if not os.path.exists(filepath):
|
|
return channels
|
|
with open(filepath, "r", encoding="utf-8") as f:
|
|
current = None
|
|
for line in f:
|
|
line = line.strip()
|
|
if line.startswith("#EXTINF"):
|
|
name = line.split(",", 1)[1] if "," in line else ""
|
|
import re
|
|
tvg = re.search(r'tvg-name="([^"]*)"', line)
|
|
grp = re.search(r'group-title="([^"]*)"', line)
|
|
current = {
|
|
"name": name,
|
|
"tvg_name": tvg.group(1) if tvg else name,
|
|
"group": grp.group(1) if grp else "其他",
|
|
"url": "",
|
|
}
|
|
elif line and not line.startswith("#") and current:
|
|
current["url"] = line
|
|
channels.append(current)
|
|
current = None
|
|
return channels
|
|
|
|
|
|
@app.route("/")
|
|
def index():
|
|
"""首页"""
|
|
return render_template("index.html")
|
|
|
|
|
|
@app.route("/api/channels")
|
|
def api_channels():
|
|
"""获取频道列表"""
|
|
return jsonify({
|
|
"total": len(_cache["channels"]),
|
|
"channels": _cache["channels"],
|
|
})
|
|
|
|
|
|
@app.route("/api/groups")
|
|
def api_groups():
|
|
"""获取分组统计"""
|
|
groups = {}
|
|
for g, items in _cache["groups"].items():
|
|
groups[g] = len(items)
|
|
return jsonify(groups)
|
|
|
|
|
|
@app.route("/api/collect", methods=["POST"])
|
|
def api_collect():
|
|
"""触发采集+检测"""
|
|
do_check = request.json.get("check", True) if request.is_json else True
|
|
|
|
channels = collect_all()
|
|
if do_check:
|
|
channels = check_all(channels)
|
|
channels = deduplicate(channels)
|
|
|
|
_cache["channels"] = channels
|
|
_cache["groups"] = group_by_category(channels)
|
|
_cache["checked"] = do_check
|
|
|
|
return jsonify({
|
|
"success": True,
|
|
"total": len(channels),
|
|
})
|
|
|
|
|
|
@app.route("/api/load")
|
|
def api_load():
|
|
"""从输出文件加载"""
|
|
filepath = os.path.join(OUTPUT_DIR, OUTPUT_FILE)
|
|
channels = load_from_file(filepath)
|
|
_cache["channels"] = channels
|
|
_cache["groups"] = group_by_category(channels)
|
|
return jsonify({
|
|
"success": True,
|
|
"total": len(channels),
|
|
})
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# 启动时尝试加载已有结果
|
|
filepath = os.path.join(OUTPUT_DIR, OUTPUT_FILE)
|
|
if os.path.exists(filepath):
|
|
channels = load_from_file(filepath)
|
|
_cache["channels"] = channels
|
|
_cache["groups"] = group_by_category(channels)
|
|
print(f"已加载 {len(channels)} 个频道")
|
|
|
|
app.run(host="0.0.0.0", port=5000, debug=False)
|