build

.gitlab-ci.yml

@@ -7,9 +7,9 @@ variables:
   GIT_SSH_COMMAND: "ssh -i /home/gitlab-runner/.ssh/id_rsa -o IdentitiesOnly=yes"
 lint:
   stage: lint
-  # image: python:3.12
+  image: r.sectorq.eu/jaydee/builder-portainer:latest
   before_script:
-    - python3 -m pip install --break-system-packages flake8 black pylint tabulate prompt_toolkit
+    - python3 -m pip install --break-system-packages flake8 black pylint tabulate prompt_toolkit hvac
     - export PATH="$PATH:/home/gitlab-runner/.local/bin"
     # - echo "PATH is now: $PATH"
   script:
@@ -27,9 +27,9 @@ build-job:       # This job runs in the build stage, which runs first.
       - echo "$SSH_PRIVATE_KEY" | tr -d '\r' > ~/.ssh/id_rsa
       - chmod 600 ~/.ssh/id_rsa
       - pyinstaller --onefile portainer.py
-      - scp -o ConnectTimeout=5 -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null dist/portainer jd@192.168.80.222:/myapps/bin/ || true
-      - scp -o ConnectTimeout=5 -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null dist/portainer jd@morefine.home.lan:/myapps/bin/ || true
-      - scp -o ConnectTimeout=5 -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null dist/portainer jd@m-server.home.lan:/myapps/bin/ || true
+      #- scp -o ConnectTimeout=5 -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null dist/portainer jd@192.168.80.222:/myapps/bin/ || true
+      - scp -o ConnectTimeout=5 -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null dist/portainer jd@192.168.77.12:/myapps/bin/ || true
+      - scp -o ConnectTimeout=5 -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null dist/portainer jd@192.168.77.101:/myapps/bin/ || true
       - rm -rf /home/gitlab-runner/builds/1fLwHSKm2/0/jaydee/portainer.tmp
     artifacts:
       paths:

port.py

@@ -12,6 +12,7 @@ import base64
 import tabulate
 from git import Repo
 import requests
+import hvac
 from prompt_toolkit import prompt
 from prompt_toolkit.completion import WordCompleter
 from prompt_toolkit.shortcuts import checkboxlist_dialog
@@ -27,9 +28,10 @@ class Portainer:
     to perform API operations.
     """
 
-    def __init__(self, site, timeout=10):
+    def __init__(self, site, args=None, timeout=120):
         self.base_url = None
         self.token = None
+        self.args = args
         self.action = None
         self._debug = False
         self.timeout = timeout
@@ -39,7 +41,7 @@ class Portainer:
         self.stack_id = None
         self.stack_ids = []
         self.endpoint_name = None
-        self.endpoint_id = None
+        self.endpoint_id = args.endpoint_id
         
         # self.git_url = "https://gitlab.sectorq.eu/home/docker-compose.git"
         self.git_url = "git@gitlab.sectorq.eu:home/docker-compose.git"
@@ -109,11 +111,11 @@ class Portainer:
         ]
         self.log_mode = False
         self.hw_mode = False
-        self.all_data = {"containers": {}, "stacks": {}, "endpoints": {}}
+        self.all_data = {"containers": {}, "stacks": {}, "endpoints": {}, "services":{}}
         self.get_site(site)
         self.get_endpoints()
         self.get_stacks()
-        self.get_containers()
+        self.refresh_in_containers()
         
     def set_defaults(self, config):
         '''Set default configuration from provided config dictionary.'''
@@ -124,10 +126,13 @@ class Portainer:
             self.base_url = os.getenv(
                 "PORTAINER_URL", "https://portainer.sectorq.eu/api"
             )
-            self.token = "ptr_GCNUoFcTOaXm7k8ZxPdQGmrFIamxZPTydbserYofMHc="
+            # self.token = "ptr_GCNUoFcTOaXm7k8ZxPdQGmrFIamxZPTydbserYofMHc="
+            token_path = "portainer/token"
+            self.token = self.args.client.secrets.kv.v2.read_secret_version(path=token_path)['data']['data']['value']
         elif site == "port":
             self.base_url = os.getenv("PORTAINER_URL", "https://port.sectorq.eu/api")
-            self.token = "ptr_/5RkMCT/j3BTaL32vMSDtXFi76yOXRKVFOrUtzMsl5Y="
+            token_path = "port/token"
+            self.token = self.args.client.secrets.kv.v2.read_secret_version(path=token_path)['data']['data']['value']
         else:
             self.base_url = os.getenv(
                 "PORTAINER_URL", "https://portainer.sectorq.eu/api"
@@ -144,11 +149,30 @@ class Portainer:
         except ValueError:
             return False
 
+    def gotify_message(self, message):
+        payload = {
+            "title": "Updates in Portainer",
+            "message": message,
+            "priority": 5
+        }
+        '''Send a notification message via Gotify.'''
+        response = requests.post(
+            "https://gotify.sectorq.eu/message",
+            data=payload,
+            headers={"X-Gotify-Key": "ASn_fIAd5OVjm8c"}
+        )
+        # print("Status:", response.status_code)
+        # print("Response:", response.text)
+        pass
+    
     def _api_get(self, path, timeout=120):
         url = f"{self.base_url.rstrip('/')}{path}"
         headers = {"X-API-Key": f"{self.token}"}
         resp = requests.get(url, headers=headers, timeout=timeout)
-        resp.raise_for_status()
+        if resp.status_code != 200:
+            return resp.status_code
+            print(f"Error: {resp.status_code} - {resp.text}")
+        # resp.raise_for_status()
         return resp.json()
 
     def _api_post(self, path, json="", timeout=120):
@@ -208,7 +232,7 @@ class Portainer:
     def get_stacks(self, endpoint_id="all", timeout=20):
         '''Get a list of stacks for a specific endpoint or all endpoints.'''
         if endpoint_id != "all":
-            endpoint_id = self.get_endpoint_id(endpoint_id)
+            endpoint_id = self.get_endpoint_id()
         path = "/stacks"
         stcks = []
         stacks = self._api_get(path, timeout=timeout)
@@ -272,9 +296,11 @@ class Portainer:
         # input(json.dumps(self.stacks_all,indent=2))
         return stcks
 
-    def get_services(self, endpoint, stack, timeout=30):
+    def get_services(self, endpoint, timeout=30):
         '''Get a list of services for a specific stack on an endpoint.'''
-        path = f"/endpoints/{self.all_data['endpoints']['by_name'][endpoint]}/docker/services"
+        # print(json.dumps(self.all_data,indent=2))
+        path = f"/endpoints/{self.get_endpoint_id()}/docker/services"
+        # print(path)
         # path += f'?filters={{"label": ["com.docker.compose.project={stack}"]}}'
         services = self._api_get(path, timeout=timeout)
         return services
@@ -286,78 +312,137 @@ class Portainer:
         stats = self._api_get(path)
         print(stats)
 
-    def get_endpoint_id(self, endpoint):
+    def get_endpoint_id(self):
         '''Get endpoint ID from either ID or name input.'''
-        if self._is_number(endpoint):
-            self.endpoint_id = endpoint
-            self.endpoint_name = self.endpoints["by_id"][endpoint]
-            return endpoint
+        if self._is_number(self.args.endpoint_id):
+            self.endpoint_id = self.args.endpoint_id
+            self.endpoint_name = self.endpoints["by_id"][self.args.endpoint_id]
+            return self.args.endpoint_id
         else:
-            self.endpoint_name = endpoint
-            self.endpoint_id = self.endpoints["by_name"][endpoint]
-            return self.endpoints["by_name"][endpoint]
+            self.endpoint_name = self.args.endpoint_id
+            self.endpoint_id = self.endpoints["by_name"][self.args.endpoint_id]
+            return self.endpoints["by_name"][self.args.endpoint_id]
 
     def get_endpoint_name(self, endpoint):
         '''Get endpoint name from either ID or name input.'''
         if self._is_number(endpoint):
             self.endpoint_id = endpoint
-            self.endpoint_name = self.endpoints["by_id"][endpoint]
-            return self.endpoints["by_id"][endpoint]
+            self.endpoint_name = self.all_data["endpoints"]["by_id"][endpoint]
+            return self.all_data["endpoints"]["by_id"][endpoint]
         else:
             self.endpoint_name = endpoint
-            self.endpoint_id = self.endpoints["by_name"][endpoint]
+            self.endpoint_id = self.all_data["endpoints"]["by_name"][endpoint]
             return endpoint
 
-    def get_containers(self, endpoint="all", stack="all", timeout=30):
+    def refresh_in_containers(self):
+        print("Refreshing containers")
         '''Get a list of containers for a specific endpoint and stack.'''
         # print(json.dumps(self.all_data,indent=2))
         # print(endpoint)
         # print(stack)
         cont = []
         data = {}
-        if endpoint == "all":
-            for s in self.all_data["endpoints"]["by_id"]:
-                # print(s)
-                if stack == "all":
-                    if s not in self.all_data["stacks"]:
-                        continue
-                    if self.all_data["endpoints_status"][s] != 1:
+
+        eps = [ep for ep in self.all_data['endpoints']['by_id'].keys()]
+        #input(eps)
+        for endpoint in eps:
+            if self.all_data["endpoints_status"][endpoint] != 1:
+                print("Endpoint down")
                     # print(f"Endpoint {self.all_data["endpoints"]["by_id"][s]} is offline")
                 continue
-                    for e in self.all_data["stacks"][s]["by_name"]:
             path = (
-                            f"/endpoints/{s}/docker/containers/json"
-                            f'?all=1&filters={{"label": ["com.docker.compose.project={e}"]}}'
+                f"/endpoints/{endpoint}/docker/containers/json?all=1"
             )
             logging.info(f"request : {path}")
             try:
                 containers = self._api_get(path)
+                #input(json.dumps(containers, indent=2))
             except Exception as e:
                 print(f"failed to get containers from {path}: {e}")
                 continue
             contr = []
             try:
                 for c in containers:
+                    #input(c)
+                    cont.append([c["Names"][0].replace("/", ""),c["Id"], c['Image']])
+                    contr.append([c["Names"][0].replace("/", ""), c["Id"], c['Image']])
+                if self.all_data["endpoints"]["by_id"][endpoint] in data:
+                    data[self.all_data["endpoints"]["by_id"][endpoint]] = contr
+                    data[endpoint] = contr
+                else:
+                    data[self.all_data["endpoints"]["by_id"][endpoint]] = contr
+                    data[endpoint] = contr
+            except Exception as e:
+                logger.debug(
+                    f"Exception while getting containers for stack {e} ",
+                    f"on endpoint {self.all_data['endpoints']['by_id'][endpoint]}: {e}",
+                )
+        self.all_data["containers"] = data
+
+        #print(cont)
+        return cont
+
+    def get_containers(self):
+        '''Get a list of containers for a specific endpoint and stack.'''
+        # print(json.dumps(self.all_data,indent=2))
+        # print(endpoint)
+        # print(stack)
+        cont = []
+        data = {}
+        if self.args.endpoint_id == "all":
+            eps = [ep for ep in self.all_data['endpoints']['by_id'].keys()]
+        else:
+
+            eps = [self.get_endpoint_id()]
+        #input(eps)
+        for endpoint in eps:
+
+            # print(s)
+            #print(self.args.stack)
+            if self.args.stack in ["all", None]:
+                # input([id for id in self.all_data["stacks"][endpoint]['by_id'].keys()])
+                for s in [id for id in self.all_data["stacks"][endpoint]['by_id'].keys()]:
+                    # if s not in self.all_data["stacks"]:
+                    #     continue
+                    #input(self.all_data)
+                    if self.all_data["endpoints_status"][endpoint] != 1:
+                        # print(f"Endpoint {self.all_data["endpoints"]["by_id"][s]} is offline")
+                        continue
+                    # input(self.all_data["stacks"][endpoint]["by_name"])
+                    for e in self.all_data["stacks"][endpoint]["by_name"]:
+                        #input(e)
+                        path = (
+                            f"/endpoints/{endpoint}/docker/containers/json"
+                            f'?all=1&filters={{"label": ["com.docker.compose.project={e}"]}}'
+                        )
+                        logging.info(f"request : {path}")
+                        try:
+                            containers = self._api_get(path)
+                            #input(containers)
+                        except Exception as e:
+                            print(f"failed to get containers from {path}: {e}")
+                            continue
+                        contr = []
+                        try:
+                            for c in containers:
+                                # input(c)
                                 cont.append(c["Names"][0].replace("/", ""))
                                 contr.append(c["Names"][0].replace("/", ""))
-                            if self.all_data["endpoints"]["by_id"][s] in data:
-                                data[self.all_data["endpoints"]["by_id"][s]][e] = contr
+                            if self.all_data["endpoints"]["by_id"][endpoint] in data:
+                                data[self.all_data["endpoints"]["by_id"][endpoint]][e] = contr
                             else:
-                                data[self.all_data["endpoints"]["by_id"][s]] = {
+                                data[self.all_data["endpoints"]["by_id"][endpoint]] = {
                                     e: contr
                                 }
                         except Exception as e:
                             logger.debug(
                                 f"Exception while getting containers for stack {e} ",
-                                f"on endpoint {self.all_data['endpoints']['by_id'][s]}: {e}",
+                                f"on endpoint {self.all_data['endpoints']['by_id'][endpoint]}: {e}",
                             )
-                    # print(data)
-            self.all_data["containers"] = data
-        else:
-            self.get_containers()
 
-            for i in self.all_data["containers"][endpoint][stack]:
-                cont.append(i)
+                self.all_data["containers"] = data
+
+        #print(cont)
         return cont
 
     def stop_containers(self, endpoint, containers, timeout=130):
@@ -396,7 +481,11 @@ class Portainer:
             for s in self.all_data["webhooks"][endpoint]:
                 stcs.append([s, self.all_data["webhooks"][endpoint][s]])
         else:
+            try:
                 stcs.append([stack, self.all_data["webhooks"][endpoint][stack]])
+            except Exception as e:
+                print(f"Error: Stack {stack} not found on endpoint {endpoint}: {e}")
+
 
         # input(stcs)
         def update(c):
@@ -514,7 +603,7 @@ class Portainer:
                 p = "standalone"
                 env_path = f"{self.repo_dir}/{stack}/.env"
             # input(swarm_id)
-            self.endpoint_id = self.get_endpoint_id(endpoint)
+            self.endpoint_id = self.get_endpoint_id()
             if os.path.exists(self.repo_dir):
                 shutil.rmtree(self.repo_dir)
             else:
@@ -721,7 +810,6 @@ class Portainer:
                     }
                     self._api_post_file(path, self.endpoint_id, stack, envs, file)
 
-
     def print_stacks(self, endpoint="all"):
         """Print a table of stacks, optionally filtered by endpoint."""
         stacks = self.get_stacks()
@@ -729,6 +817,7 @@ class Portainer:
         data = []
         stack_names = []
         for stack in stacks:
+            print(stack)
             if endpoint is not None:
                 if not stack["EndpointId"] in self.endpoints["by_id"]:
                     continue
@@ -749,15 +838,245 @@ class Portainer:
                 logger.debug(
                     "KeyError getting endpoint name for stack %s : %s", stack["Name"], e
                 )
-            count += 1
 
+        data = sorted(data, key=lambda x: x[1])
         headers = ["StackID", "Name", "Endpoint"]
         print(tabulate.tabulate(data, headers=headers, tablefmt="github"))
         print(f"Total stacks: {count}")
         # print(sorted(stack_names))
 
+    def update_containers(self):
+        all_containers = self.all_data["containers"][self.args.endpoint_id]
+        #input(all_containers)
+        service_tuples = [(s[1], s[0]) for s in all_containers if "." not in s[0]]
+        service_tuples = sorted(service_tuples, key=lambda x: x[1])
+        service_dict = dict(service_tuples)
+        # input(service_tuples)
+        if self.args.service_id is None:
+            #services = [(s["Id"], s["Name"]) for s in self.get_stacks(endpoint_id)]
+            service_tuples.insert(0, ("__ALL__", "[Select ALL]"))
+            service_tuples.insert(0, ("__ONLY_CHECK__", "[Check Only]"))
+            service_ids = checkboxlist_dialog(
+                title="Select one service",
+                text="Choose a service:",
+                values=service_tuples
+            ).run()
+        elif self.args.service_id == "all":
+            service_ids = [s[0] for s in service_tuples if s[0] != "__ALL__" and s[0] != "__ONLY_CHECK__"]
+        else:
+            service_ids = [self.args.service_id]
+        if "__ONLY_CHECK__" in service_ids or self.args.update is False:
+            pull = False
+            print("Checking for updates only...")
+        else:
+            print("Checking for updates and pulling updates...")
+            pull = True
+        if "__ALL__" in service_ids:
+            service_ids = [s[0] for s in service_tuples if s[0] != "__ALL__" and s[0] != "__ONLY_CHECK__"]
+            
+        longest = 0
+        for a in service_dict.items():
+           # print(a[1])
+            if len(a[1]) > longest:
+                longest = len(a[1])
+        #print(longest)
+        ok = "\033[92m✔\033[0m"
+        err = "\033[91m✖\033[0m"
+        for service_id in service_ids:
+            # print(self.all_data["containers"][self.args.endpoint_id])
+
+            print("\033[?25l", end="")
+            print(f"{service_dict[service_id]:<{longest}}  ", end="", flush=True)
+            path = f"/docker/{self.get_endpoint_id()}/containers/{service_id}/image_status?refresh=true"
+            
+            try:
+                resp = self._api_get(path, timeout=20)
+            except ValueError as e:
+                print(f"Error restarting service: {e}")
+                return []
+            #print(resp)
+            if resp == 500:
+                print("?")
+            elif resp['Status'] == "outdated":
+                if pull:
+                    
+                    self.recreate_container(service_id, pull)
+                    #print(f"Service {service_dict[service_id]:<{longest}} : updated")
+                    self.gotify_message(f"Service {service_dict[service_id]} updated")
+                    print(ok, end=" ")
+                    for name, hash_, image in self.all_data["containers"][self.args.endpoint_id]:
+                        if name.startswith(service_dict[service_id]):
+                            print(image)
+                else:
+                    print(f"\r\033[4m{service_dict[service_id]:<{longest}}\033[0m  ", end="", flush=True)
+                    #print(f"\033[4m{service_dict[service_id]:<{longest}} {err}\033[0m")
+                    self.gotify_message(f"Service update available for {service_dict[service_id]}")
+                    print(err, end=" ")
+                    for name, hash_, image in self.all_data["containers"][self.args.endpoint_id]:
+                        if name.startswith(service_dict[service_id]):
+                            print(image)
+            else:
+                print(ok, end=" ")
+                for name, hash_, image in self.all_data["containers"][self.args.endpoint_id]:
+                    if name.startswith(service_dict[service_id]):
+                        print(image)
+        print("\033[?25h", end="")
+        return True
+
+    def update_service(self):
+        all_services = self.get_services(self.get_endpoint_id())
+        #input(all_services)
+        service_tuples = [(s['ID'], s['Spec']['Name']) for s in all_services]
+        service_tuples = sorted(service_tuples, key=lambda x: x[1])
+        service_dict = dict(service_tuples)
+        # input(service_tuples)
+        if self.args.service_id is None:
+            #services = [(s["Id"], s["Name"]) for s in self.get_stacks(endpoint_id)]
+            service_tuples.insert(0, ("__ALL__", "[Select ALL]"))
+            service_tuples.insert(0, ("__ONLY_CHECK__", "[Check Only]"))
+            service_ids = checkboxlist_dialog(
+                title="Select one service",
+                text="Choose a service:",
+                values=service_tuples
+            ).run()
+        elif self.args.service_id == "all":
+            service_ids = [s[0] for s in service_tuples if s[0] != "__ALL__" and s[0] != "__ONLY_CHECK__"]
+        else:
+            service_ids = [self.args.service_id]
+        if "__ONLY_CHECK__" in service_ids or self.args.update is False:
+            pull = False
+            print("Checking for updates only...")
+        else:
+            print("Checking for updates and pulling updates...")
+            pull = True
+        if "__ALL__" in service_ids:
+            service_ids = [s[0] for s in service_tuples if s[0] != "__ALL__" and s[0] != "__ONLY_CHECK__"]
+            
+        longest = 0
+        for a in service_dict.items():
+           # print(a[1])
+            if len(a[1]) > longest:
+                longest = len(a[1])
+        #print(longest)
+        ok = "\033[92m✔\033[0m"
+        err = "\033[91m✖\033[0m"
+        for service_id in service_ids:
+            print("\033[?25l", end="")
+            print(f"{service_dict[service_id]:<{longest}}  ", end="", flush=True)
+            path = f"/docker/{self.endpoint_id}/services/{service_id}/image_status?refresh=true"
+            
+            try:
+                resp = self._api_get(path, timeout=20)
+            except ValueError as e:
+                print(f"Error restarting service: {e}")
+                return []
+            
+            if resp['Status'] == "outdated":
+                if pull:
+                    self.restart_srv(service_id, pull)
+                    #print(f"Service {service_dict[service_id]:<{longest}} : updated")
+                    self.gotify_message(f"Service {service_dict[service_id]} updated")
+                    print(ok)
+                else:
+                    print(f"\r\033[4m{service_dict[service_id]:<{longest}}\033[0m  ", end="", flush=True)
+                    #print(f"\033[4m{service_dict[service_id]:<{longest}} {err}\033[0m")
+                    self.gotify_message(f"Service update available for {service_dict[service_id]}")
+                    print(err)
+            else:
+                print(ok)
+        print("\033[?25h", end="")
+        return True
+
+    def update_service2(self):
+        all_services = self.get_services(self.get_endpoint_id(self.args.endpoint_id))
+        
+        service_tuples = [(s['ID'], s['Spec']['Name']) for s in all_services]
+        service_tuples = sorted(service_tuples, key=lambda x: x[1])
+        service_dict = dict(service_tuples)
+        # input(service_tuples)
+        if self.args.service_id is None:
+            #services = [(s["Id"], s["Name"]) for s in self.get_stacks(endpoint_id)]
+            service_tuples.insert(0, ("__ALL__", "[Select ALL]"))
+            service_tuples.insert(0, ("__ONLY_CHECK__", "[Check Only]"))
+            service_ids = checkboxlist_dialog(
+                title="Select one service",
+                text="Choose a service:",
+                values=service_tuples
+            ).run()
+        elif self.args.service_id == "all":
+            service_ids = [s[0] for s in service_tuples if s[0] != "__ALL__" and s[0] != "__ONLY_CHECK__"]
+        else:
+            service_ids = [self.args.service_id]
+        if "__ONLY_CHECK__" in service_ids or self.args.update is False:
+            pull = False
+            print("Checking for updates only...")
+        else:
+            print("Checking for updates and pulling updates...")
+            pull = True
+        if "__ALL__" in service_ids:
+            service_ids = [s[0] for s in service_tuples if s[0] != "__ALL__" and s[0] != "__ONLY_CHECK__"]
+            
+        longest = 0
+        for a in service_dict.items():
+           # print(a[1])
+            if len(a[1]) > longest:
+                longest = len(a[1])
+        #print(longest)
+        ok = "\033[92m✔\033[0m"
+        err = "\033[91m✖\033[0m"
+        for service_id in service_ids:
+            print("\033[?25l", end="")
+            print(f"{service_dict[service_id]:<{longest}}  ", end="", flush=True)
+            path = f"/docker/{self.endpoint_id}/services/{service_id}/image_status?refresh=true"
+            
+            try:
+                resp = self._api_get(path, timeout=20)
+            except ValueError as e:
+                print(f"Error restarting service: {e}")
+                return []
+            
+            if resp['Status'] == "outdated":
+                if pull:
+                    self.restart_srv(service_id, pull)
+                    #print(f"Service {service_dict[service_id]:<{longest}} : updated")
+                    self.gotify_message(f"Service {service_dict[service_id]} updated")
+                    print(ok)
+                else:
+                    print(f"\r\033[4m{service_dict[service_id]:<{longest}}\033[0m  ", end="", flush=True)
+                    #print(f"\033[4m{service_dict[service_id]:<{longest}} {err}\033[0m")
+                    self.gotify_message(f"Service update available for {service_dict[service_id]}")
+                    print(err)
+            else:
+                print(ok)
+        print("\033[?25h", end="")
+        return True
+
+    def recreate_container(self,service_id, pull=False):
+        """Restart a service on an endpoint."""
+        path = f"/endpoints/{self.endpoint_id}/containers/{service_id}/recreate"
+        print(path)
+        params={"pullImage": pull}
+        try:
+            resp = self._api_post(path, json=params, timeout=20)
+            print(resp)
+        except ValueError as e:
+            print(f"Error restarting service: {e}")
+            return []
+
+    def restart_srv(self,service_id, pool=False):
+        """Restart a service on an endpoint."""
+        path = f"/endpoints/{self.endpoint_id}/forceupdateservice"
+        params={"serviceID": service_id, "pullImage": pool}
+        try:
+            resp = self._api_put(path, json=params, timeout=20)
+            print(resp)
+        except ValueError as e:
+            print(f"Error restarting service: {e}")
+            return []
+
     def restart_service(self, endpoint_id, service_id):
         stacks = [(s["Id"], s["Name"]) for s in self.get_stacks(endpoint_id)]
+        stacks = sorted(stacks, key=lambda x: x[1])
         stack_id = radiolist_dialog(
             title="Select one service",
             text="Choose a service:",
@@ -783,16 +1102,7 @@ class Portainer:
             values=svc_menu
         ).run()
 
-
-        """Restart a service on an endpoint."""
-        path = f"/endpoints/{self.endpoint_id}/forceupdateservice"
-        params={"serviceID": service_id, "pullImage": False}
-        try:
-            resp = self._api_put(path, json=params, timeout=20)
-            print(resp)
-        except ValueError as e:
-            print(f"Error restarting service: {e}")
-            return []
+        self.restart_srv(service_id, False)
         
         print(f"Service {service_id} : restarted")
         return True

portainer.py

@@ -14,6 +14,7 @@ import json
 import argparse
 import tty
 import termios
+import hvac
 from tabulate import tabulate
 from port import Portainer
 from prompt_toolkit import prompt
@@ -21,7 +22,24 @@ from prompt_toolkit.completion import WordCompleter
 from prompt_toolkit.shortcuts import checkboxlist_dialog
 from prompt_toolkit.shortcuts import radiolist_dialog
 
-VERSION = "0.1.13"
+VAULT_ADDR = os.environ.get("VAULT_ADDR", "http://192.168.77.101:8200")
+try:
+    VAULT_TOKEN = os.environ.get("VAULT_TOKEN")
+    if VAULT_TOKEN is None:
+        raise KeyError
+except KeyError:
+    VAULT_TOKEN = prompt("Valult root token : ", is_password=True)
+os.environ["VAULT_TOKEN"] = VAULT_TOKEN
+
+client = hvac.Client(url=VAULT_ADDR, token=VAULT_TOKEN)
+# Check if connected
+if client.is_authenticated():
+    print("Connected to Vault")
+else:
+    raise Exception("Failed to authenticate with Vault")
+# Specify the mount point of your KV engine
+
+VERSION = "0.1.14"
 
 defaults = {
     "endpoint_id": "vm01",
@@ -34,8 +52,6 @@ defaults = {
 
 cur_config = {}
 
-
-
 def load_config(defaults=defaults):
     '''Load configuration from /myapps/portainer.conf if it exists, else from env vars or defaults.'''  
     if os.path.exists("/myapps/portainer.conf"):
@@ -71,7 +87,7 @@ def load_config(defaults=defaults):
 
 
 
-cur_config = load_config(defaults)
+a = load_config(defaults)
 
 # ENV_VARS = [
 #     "PORTAINER_URL",
@@ -110,52 +126,27 @@ parser.add_argument(
     help="Endpoint ID to limit stack operations",
 )
 parser.add_argument(
-    "--refresh-environment", "-R", action="store_true", help="List endpoints"
-)
-parser.add_argument(
-    "--list-endpoints", "-E", action="store_true", help="List endpoints"
-)
-parser.add_argument("--list-stacks", "-l", action="store_true", help="List stacks")
-parser.add_argument("--print-all-data", "-A", action="store_true", help="List stacks")
-parser.add_argument(
-    "--list-containers", "-c", action="store_true", help="List containers"
-)
-parser.add_argument("--update-stack", "-U", action="store_true", help="Update stacks")
-parser.add_argument(
-    "--stop-containers", "-O", action="store_true", help="Stop containers"
-)
-parser.add_argument(
-    "--start-containers", "-X", action="store_true", help="Start containers"
-)
-parser.add_argument("--update-status", "-S", action="store_true", help="Update status")
-parser.add_argument(
-    "--get-stack", metavar="NAME_OR_ID", help="Get stack by name or numeric id"
+    "--service-id",
+    "-i",
+    type=str,
+    default=None,
+    help="Service ID to limit service operations",
 )
+parser.add_argument("--stack", "-s", type=str, default=None, nargs="+", help="Stack ID for operations")
 parser.add_argument("--action", "-a", type=str, default=None, help="Action to perform")
 parser.add_argument(
     "--autostart", "-Z", action="store_true", help="Auto-start created stacks"
 )
-parser.add_argument("--start-stack", "-x", action="store_true")
-parser.add_argument("--stop-stack", "-o", action="store_true")
-parser.add_argument("--secrets", "-q", action="store_true")
+parser.add_argument("--update", "-u", action="store_true", help="Update service if it exists")
 parser.add_argument("--debug", "-D", action="store_true")
-parser.add_argument("--create-stack", "-n", action="store_true")
-parser.add_argument("--create-stack_new2", "-N", action="store_true")
 parser.add_argument("--gpu", "-g", action="store_true")
-parser.add_argument("--create-stacks", "-C", action="store_true")
-parser.add_argument("--refresh-status", "-r", action="store_true")
-
-parser.add_argument("--stack", "-s", type=str, nargs="+", help="Stack ID for operations")
-parser.add_argument(
-    "--token-only", action="store_true", help="Print auth token and exit"
-)
 parser.add_argument("--timeout", type=int, default=10, help="Request timeout seconds")
 parser.add_argument("--deploy-mode", "-m", type=str, default="git", help="Deploy mode")
 parser.add_argument("--stack-mode", "-w", default=None, help="Stack mode")
 args = parser.parse_args()
 print("Running version:", VERSION)
 print("Environment:", args.site)
-
+args.client = client
 if args.site is not None:
     cur_config["PORTAINER_SITE"] = args.site
 if args.endpoint_id is not None:
@@ -228,10 +219,13 @@ def prompt_missing_args(args_in, defaults_in, fields, action=None,stacks=None):
             longest = len(a)
 
     for field, text in fields:
+        # print(field)
         value_in = getattr(args_in, field)
         default = defaults_in.get(f"PORTAINER_{field}".upper())
         cur_site = defaults_in.get("PORTAINER_SITE".upper())
         cur_env = defaults_in.get("PORTAINER_ENVIRONMENT_ID".upper())
+        
+        # print(value_in)
         if value_in is None:
             if default is not None:
                 prompt_text = f"{text} (default={default}) : "
@@ -249,7 +243,7 @@ def prompt_missing_args(args_in, defaults_in, fields, action=None,stacks=None):
                         # input(json.dumps(stacks, indent=2))
                         commands = [
                             'authentik', 'bitwarden', 'bookstack', 'dockermon', 'fail2ban', 'gitea', 'gitlab', 'grafana',
-                            'home-assistant', 'homepage', 'immich', 'influxdb', 'jupyter', 'kestra', 'mailu3',
+                            'hashicorp', 'home-assistant', 'homepage', 'immich', 'influxdb', 'jupyter', 'kestra', 'mailu3',
                             'mealie', 'mediacenter', 'mosquitto', 'motioneye', 'n8n', 'nebula', 'nextcloud', 'nginx',
                             'node-red', 'octoprint', 'ollama', 'onlyoffice', 'paperless-ngx', 'pihole', 'portainer-ce', 'rancher', 'registry',
                             'regsync', 'semaphore', 'unifibrowser', 'uptime-kuma', 'watchtower', 'wazuh', 'webhub', 'wordpress',
@@ -376,13 +370,13 @@ def prompt_missing_args(args_in, defaults_in, fields, action=None,stacks=None):
     return args
 
 
-
 if __name__ == "__main__":
     # Example usage: set PORTAINER_USER and PORTAINER_PASS in env, or pass literals below.
     # token = get_portainer_token(base,"admin","l4c1j4yd33Du5lo")  # or get_portainer_token(base, "admin", "secret")
     def signal_handler(sig, frame):
         logger.warning("Killed manually %s, %s", sig, frame)
         print("\nTerminated by user")
+        print("\033[?25h", end="")
         sys.exit(0)
     signal.signal(signal.SIGINT, signal_handler)
     os.system("cls" if os.name == "nt" else "clear")
@@ -393,6 +387,8 @@ if __name__ == "__main__":
             ("stop_stack","stop_stack"),
             ("start_stack","start_stack"),
             ("restart_service","restart_service"),
+            ("update_service","update_service"),
+            ("update_containers","update_containers"),
             ("list_stacks","list_stacks"),
             ("update_stack","update_stack"),
             ("secrets","secrets"),
@@ -430,7 +426,7 @@ if __name__ == "__main__":
         
         os.system("cls" if os.name == "nt" else "clear")
     # Example: list endpoints
-    por = Portainer(cur_config["PORTAINER_SITE"], timeout=args.timeout)
+    por = Portainer(cur_config["PORTAINER_SITE"], args)
     por.set_defaults(cur_config)
     if args.debug:
         por._debug = True
@@ -479,6 +475,8 @@ if __name__ == "__main__":
 
     if args.action == "create_stack":
         por.action = "create_stack"
+        print(cur_config)
+        print(args)
         args = prompt_missing_args(
             args,
             cur_config,
@@ -539,6 +537,36 @@ if __name__ == "__main__":
         por.restart_service(args.endpoint_id, "lala")
         sys.exit()
         
+    if args.action == "update_service":
+        
+        
+        
+        args = prompt_missing_args(
+            args,
+            cur_config,
+            [
+                ("site", "Site"),
+                ("endpoint_id", "Endpoint ID")
+            ],
+        )
+        por.update_service()
+        sys.exit()
+
+    if args.action == "update_containers":
+        
+        
+        
+        args = prompt_missing_args(
+            args,
+            cur_config,
+            [
+                ("site", "Site"),
+                ("endpoint_id", "Endpoint ID")
+            ],
+        )
+        por.update_containers()
+        sys.exit()
+
     if args.action == "list_stacks":
         args = prompt_missing_args(
             args,
@@ -554,12 +582,10 @@ if __name__ == "__main__":
 
     if args.action == "list_containers":
         print("Getting containers")
-        por.get_containers(args.endpoint_id, args.stack)
+        print(por.get_containers())
         sys.exit()
 
 
-
-
     if args.action == "update_stack":
         print("Updating stacks")
         por.update_stack(args.endpoint_id, args.stack, args.autostart)
@@ -572,7 +598,7 @@ if __name__ == "__main__":
         sys.exit()
 
     if args.action == "list_endpoints":
-        eps = por.get_endpoints()
+        eps = por.get_endpoints(args)
         export_data = []
         for i in eps["by_id"]:
             export_data.append([i, eps["by_id"][i]])