From c4e5189f531f3684b43eeeaa80a0ab42d49d78f8 Mon Sep 17 00:00:00 2001
From: bxdd <bxddream@gmail.com>
Date: Mon, 16 Oct 2023 21:58:50 +0800
Subject: [PATCH] [MNT] add docker

---
 learnware/client/container.py | 26 +++++++++++++++++---------
 setup.py                      |  1 +
 2 files changed, 18 insertions(+), 9 deletions(-)

diff --git a/learnware/client/container.py b/learnware/client/container.py
index e2c3f64..d5d5935 100644
--- a/learnware/client/container.py
+++ b/learnware/client/container.py
@@ -1,4 +1,5 @@
 import os
+import docker
 import pickle
 import atexit
 import tempfile
@@ -181,13 +182,26 @@ class ModelDockerContainer(ModelContainer):
         # call init method of parent of parent class
         super(ModelDockerContainer, self).__init__(model_config, learnware_zippath, build)
 
+    @staticmethod
+    def _generate_docker_container():
+        client = docker.from_env()
+        image = client.images.pull('continuumio/miniconda3')
+        return client.containers.create(image)
+    
+    @staticmethod
+    def _destroy_docker_container():
+        # destroy
+        pass
+
     def _setup_env_and_metadata(self):
         """setup env and set the input and output shape by communicating with docker"""
         raise NotImplementedError("_setup_env_and_metadata method is not implemented!")
 
     def _init_env(self):
         """create docker img according to the str self.docker_img, and creat the correponding conda python env"""
-
+        client = docker.from_env()
+        client.containers.run
+        image, build_log = client.images.pull(path=image_path, tag=tag)
         raise NotImplementedError("_init_env method is not implemented!")
 
     def _remove_env(self):
@@ -238,12 +252,6 @@ class LearnwaresContainer:
         self.cleanup = cleanup
         print("234", self.learnware_list)
 
-    def _generate_docker_img():
-        return None
-
-    def _destroy_docker_img():
-        pass
-
     def __enter__(self):
         if self.mode == "conda":
             self.learnware_containers = [
@@ -253,7 +261,7 @@ class LearnwaresContainer:
                 for _learnware, _zippath in zip(self.learnware_list, self.learnware_zippaths)
             ]
         else:
-            self.docker_img = self._generate_docker_img()
+            self.docker_img = ModelDockerContainer._generate_docker_container()
             self.learnware_containers = [
                 Learnware(
                     _learnware.id,
@@ -289,7 +297,7 @@ class LearnwaresContainer:
         self.results = None
 
         if self.mode == "docker":
-            self._destroy_docker_img()
+            ModelDockerContainer._destroy_docker_container()
 
     @staticmethod
     def _initialize_model_container(model: ModelCondaContainer):
diff --git a/setup.py b/setup.py
index 2c9725b..b79c79d 100644
--- a/setup.py
+++ b/setup.py
@@ -68,6 +68,7 @@ REQUIRED = [
     "sqlalchemy>=2.0.21",
     "shortuuid>=1.0.11",
     "geatpy>=2.7.0",
+    "docker>=6.1.3",
 ]
 
 if get_platform() != MACOS: