Initial YakPanel commit

class_v2/btdockerModelV2/volumeModel.py

@@ -0,0 +1,158 @@
+# coding: utf-8
+# -------------------------------------------------------------------
+# YakPanel
+# -------------------------------------------------------------------
+# Copyright (c) 2015-2099 YakPanel(www.yakpanel.com) All rights reserved.
+# -------------------------------------------------------------------
+# Author: wzz <wzz@yakpanel.com>
+# -------------------------------------------------------------------
+
+# ------------------------------
+# Docker模型
+# ------------------------------
+import docker.errors
+import public
+from btdockerModelV2 import dk_public as dp
+from btdockerModelV2.dockerBase import dockerBase
+from public.validate import Param
+
+class main(dockerBase):
+
+    def docker_client(self, url):
+        return dp.docker_client(url)
+
+    def get_volume_container_name(self, volume_detail, container_list):
+        '''
+        拼接对应的容器名与卷名
+        @param volume_detail: 卷字典
+        @param container_list: 容器详情列表
+        @return:
+        '''
+        try:
+            for container in container_list:
+                if not container['Mounts']:
+                    continue
+                for mount in container['Mounts']:
+                    if "Name" not in mount:
+                        continue
+                    if volume_detail['Name'] == mount['Name']:
+                        volume_detail['container'] = container['Names'][0].replace("/", "")
+            if 'container' not in volume_detail:
+                volume_detail['container'] = ''
+        except:
+            volume_detail['container'] = ''
+
+        return volume_detail
+
+    def get_volume_list(self, args):
+        """
+        :param self._url: 链接docker的URL
+        :return:
+        """
+        try:
+            data = list()
+            from btdockerModelV2.dockerSock import volume
+            sk_volume = volume.dockerVolume()
+            volume_list = sk_volume.get_volumes()
+
+            from btdockerModelV2.dockerSock import container
+            sk_container = container.dockerContainer()
+            container_list = sk_container.get_container()
+
+            if "Volumes" in volume_list and type(volume_list["Volumes"]) == list:
+                for v in volume_list["Volumes"]:
+                    v["CreatedAt"] = dp.convert_timezone_str_to_timestamp(v["CreatedAt"])
+                    data.append(self.get_volume_container_name(v, container_list))
+
+                return public.return_message(0, 0, sorted(data, key=lambda x: x['CreatedAt'], reverse=True))
+            else:
+                return public.return_message(0, 0, [])
+        except Exception as e:
+            return public.return_message(-1, 0, [])
+
+    def add(self, args):
+        """
+        添加一个卷
+        :param name
+        :param driver  local
+        :param driver_opts (dict) – Driver options as a key-value dictionary
+        :param labels str
+        :return:
+        """
+        try:
+            args.driver_opts = args.get("driver_opts", "")
+            args.labels = args.get("labels", "")
+            if args.driver_opts != "":
+                args.driver_opts = dp.set_kv(args.driver_opts)
+            if args.labels != "":
+                args.labels = dp.set_kv(args.labels)
+
+            if len(args.name) < 2:
+                return public.return_message(-1, 0, public.lang("Volume names can be no less than 2 characters long!"))
+
+            self.docker_client(self._url).volumes.create(
+                name=args.name,
+                driver=args.driver,
+                driver_opts=args.driver_opts if args.driver_opts else None,
+                labels=args.labels if args.labels != "" else None
+            )
+            dp.write_log("Add storage volume [{}] success!".format(args.name))
+            return public.return_message(0, 0, public.lang("successfully added!"))
+        except docker.errors.APIError as e:
+            if "volume name is too short, names should be at least two alphanumeric characters" in str(e):
+                return public.return_message(-1, 0, public.lang("Volume names can be no less than 2 characters long!"))
+            if "volume name" in str(e):
+                return public.return_message(-1, 0, public.lang("Volume name already exists!"))
+            return public.return_message(-1, 0, public.lang("addition failed {}", e))
+
+        except Exception as e:
+            if "driver_opts must be a dictionary" in str(e):
+                return public.return_message(-1, 0, public.lang("Driver option tags must be dictionary/key-value pairs!"))
+            return public.return_message(-1, 0, public.lang("Add failed! {}", e))
+
+    def remove(self, args):
+        """
+        删除一个卷
+        :param name  volume name
+        :param args:
+        :return:
+        """
+
+        # 校验参数
+        try:
+            args.validate([
+                Param('name').Require().String(),
+            ], [
+                public.validate.trim_filter(),
+            ])
+        except Exception as ex:
+            public.print_log("error info: {}".format(ex))
+            return public.return_message(-1, 0, str(ex))
+        try:
+            obj = self.docker_client(self._url).volumes.get(args.name)
+            obj.remove()
+            dp.write_log("Delete volume [{}] successful!".format(args.name))
+            return public.return_message(0, 0, public.lang("successfully delete"))
+
+        except docker.errors.APIError as e:
+            if "volume is in use" in str(e):
+                return public.return_message(-1, 0, public.lang("The storage volume is in use and cannot be deleted!"))
+            if "no such volume" in str(e):
+                return public.return_message(-1, 0, public.lang("The storage volume does not exist!"))
+            return public.return_message(-1, 0, public.lang("Delete failed! {}", e))
+
+    def prune(self, args):
+        """
+        删除无用的卷
+        :param args:
+        :return:
+        """
+        try:
+            res = self.docker_client(self._url).volumes.prune()
+            if not res['VolumesDeleted']:
+                return public.return_message(-1, 0, public.lang("No useless storage volumes!"))
+
+            dp.write_log("Delete useless storage volume successfully!")
+            return public.return_message(0, 0, public.lang("successfully delete!"))
+        except docker.errors.APIError as e:
+            return public.return_message(-1, 0, public.lang("Delete failed! {}", e))