direct

Sleeping

App Files Files Community

Kgshop commited on Apr 25

Commit

090b5ab

verified ·

1 Parent(s): a45e8a6

Update app.py

Browse files

Files changed (1) hide show

app.py +134 -1

app.py CHANGED Viewed

@@ -3344,4 +3344,137 @@ def get_env_stats(env_id):
         env_data = data.get(env_id)
     if not env_data:
-        return jsonify({"error": "Среда не

         env_data = data.get(env_id)
     if not env_data:
+        return jsonify({"error": "Среда ненайдена"}), 404
+    raw_logs = env_data.get("logs", [])
+    formatted_logs = []
+    for log in reversed(raw_logs):
+        try:
+            utc_dt = datetime.fromisoformat(log['time'])
+            almaty_dt = utc_dt + timedelta(hours=5)
+            time_str = almaty_dt.strftime('%Y-%m-%d %H:%M:%S')
+            formatted_logs.append({
+                "time": time_str,
+                "ip": log.get('ip', 'unknown'),
+                "ua": log.get('ua', 'unknown')
+            })
+        except:
+            continue
+    response_data = {
+        "id": env_id,
+        "keyword": env_data.get("keyword"),
+        "type": env_data.get("type", "closed"),
+        "hits": env_data.get("hits", 0),
+        "logs": formatted_logs
+    }
+    return jsonify(response_data)
+@app.route('/env/<env_id>')
+def serve_env(env_id):
+    new_token_to_set = None
+    with data_lock:
+        data = load_data()
+        env_data = data.get(env_id)
+        if not env_data or not isinstance(env_data, dict) or env_data.get("archived"):
+            return "Среда не найдена или заархивирована.", 404
+        current_log = {
+            "time": datetime.utcnow().isoformat(),
+            "ip": request.headers.get('X-Forwarded-For', request.remote_addr),
+            "ua": request.headers.get('User-Agent', '')[:150]
+        }
+        env_data['hits'] = env_data.get('hits', 0) + 1
+        if 'logs' not in env_data or not isinstance(env_data.get('logs'), list):
+            env_data['logs'] = []
+        env_data['logs'].append(current_log)
+        if len(env_data['logs']) > 30:
+            env_data['logs'] = env_data['logs'][-30:]
+        env_type = env_data.get("type", "closed")
+        if env_type == 'closed':
+            stored_token = env_data.get("device_token")
+            if not stored_token:
+                new_token_to_set = ''.join(random.choices(string.ascii_letters + string.digits, k=40))
+                env_data['device_token'] = new_token_to_set
+        data[env_id] = env_data
+        save_data(data)
+    keyword = env_data.get("keyword", "")
+    prompts_data = load_prompts()
+    if env_type == 'open':
+        return render_template_string(SYNKRIS_LOOK_TEMPLATE, keyword=keyword, prompts_data=prompts_data)
+    user_token = request.cookies.get(f'access_token_{env_id}')
+    stored_token = env_data.get("device_token")
+    if new_token_to_set:
+        resp = make_response(render_template_string(SYNKRIS_LOOK_TEMPLATE, keyword=keyword, prompts_data=prompts_data))
+        resp.set_cookie(f'access_token_{env_id}', new_token_to_set, max_age=31536000, httponly=True, samesite='Lax')
+        return resp
+    elif stored_token and user_token == stored_token:
+        return render_template_string(SYNKRIS_LOOK_TEMPLATE, keyword=keyword, prompts_data=prompts_data)
+    else:
+        return """
+        <!DOCTYPE html>
+        <html lang="ru">
+        <head>
+            <meta charset="UTF-8">
+            <meta name="viewport" content="width=device-width, initial-scale=1.0">
+            <title>Доступ запрещен</title>
+            <style>
+                body {
+                    font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, Helvetica, Arial, sans-serif;
+                    background: #111;
+                    color: #eee;
+                    display: flex;
+                    align-items: center;
+                    justify-content: center;
+                    height: 100vh;
+                    margin: 0;
+                    text-align: center;
+                    flex-direction: column;
+                }
+                .container {
+                    padding: 40px;
+                    background: #1a1a1a;
+                    border-radius: 16px;
+                    border: 1px solid #333;
+                    box-shadow: 0 10px 30px rgba(0,0,0,0.5);
+                }
+                h1 {
+                    color: #ff4d4d;
+                    margin-top: 0;
+                    margin-bottom: 15px;
+                    font-size: 2.5rem;
+                }
+                p {
+                    color: #aaa;
+                    font-size: 1.1rem;
+                    max-width: 400px;
+                }
+            </style>
+        </head>
+        <body>
+            <div class="container">
+                <h1>⛔ Доступ запрещен</h1>
+                <p>Эта ссылка уже привязана к другому устройству или браузеру.</p>
+            </div>
+        </body>
+        </html>
+        """, 403
+if __name__ == '__main__':
+    setup_initial_files()
+    download_db_from_hf()
+    if HF_TOKEN_WRITE:
+        backup_thread = threading.Thread(target=periodic_backup, daemon=True)
+        backup_thread.start()
+    port = int(os.environ.get('PORT', 7860))
+    app.run(debug=False, host='0.0.0.0', port=port)