Spaces:

GamerC0der
/

db

Sleeping

App Files Files Community

GamerC0der commited on 26 days ago

Commit

192f097

verified ·

1 Parent(s): 4f23300

Update app.py

Browse files

Files changed (1) hide show

app.py +54 -56

app.py CHANGED Viewed

@@ -2,6 +2,7 @@ import os
 import time
 import uuid
 import psutil
 import pymysql
 from flask import Flask, jsonify, request, abort, render_template_string
@@ -21,23 +22,47 @@ TOTAL_MAX_MB = 25000
 start_time = time.time()
 db_registry = {}
 DASHBOARD_TEMPLATE = """
 <!doctype html>
 <html>
 <head>
   <title>MySQL DB Manager Dashboard</title>
   <style>
-    body { font-family: Arial, sans-serif; margin: 20px; }
-    h1, h2 { margin-bottom: 0.3em; }
     .section { margin-bottom: 2em; }
-    table { border-collapse: collapse; width: 100%; margin-top: 0.5em; }
-    th, td { border: 1px solid #ddd; padding: 6px 8px; font-size: 14px; }
-    th { background: #f5f5f5; text-align: left; }
-    code { background: #f0f0f0; padding: 2px 4px; }
-    .small { font-size: 12px; color: #666; }
-    input[type="text"], input[type="number"] { padding: 4px; }
-    button { padding: 6px 10px; cursor: pointer; }
-    .error { color: #b00020; }
   </style>
 </head>
 <body>
@@ -157,20 +182,9 @@ def get_system_stats():
     disk_used_gb = disk.used / (1024**3)
     disk_percent = disk.percent
     return {
-        "cpu": {
-            "usage_percent": cpu_percent,
-            "cores_detected": cpu_count,
-        },
-        "ram": {
-            "total_gb": ram_total_gb,
-            "used_gb": ram_used_gb,
-            "usage_percent": ram_percent,
-        },
-        "disk": {
-            "total_gb": disk_total_gb,
-            "used_gb": disk_used_gb,
-            "usage_percent": disk_percent,
-        },
     }
 def total_allocated_mb():
@@ -180,15 +194,7 @@ def total_allocated_mb():
 def dashboard():
     system_stats = get_system_stats()
     uptime_seconds = int(time.time() - start_time)
-    dbs = [
-        {
-            "uuid": k,
-            "name": v["db_name"],
-            "size_mb": v["size_mb"],
-            "created_at": int(v["created_at"]),
-        }
-        for k, v in db_registry.items()
-    ]
     return render_template_string(
         DASHBOARD_TEMPLATE,
         system_stats=system_stats,
@@ -242,18 +248,18 @@ def create_db():
         cur.close()
         conn.close()
     except Exception as e:
-        return jsonify({"error": f"Failed to create database: {e}"}), 500
-    db_registry[db_uuid] = {
-        "db_name": db_name,
-        "size_mb": size_mb,
-        "created_at": time.time(),
-    }
-    return jsonify({
-        "uuid": db_uuid,
-        "db_name": db_name,
-        "size_mb": size_mb,
-        "query_endpoint": f"/api/{db_uuid}/query",
-    })
 @app.route("/api/<db_uuid>/query", methods=["POST"])
 def db_query(db_uuid):
@@ -268,24 +274,16 @@ def db_query(db_uuid):
         conn = get_mysql_connection(database=db_name)
         cur = conn.cursor()
         cur.execute(query)
-        rows = None
-        if cur.description is not None:
-            rows = cur.fetchall()
         try:
             conn.commit()
         except Exception:
             conn.rollback()
         cur.close()
         conn.close()
-        return jsonify({
-            "status": "ok",
-            "rows": rows,
-        })
     except Exception as e:
-        return jsonify({
-            "status": "error",
-            "error": str(e),
-        }), 500
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)

 import time
 import uuid
 import psutil
+import subprocess
 import pymysql
 from flask import Flask, jsonify, request, abort, render_template_string
 start_time = time.time()
 db_registry = {}
+def ensure_mysql_running():
+    try:
+        conn = pymysql.connect(
+            host=MYSQL_CONFIG["host"],
+            port=MYSQL_CONFIG["port"],
+            user=MYSQL_CONFIG["user"],
+            password=MYSQL_CONFIG["password"]
+        )
+        conn.close()
+    except Exception:
+        try:
+            subprocess.Popen(["mysqld_safe"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+            time.sleep(5)
+        except Exception:
+            try:
+                subprocess.Popen(["service", "mysql", "start"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+                time.sleep(5)
+            except Exception:
+                pass
+ensure_mysql_running()
 DASHBOARD_TEMPLATE = """
 <!doctype html>
 <html>
 <head>
   <title>MySQL DB Manager Dashboard</title>
   <style>
+    body { font-family: Arial, sans-serif; margin: 20px; background-color: #121212; color: #e0e0e0; }
+    h1, h2 { margin-bottom: 0.3em; color: #ffffff; }
     .section { margin-bottom: 2em; }
+    table { border-collapse: collapse; width: 100%; margin-top: 0.5em; background-color: #1e1e1e; color: #e0e0e0; }
+    th, td { border: 1px solid #333; padding: 6px 8px; font-size: 14px; }
+    th { background: #2a2a2a; text-align: left; color: #fff; }
+    code { background: #2a2a2a; padding: 2px 4px; color: #f5c518; }
+    .small { font-size: 12px; color: #aaa; }
+    input[type="text"], input[type="number"] { padding: 4px; background: #1e1e1e; color: #e0e0e0; border: 1px solid #333; }
+    button { padding: 6px 10px; cursor: pointer; background: #1565c0; color: #fff; border: none; border-radius: 3px; }
+    button:hover { background: #1e88e5; }
+    .error { color: #ff5252; }
+    a, a:visited { color: #90caf9; }
   </style>
 </head>
 <body>
     disk_used_gb = disk.used / (1024**3)
     disk_percent = disk.percent
     return {
+        "cpu": {"usage_percent": cpu_percent, "cores_detected": cpu_count},
+        "ram": {"total_gb": ram_total_gb, "used_gb": ram_used_gb, "usage_percent": ram_percent},
+        "disk": {"total_gb": disk_total_gb, "used_gb": disk_used_gb, "usage_percent": disk_percent},
     }
 def total_allocated_mb():
 def dashboard():
     system_stats = get_system_stats()
     uptime_seconds = int(time.time() - start_time)
+    dbs = [{"uuid": k, "name": v["db_name"], "size_mb": v["size_mb"], "created_at": int(v["created_at"])} for k, v in db_registry.items()]
     return render_template_string(
         DASHBOARD_TEMPLATE,
         system_stats=system_stats,
         cur.close()
         conn.close()
     except Exception as e:
+        ensure_mysql_running()
+        try:
+            conn = get_mysql_connection()
+            conn.autocommit(True)
+            cur = conn.cursor()
+            cur.execute(f"CREATE DATABASE `{db_name}`")
+            cur.close()
+            conn.close()
+        except Exception as e2:
+            return jsonify({"error": f"Failed to create database: {e2}"}), 500
+    db_registry[db_uuid] = {"db_name": db_name, "size_mb": size_mb, "created_at": time.time()}
+    return jsonify({"uuid": db_uuid, "db_name": db_name, "size_mb": size_mb, "query_endpoint": f"/api/{db_uuid}/query"})
 @app.route("/api/<db_uuid>/query", methods=["POST"])
 def db_query(db_uuid):
         conn = get_mysql_connection(database=db_name)
         cur = conn.cursor()
         cur.execute(query)
+        rows = cur.fetchall() if cur.description else None
         try:
             conn.commit()
         except Exception:
             conn.rollback()
         cur.close()
         conn.close()
+        return jsonify({"status": "ok", "rows": rows})
     except Exception as e:
+        return jsonify({"status": "error", "error": str(e)}), 500
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)