Create generate.py
This commit is contained in:
parent
e34cf463ae
commit
572a9ed85c
1 changed files with 59 additions and 0 deletions
59
generate.py
Normal file
59
generate.py
Normal file
|
@ -0,0 +1,59 @@
|
|||
import requests
|
||||
from flask import Flask, Response
|
||||
|
||||
app = Flask(__name__)
|
||||
|
||||
# List of M3U file URLs
|
||||
urls = [
|
||||
"https://iptv-org.github.io/iptv/index.country.m3u",
|
||||
"https://forge.fsky.io/frost/repo/raw/branch/main/tv.m3u",
|
||||
"https://raw.githubusercontent.com/Free-TV/IPTV/refs/heads/master/playlist.m3u8"
|
||||
]
|
||||
|
||||
combined_m3u = None # Variable to store the combined M3U content
|
||||
|
||||
def fetch_m3u(url):
|
||||
"""Fetch M3U content from a given URL."""
|
||||
response = requests.get(url)
|
||||
response.raise_for_status() # Raise an error for bad responses
|
||||
return response.text
|
||||
|
||||
def combine_m3u(urls):
|
||||
"""Combine M3U content from multiple URLs and remove duplicates."""
|
||||
unique_streams = {}
|
||||
|
||||
for url in urls:
|
||||
m3u_content = fetch_m3u(url)
|
||||
lines = m3u_content.splitlines()
|
||||
current_entry = None
|
||||
|
||||
for line in lines:
|
||||
if line.startswith('#EXTINF:'):
|
||||
current_entry = line
|
||||
elif line and current_entry:
|
||||
# Use the stream link as the key to ensure uniqueness
|
||||
stream_link = line.strip()
|
||||
if stream_link not in unique_streams:
|
||||
unique_streams[stream_link] = current_entry + '\n' + stream_link
|
||||
current_entry = None # Reset current entry after adding
|
||||
|
||||
# Return the combined M3U content
|
||||
return '#EXTM3U\n' + '\n'.join(unique_streams.values())
|
||||
|
||||
@app.route('/all.m3u')
|
||||
def serve_m3u():
|
||||
"""Serve the combined M3U file."""
|
||||
global combined_m3u
|
||||
if combined_m3u is None: # Generate if not already done
|
||||
combined_m3u = combine_m3u(urls)
|
||||
return Response(combined_m3u, mimetype='application/vnd.apple.mpegurl')
|
||||
|
||||
@app.route('/generate')
|
||||
def generate_m3u():
|
||||
"""Regenerate the M3U file from the sources."""
|
||||
global combined_m3u
|
||||
combined_m3u = combine_m3u(urls) # Regenerate the M3U content
|
||||
return Response("M3U playlist regenerated!", mimetype='text/plain')
|
||||
|
||||
if __name__ == '__main__':
|
||||
app.run(host='0.0.0.0', port=5000)
|
Loading…
Add table
Reference in a new issue