Additional Work Done on requests

index.py

@@ -1,14 +1,18 @@
 from dotenv import load_dotenv
-from flask import Flask, render_template
+from flask import Flask, render_template, request, session, redirect, url_for, abort
 from flask_apscheduler import APScheduler
-from flask_login import LoginManager
+from flask_migrate import Migrate
+from flask_sqlalchemy import SQLAlchemy
 import docker
+import pyjokes
 import time
 import os
+from datetime import timedelta
 
 # DEBUG
 from pprint import pprint
 
+
 def calculate_cpu_percent(stats, status):
     if (status != "running"):
         return 0
@@ -25,47 +29,94 @@ def calculate_cpu_percent(stats, status):
 
     return round(cpu_percent, 2)
 
+
 def calculate_ram_percent(stats, status):
     if (status != "running"):
         return 0
 
-    mem_used = stats["memory_stats"]["usage"] - stats["memory_stats"]["stats"]["cache"] + stats["memory_stats"]["stats"]["active_file"]
+    mem_used = stats["memory_stats"]["usage"] - stats["memory_stats"]["stats"]["cache"] + \
+        stats["memory_stats"]["stats"]["active_file"]
     limit = stats['memory_stats']['limit']
 
     return round(mem_used / limit * 100, 2)
 
-client = docker.from_env()
-#login_manager = LoginManager()
-app = Flask(__name__)
-scheduler = APScheduler()
+
 load_dotenv()
+scheduler = APScheduler()
+app = Flask(__name__)
+app.secret_key = "sdasdsadasdasdasdadadaasd53563da"
+app.permanent_session_lifetime = timedelta(days=1)
+client = docker.from_env()
+
+
+# db = SQLAlchemy(app)
+# migrate = Migrate(app, db)
+
 # login_manager.init_app(app)
 # https://flask-login.readthedocs.io/en/latest/
 
+
+@app.route("/login", methods=["POST", "GET"])
+def login():
+    if (request.method == "POST"):
+        user = request.form["email"]
+
+        #TODO: Validation
+        session.permanent = True
+        session['user'] = user
+        return redirect(url_for("index"))
+    else:
+        if "user" in session:
+            return redirect(url_for("index"))
+        return render_template("login.html")
+
+
+@app.route("/logout", methods=['GET', 'POST'])
+def logout():
+    session.pop("user", None)
+    return redirect(url_for("login"))
+
+
 @app.route("/", methods=['GET'])
 def index():
+    if "user" not in session:
+        return redirect(url_for("login"))
+     
     containers = client.containers.list(all=True)
-    return render_template("index.html", title="test", containers_len=len(containers), containers=containers)
+    punchline = pyjokes.get_joke(language="en", category="neutral")
+    return render_template("index.html", title="test", subtitle=punchline, containers_len=len(containers), containers=containers)
+
 
 @app.route("/parts/modal/container/<container_id>", methods=['GET'])
 def parts_container_modal(container_id):
+    if "user" not in session:
+        abort(401)
+    
     container = client.containers.get(container_id)
     stats = container.stats(stream=False)
     # com.docker.extension.detailed-description
     project = ''
+
+    pprint(container.attrs)
     if ('com.docker.compose.project' in container.attrs['Config']['Labels']):
         project = container.attrs['Config']['Labels']['com.docker.compose.project']
 
     return render_template("container_modal.html", name=container.name, project=project, ports=container.attrs['NetworkSettings']['Ports'], status=container.status, CPU=calculate_cpu_percent(stats, container.status), RAM=calculate_ram_percent(stats, container.status))
 
+
 @app.route("/api/<container_id>/<action>", methods=['DELETE', 'POST'])
 def api_stop_container(container_id, action):
+    if "user" not in session:
+        abort(401)
+
     container = client.containers.get(container_id)
-    #pprint(vars(container))
+
     try:
         match action:
             case 'stop':
                 container.stop()
+            case 'restart':
+                container.restart()
             case 'start':
                 container.start()
             case _:
@@ -76,12 +127,10 @@ def api_stop_container(container_id, action):
     except:
         return 'False'
 
-
-
 def scheduleTasks():
     print("This test runs every 3 seconds")
 
 if __name__ == '__main__':
-    #scheduler.add_job(id = 'scheduled_tasks', func=scheduleTasks, trigger="interval", seconds=3)
+    # scheduler.add_job(id = 'scheduled_tasks', func=scheduleTasks, trigger="interval", seconds=60)
     # scheduler.start()
     app.run(host=os.getenv('IP'), port=os.getenv('port'), debug=True)

templates/index.html

@@ -7,13 +7,20 @@
   <title>{{title}}</title>
   <link href="https://cdn.jsdelivr.net/npm/bootstrap@5.3.1/dist/css/bootstrap.min.css" rel="stylesheet"
     integrity="sha384-4bw+/aepP/YC94hEpVNVgiZdgIC5+VKNBQNGCHeKRQN+PtmoHDEXuppvnDJzQIu9" crossorigin="anonymous">
+    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/bootstrap-icons@1.10.5/font/bootstrap-icons.css">
   <style>
   </style>
 </head>
 
 <body data-bs-theme="dark">
   <div class="container-fluid">
-    <h1>Containers</h1>
+    <a href="/" class="d-flex text-decoration-none text-primary">
+      <i style="font-size: 3rem;" class="bi bi-stack me-2"></i>
+      <div class="mb-2">
+        <h2 class="mb-0">Containers</h2>
+        <small>{{subtitle}}</small>
+      </div>
+    </a>
     <div class="row row-cols-auto g-2">
       {% for container in containers %}
       <div class="col-6 col-sm-4 col-md-4 col-lg-3 col-xxl-2">
@@ -28,7 +35,8 @@
                 <input class="form-check-input" type="checkbox" id="flexSwitchCheckReverse"
                   onclick="toggleContainer('{{container.id}}', '{{container.status}}', this, event);" {% if
                   container.status=="running" %} checked {% endif %}>
-                <label class="form-check-label d-none d-md-block " for="flexSwitchCheckReverse">isRunning</label>
+                <label class="form-check-label d-none d-md-block "
+                  for="flexSwitchCheckReverse">{{container.status}}</label>
               </div>
             </div>
             <h4 class="text-truncate">{{container.attrs['Config']['Labels']['com.docker.compose.project']}}</h4>

templates/login.html

@@ -0,0 +1,72 @@
+<!doctype html>
+<html lang="en">
+
+<head>
+  <meta charset="utf-8">
+  <meta name="viewport" content="width=device-width, initial-scale=1, shrink-to-fit=no">
+  <title>Signin Template for Bootstrap</title>
+  <link href="https://cdn.jsdelivr.net/npm/bootstrap@5.3.1/dist/css/bootstrap.min.css" rel="stylesheet"
+    integrity="sha384-4bw+/aepP/YC94hEpVNVgiZdgIC5+VKNBQNGCHeKRQN+PtmoHDEXuppvnDJzQIu9" crossorigin="anonymous">
+  <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/bootstrap-icons@1.10.5/font/bootstrap-icons.css">
+  <style>
+html,
+body {
+  height: 100%;
+}
+
+body {
+  display: -ms-flexbox;
+  display: -webkit-box;
+  display: flex;
+  -ms-flex-align: center;
+  -ms-flex-pack: center;
+  -webkit-box-align: center;
+  align-items: center;
+  -webkit-box-pack: center;
+  justify-content: center;
+  padding-top: 40px;
+  padding-bottom: 40px;
+  background-color: #f5f5f5;
+}
+
+.form-signin {
+  width: 100%;
+  max-width: 330px;
+  padding: 15px;
+  margin: 0 auto;
+}
+.form-signin .checkbox {
+  font-weight: 400;
+}
+.form-signin .form-control {
+  position: relative;
+  box-sizing: border-box;
+  height: auto;
+  padding: 10px;
+  font-size: 16px;
+}
+.form-signin .form-control:focus {
+  z-index: 2;
+}
+.form-signin input[type="email"] {
+  margin-bottom: -1px;
+}
+.form-signin input[type="password"] {
+  margin-bottom: 10px;
+}
+
+  </style>
+</head>
+
+<body class="text-center">
+  <form class="form-signin" method="post">
+    <h1 class="h3 mb-3 font-weight-normal">Please sign in</h1>
+    <label for="inputEmail" class="">Email address</label>
+    <input type="email" id="inputEmail" name="email" class="form-control" placeholder="Email address" required autofocus>
+    <label for="inputPassword" class="">Password</label>
+    <input type="password" id="inputPassword" name="password"  class="form-control" placeholder="Password" required>
+    <button class="btn btn-lg btn-primary btn-block" type="submit">Sign in</button>
+  </form>
+</body>
+
+</html>