[Image] Docker run restart policy support (#40)

Makefile

@@ -15,7 +15,7 @@ fmt:
 .PHONY: fmt-check
 fmt-check:
 	@echo "Running black fmt check..."
-	$(VENV_PYTHON) -m black --skip-string-normalization --check --diff -S --exclude='.*venv.*' .
+	$(VENV_PYTHON) -m black --skip-string-normalization --check --diff .
 
 .PHONY: test
 test: test-unit test-integ

manof/image.py

@@ -180,7 +180,7 @@ def run(self):
         for env in self._update_env_override():
 
             # single environment variable means set it to itself (x=x), thus forwarding the variable from the
-            # outter env into the docker env
+            # outer env into the docker env
             if isinstance(env, str):
                 lvalue = env
                 rvalue = os.environ.get(lvalue, None)
@@ -212,6 +212,10 @@ def run(self):
             if cap:
                 command += '--cap-drop={0} '.format(cap)
 
+        # set restart policy
+        if self.restart:
+            command += '--restart={0} '.format(self.restart)
+
         command = self._add_device_arguments(command)
 
         # set tag
@@ -727,6 +731,10 @@ def device_write_iops(self):
 
         return None
 
+    @property
+    def restart(self):
+        return None
+
     def to_dict(self):
         d = super(Image, self).to_dict()
         for idx, item in enumerate(d['volumes']):

pyproject.toml

@@ -1,9 +1,22 @@
+[project]
+name = "manof"
+requires-python = ">=3.7,<3.10"
+version = "0.1.2"
+authors = [
+    {name = "LiranBG", email = "liran_ben_gida@mckinsey.com"},
+]
+
+[tool.setuptools]
+py-modules = []
+
 # please keep this to a minimum - defaults are good
 [tool.black]
 exclude = '''
-/(
+^/(   # anchor to the root
+  (   # exclude these directories
     \.git
-  | \.venv
-  | \venv
-)/
+    | \.venv
+    | venv
+  )/
+)
 '''

tests/integration/cases/shell_commands/artifacts/manofest.py

@@ -26,12 +26,31 @@ def image_name(self):
 
     @property
     def command(self):
-        return '/bin/sh -c "echo \'{0}\'"'.format(self.name)
+        return '/bin/sh -c "echo \'{0}\' && sleep infinity"'.format(self.name)
+
+
+class TestImage2(TestImage):
+    @property
+    def restart(self):
+        return 'on-failure:5'
+
+    @property
+    def memory(self):
+        return '6Mib'
+
+    @property
+    def cpus(self):
+        return '1'
+
+    @property
+    def cap_add(self):
+        return ['SYS_ADMIN']
 
 
 class SomeGroup(manof.Group):
     @property
     def members(self):
         return [
             'TestImage',
+            'TestImage2',
         ]