Automatic deployer script added

automatic_deployer.py

@@ -0,0 +1,58 @@
+import subprocess
+import os
+import time
+
+PORT=22060
+HOST='paffenroth-23.dyn.wpi.edu'
+
+def deploy():
+    print(f'Deploying the app...')
+    try:
+        # Run the setup.sh script
+        result = subprocess.run(
+            ['./setup.sh'],  # Command to execute
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+            check=True  # Raise an exception for non-zero exit codes
+        )
+        print(result.stdout.decode())  # Print standard output from the script
+    except subprocess.CalledProcessError as e:
+        print(f"Error occurred while deploying: {e.stderr.decode()}")
+
+def checkStatus(HOST, PORT):
+    print(f'Checking the status of the app...')
+    try:
+        # Run the netcat (nc) command to check if the port is open
+        result = subprocess.run(
+            ['nc', '-vz', HOST, str(PORT)],
+            stdout=subprocess.PIPE, stderr=subprocess.PIPE
+        )
+        
+        # If the return code is 0, the connection was successful
+        if result.returncode == 0:
+            print(f"Connection to {HOST}:{PORT} successful.")
+            return True
+        else:
+            print(f"Connection to {HOST}:{PORT} refused.")
+            return False
+    except Exception as e:
+        print(f"Error occurred: {e}")
+        return False
+
+def monitorStatus(HOST, PORT, checkInterval=5): # for 5 seconds will wait if the port is not available
+    print(f'Monitoring the status of the app...')
+    status=False
+    while True:
+        if checkStatus(HOST, PORT):
+            # If the connection is successful, run FUN and exit the loop
+            if not status:
+                deploy()
+            status=True
+        else:
+            # If connection is refused, wait and retry
+            print(f"Waiting for the server {HOST}:{PORT} to become available...")
+            time.sleep(checkInterval)
+            status = False
+
+# checkStatus(HOST, PORT)
+monitorStatus(HOST, PORT)

setup.sh

@@ -40,38 +40,42 @@ SSH_CONNECTION="ssh -i my_key -p $PORT $USER@$HOST"
 
 # TODO : Comment the old ssh key in the authorized_keys file
 # Use the new SSH key to connect to the server and install Docker
-$SSH_CONNECTION << EOF
-  # Update existing package list
-  sudo apt-get update
-  
-  # Install required packages for Docker
-  sudo apt-get install -y ca-certificates curl gnupg lsb-release
-  
-  # Add Docker's official GPG key
-  sudo mkdir -p /etc/apt/keyrings
-  curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo gpg --dearmor -o /etc/apt/keyrings/docker.gpg
-  
-  # Set up the Docker repository
-  echo \
-    "deb [arch=\$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu \
-    \$(lsb_release -cs) stable" | sudo tee /etc/apt/sources.list.d/docker.list > /dev/null
-
-  # Update the package list again
-  sudo apt-get update
-
-  # Install Docker Engine, Docker CLI, and containerd
-  sudo apt-get install -y docker-ce docker-ce-cli containerd.io docker-buildx-plugin docker-compose-plugin
-
-  # Start and enable Docker service
-  sudo systemctl start docker
-  sudo systemctl enable docker
-
-  # Add current user to docker group (to run Docker commands without sudo)
-  sudo usermod -aG docker $USER
-
-  # Verify Docker installation
-  docker --version
-EOF
+
+echo "Installing Docker on the server..."
+
+$SSH_CONNECTION "sudo apt-get update"
+
+# Install required packages for Docker
+echo "Installing required packages for Docker..."
+$SSH_CONNECTION "sudo apt-get install -y ca-certificates curl gnupg lsb-release"
+
+# Add Docker's official GPG key
+echo "Adding Docker's official GPG key..."
+$SSH_CONNECTION "sudo mkdir -p /etc/apt/keyrings"
+$SSH_CONNECTION "curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo gpg --dearmor -o /etc/apt/keyrings/docker.gpg"
+
+# Set up the Docker repository
+echo "Setting up the Docker repository..."
+$SSH_CONNECTION "echo \"deb [arch=\$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu \$(lsb_release -cs) stable\" | sudo tee /etc/apt/sources.list.d/docker.list > /dev/null"
+
+# Update the package list again
+$SSH_CONNECTION "sudo apt-get update"
+
+# Install Docker Engine, Docker CLI, and containerd
+echo "Installing Docker Engine, Docker CLI, and containerd..."
+$SSH_CONNECTION "sudo apt-get install -y docker-ce docker-ce-cli containerd.io"
+
+# Start and enable Docker service
+echo "Starting and enabling Docker service..."
+$SSH_CONNECTION "sudo systemctl start docker"
+$SSH_CONNECTION "sudo systemctl enable docker"
+
+# Add current user to docker group (to run Docker commands without sudo)
+echo "Adding current user to docker group..."
+$SSH_CONNECTION "sudo usermod -aG docker $USER"
+
+# Verify Docker installation
+$SSH_CONNECTION "docker --version"
 
 echo "Docker has been installed on the server."
 
@@ -79,25 +83,28 @@ echo "Docker has been installed on the server."
 
 DOCKER_NAME=mlopscs2
 
-# Build Docker image locally
-echo "Building Docker image..."
-# docker build -t $DOCKER_NAME .
-docker build --build-arg HF_TOKEN=$HF_TOKEN -t $DOCKER_NAME .
 
-# Save the Docker image to a tar file
-docker save -o $DOCKER_NAME.tar $DOCKER_NAME
+# # Build Docker image locally
+# echo "Building Docker image..."
+# # docker build -t $DOCKER_NAME .
+# docker build --build-arg HF_TOKEN=$HF_TOKEN -t $DOCKER_NAME .
 
-# Copy the Docker image tar to the server
-echo "Copying Docker image to the server..."
-scp -i "$old_ssh_key_name" -P "$PORT" $DOCKER_NAME.tar "$USER@$HOST:/home/$USER/"
+# # Save the Docker image to a tar file
+# docker save -o $DOCKER_NAME.tar $DOCKER_NAME
 
-# Load the Docker image on the server
-$SSH_CONNECTION "docker load -i /home/$USER/$DOCKER_NAME.tar"
+# # Copy the Docker image tar to the server
+# echo "Copying Docker image to the server..."
+# scp -i "$old_ssh_key_name" -P "$PORT" $DOCKER_NAME.tar "$USER@$HOST:/home/$USER/"
 
-# Run the Docker container on the server with restart policy
-$SSH_CONNECTION "docker run -d --restart unless-stopped -p 5000:5000 $DOCKER_NAME"
+# # Load the Docker image on the server
+# $SSH_CONNECTION "docker load -i /home/$USER/$DOCKER_NAME.tar"
 
-# Clean up the Docker image tar file
-$SSH_CONNECTION "rm /home/$USER/$DOCKER_NAME.tar"
+# pull the docker image from venkateshroshan/mlops-case-study2:latest
+$SSH_CONNECTION "docker pull venkateshroshan/mlops-case-study2:latest"
+
+echo "Docker image has been copied to the server."
+
+# Run the Docker container on the server with restart policy
+$SSH_CONNECTION "docker run -d --restart unless-stopped -p 8013:5000 $DOCKER_NAME"
 
 echo "Docker container is running on the server."