|
from flask import Flask, request, jsonify, render_template |
|
from flask_socketio import SocketIO, emit |
|
import json |
|
import uuid |
|
import os |
|
from datetime import datetime |
|
|
|
app = Flask(__name__) |
|
app.config['SECRET_KEY'] = 'your-secret-key' |
|
socketio = SocketIO(app) |
|
|
|
|
|
jobs = [] |
|
connected_clients = {} |
|
|
|
def save_jobs(): |
|
"""Save jobs to JSON file""" |
|
try: |
|
with open('jobs.json', 'w') as f: |
|
json.dump(jobs, f) |
|
except Exception as e: |
|
print(f"Error saving jobs: {e}") |
|
|
|
def load_jobs(): |
|
"""Load jobs from JSON file""" |
|
try: |
|
if os.path.exists('jobs.json') and os.path.getsize('jobs.json') > 0: |
|
with open('jobs.json', 'r') as f: |
|
return json.load(f) |
|
except Exception as e: |
|
print(f"Error loading jobs: {e}") |
|
return [] |
|
|
|
@app.route('/') |
|
def index(): |
|
return render_template('index.html') |
|
|
|
@app.route('/submit_job', methods=['POST']) |
|
def submit_job(): |
|
youtube_url = request.form.get("youtube_url") |
|
exit_time = int(request.form.get("exit_time", 10)) |
|
mode = request.form.get("mode", "once") |
|
job_id = str(uuid.uuid4()) |
|
|
|
job = { |
|
"job_id": job_id, |
|
"youtube_url": youtube_url, |
|
"exit_time": exit_time, |
|
"mode": mode, |
|
"status": "pending" |
|
} |
|
jobs.append(job) |
|
save_jobs() |
|
|
|
return jsonify({"message": "Job submitted successfully", "job_id": job_id}) |
|
|
|
@socketio.on('client_info') |
|
def handle_client_info(data): |
|
client_id = request.sid |
|
connected_clients[client_id] = { |
|
'ip': data.get('ip', 'Unknown'), |
|
'country': data.get('country', 'Unknown'), |
|
'city': data.get('city', 'Unknown'), |
|
'connected_at': datetime.now().isoformat() |
|
} |
|
|
|
emit('client_connected', { |
|
'client_id': client_id, |
|
'ip': data.get('ip', 'Unknown'), |
|
'country': data.get('country', 'Unknown'), |
|
'city': data.get('city', 'Unknown') |
|
}, broadcast=True) |
|
|
|
@socketio.on('disconnect') |
|
def handle_disconnect(): |
|
client_id = request.sid |
|
if client_id in connected_clients: |
|
del connected_clients[client_id] |
|
emit('client_disconnected', client_id, broadcast=True) |
|
|
|
@app.route('/get_job', methods=['GET']) |
|
def get_job(): |
|
for job in jobs: |
|
if job["status"] == "pending": |
|
return jsonify(job) |
|
return jsonify({"message": "No pending jobs"}) |
|
|
|
@app.route('/update_job_status', methods=['POST']) |
|
def update_job_status(): |
|
data = request.get_json() |
|
job_id = data.get("job_id") |
|
status = data.get("status", "done") |
|
|
|
for job in jobs: |
|
if job["job_id"] == job_id: |
|
job["status"] = status |
|
break |
|
|
|
save_jobs() |
|
return jsonify({"message": f"Job {job_id} marked as {status}"}) |
|
|
|
if __name__ == "__main__": |
|
jobs = load_jobs() |
|
|
|
if not os.path.exists('templates'): |
|
os.makedirs('templates') |
|
|
|
template_path = os.path.join('templates', 'index.html') |
|
if not os.path.exists(template_path): |
|
with open(template_path, 'w') as f: |
|
f.write('''<!DOCTYPE html> |
|
<html lang="en"> |
|
<head> |
|
<meta charset="UTF-8"> |
|
<meta name="viewport" content="width=device-width, initial-scale=1.0"> |
|
<title>YouTube Job Manager</title> |
|
<link href="https://cdn.jsdelivr.net/npm/bootstrap@5.1.3/dist/css/bootstrap.min.css" rel="stylesheet"> |
|
<style> |
|
body { padding: 20px; } |
|
.connection-status { margin-bottom: 20px; } |
|
</style> |
|
</head> |
|
<body> |
|
<div class="container"> |
|
<div class="row"> |
|
<div class="col-md-8 offset-md-2"> |
|
<h1 class="mb-4">YouTube Job Manager</h1> |
|
|
|
<div class="card mb-4"> |
|
<div class="card-header"> |
|
<h5 class="mb-0">Connected Clients</h5> |
|
</div> |
|
<div class="card-body"> |
|
<table class="table" id="clientsTable"> |
|
<thead> |
|
<tr> |
|
<th>IP Address</th> |
|
<th>Country</th> |
|
<th>City</th> |
|
<th>Status</th> |
|
</tr> |
|
</thead> |
|
<tbody id="clientsTableBody"> |
|
</tbody> |
|
</table> |
|
</div> |
|
</div> |
|
|
|
<div class="card"> |
|
<div class="card-header"> |
|
<h5 class="mb-0">Submit New Job</h5> |
|
</div> |
|
<div class="card-body"> |
|
<form action="/submit_job" method="post" class="needs-validation" novalidate> |
|
<div class="mb-3"> |
|
<label for="youtube_url" class="form-label">YouTube URL:</label> |
|
<input type="text" class="form-control" id="youtube_url" name="youtube_url" required> |
|
</div> |
|
|
|
<div class="mb-3"> |
|
<label for="exit_time" class="form-label">Exit Time (seconds):</label> |
|
<input type="number" class="form-control" id="exit_time" name="exit_time" value="10" required> |
|
</div> |
|
|
|
<div class="mb-3"> |
|
<label for="mode" class="form-label">Mode:</label> |
|
<select class="form-select" id="mode" name="mode"> |
|
<option value="once">Once</option> |
|
<option value="again">Again</option> |
|
<option value="continuous">Continuous</option> |
|
</select> |
|
</div> |
|
|
|
<button type="submit" class="btn btn-primary">Submit Job</button> |
|
</form> |
|
</div> |
|
</div> |
|
</div> |
|
</div> |
|
</div> |
|
|
|
<script src="https://cdn.jsdelivr.net/npm/bootstrap@5.1.3/dist/js/bootstrap.bundle.min.js"></script> |
|
<script src="https://cdnjs.cloudflare.com/ajax/libs/socket.io/4.0.1/socket.io.js"></script> |
|
<script> |
|
const socket = io(); |
|
|
|
socket.on('connect', () => { |
|
console.log('Connected to server'); |
|
}); |
|
|
|
socket.on('client_connected', (data) => { |
|
const tableBody = document.getElementById('clientsTableBody'); |
|
const row = document.createElement('tr'); |
|
row.setAttribute('data-client-id', data.client_id); |
|
row.innerHTML = ` |
|
<td>${data.ip}</td> |
|
<td>${data.country}</td> |
|
<td>${data.city}</td> |
|
<td><span class="badge bg-success">Connected</span></td> |
|
`; |
|
tableBody.appendChild(row); |
|
}); |
|
|
|
socket.on('client_disconnected', (clientId) => { |
|
const row = document.querySelector(`tr[data-client-id="${clientId}"]`); |
|
if (row) { |
|
row.querySelector('.badge').className = 'badge bg-danger'; |
|
row.querySelector('.badge').textContent = 'Disconnected'; |
|
} |
|
}); |
|
</script> |
|
</body> |
|
</html> |
|
''') |
|
|
|
socketio.run(app, host="0.0.0.0", port=7860 debug=False) |
