diff mbox series

[v1,8/8] docker: docker.py adding age check command

Message ID 20180618130709.19730-9-alex.bennee@linaro.org
State Superseded
Headers show
Series Docker tweaks for TCG testing | expand

Commit Message

Alex Bennée June 18, 2018, 1:07 p.m. UTC
This is useful for querying if an image is too old.

Signed-off-by: Alex Bennée <alex.bennee@linaro.org>


---
v7
  - use cached info
---
 tests/docker/docker.py | 46 +++++++++++++++++++++++++++++++++---------
 1 file changed, 36 insertions(+), 10 deletions(-)

-- 
2.17.1

Comments

Philippe Mathieu-Daudé June 18, 2018, 4:06 p.m. UTC | #1
On 06/18/2018 10:07 AM, Alex Bennée wrote:
> This is useful for querying if an image is too old.

> 

> Signed-off-by: Alex Bennée <alex.bennee@linaro.org>

> 

> ---

> v7

>   - use cached info

> ---

>  tests/docker/docker.py | 46 +++++++++++++++++++++++++++++++++---------

>  1 file changed, 36 insertions(+), 10 deletions(-)

> 

> diff --git a/tests/docker/docker.py b/tests/docker/docker.py

> index c157ff826a..da1ed4e879 100755

> --- a/tests/docker/docker.py

> +++ b/tests/docker/docker.py

> @@ -29,6 +29,7 @@ from tarfile import TarFile, TarInfo

>  from StringIO import StringIO

>  from shutil import copy, rmtree

>  from pwd import getpwuid

> +from datetime import datetime,timedelta


datetime,<space>timedelta

Reviewed-by: Philippe Mathieu-Daudé <f4bug@amsat.org>


>  

>  

>  FILTERED_ENV_NAMES = ['ftp_proxy', 'http_proxy', 'https_proxy']

> @@ -185,6 +186,9 @@ class Docker(object):

>          except subprocess.CalledProcessError:

>              return None

>  

> +    def get_image_creation_time(self, info):

> +        return json.loads(info)[0]["Created"]

> +

>      def get_image_dockerfile_checksum(self, tag):

>          resp = self.inspect_tag(tag)

>          labels = json.loads(resp)[0]["Config"].get("Labels", {})

> @@ -456,11 +460,14 @@ class CheckCommand(SubCommand):

>      def args(self, parser):

>          parser.add_argument("tag",

>                              help="Image Tag")

> -        parser.add_argument("dockerfile",

> -                            help="Dockerfile name")

> +        parser.add_argument("dockerfile", default=None,

> +                            help="Dockerfile name", nargs='?')

> +        parser.add_argument("--checktype", choices=["checksum", "age"],

> +                            default="checksum", help="check type")

> +        parser.add_argument("--olderthan", default=60, type=int,

> +                            help="number of minutes")

>  

>      def run(self, args, argv):

> -        dockerfile = open(args.dockerfile, "rb").read()

>          tag = args.tag

>  

>          dkr = Docker()

> @@ -469,13 +476,32 @@ class CheckCommand(SubCommand):

>              print("Image does not exist")

>              return 1

>  

> -        if dkr.image_matches_dockerfile(tag, dockerfile):

> -            if not args.quiet:

> -                print("Image is up to date")

> -            return 0

> -        else:

> -            print("Image needs updating")

> -            return 1

> +        if args.checktype == "checksum":

> +            if not args.dockerfile:

> +                print("Need a dockerfile for tag:%s" % (tag))

> +                return 1

> +

> +            dockerfile = open(args.dockerfile, "rb").read()

> +

> +            if dkr.image_matches_dockerfile(tag, dockerfile):

> +                if not args.quiet:

> +                    print("Image is up to date")

> +                return 0

> +            else:

> +                print("Image needs updating")

> +                return 1

> +        elif args.checktype == "age":

> +            timestr = dkr.get_image_creation_time(info).split(".")[0]

> +            created = datetime.strptime(timestr, "%Y-%m-%dT%H:%M:%S")

> +            past = datetime.now() - timedelta(minutes=args.olderthan)

> +            if created < past:

> +                print ("Image created @ %s more than %d minutes old" %

> +                       (timestr, args.olderthan))

> +                return 1

> +            else:

> +                if not args.quiet:

> +                    print ("Image less than %d minutes old" % (args.olderthan))

> +                return 0

>  

>  

>  def main():

>
diff mbox series

Patch

diff --git a/tests/docker/docker.py b/tests/docker/docker.py
index c157ff826a..da1ed4e879 100755
--- a/tests/docker/docker.py
+++ b/tests/docker/docker.py
@@ -29,6 +29,7 @@  from tarfile import TarFile, TarInfo
 from StringIO import StringIO
 from shutil import copy, rmtree
 from pwd import getpwuid
+from datetime import datetime,timedelta
 
 
 FILTERED_ENV_NAMES = ['ftp_proxy', 'http_proxy', 'https_proxy']
@@ -185,6 +186,9 @@  class Docker(object):
         except subprocess.CalledProcessError:
             return None
 
+    def get_image_creation_time(self, info):
+        return json.loads(info)[0]["Created"]
+
     def get_image_dockerfile_checksum(self, tag):
         resp = self.inspect_tag(tag)
         labels = json.loads(resp)[0]["Config"].get("Labels", {})
@@ -456,11 +460,14 @@  class CheckCommand(SubCommand):
     def args(self, parser):
         parser.add_argument("tag",
                             help="Image Tag")
-        parser.add_argument("dockerfile",
-                            help="Dockerfile name")
+        parser.add_argument("dockerfile", default=None,
+                            help="Dockerfile name", nargs='?')
+        parser.add_argument("--checktype", choices=["checksum", "age"],
+                            default="checksum", help="check type")
+        parser.add_argument("--olderthan", default=60, type=int,
+                            help="number of minutes")
 
     def run(self, args, argv):
-        dockerfile = open(args.dockerfile, "rb").read()
         tag = args.tag
 
         dkr = Docker()
@@ -469,13 +476,32 @@  class CheckCommand(SubCommand):
             print("Image does not exist")
             return 1
 
-        if dkr.image_matches_dockerfile(tag, dockerfile):
-            if not args.quiet:
-                print("Image is up to date")
-            return 0
-        else:
-            print("Image needs updating")
-            return 1
+        if args.checktype == "checksum":
+            if not args.dockerfile:
+                print("Need a dockerfile for tag:%s" % (tag))
+                return 1
+
+            dockerfile = open(args.dockerfile, "rb").read()
+
+            if dkr.image_matches_dockerfile(tag, dockerfile):
+                if not args.quiet:
+                    print("Image is up to date")
+                return 0
+            else:
+                print("Image needs updating")
+                return 1
+        elif args.checktype == "age":
+            timestr = dkr.get_image_creation_time(info).split(".")[0]
+            created = datetime.strptime(timestr, "%Y-%m-%dT%H:%M:%S")
+            past = datetime.now() - timedelta(minutes=args.olderthan)
+            if created < past:
+                print ("Image created @ %s more than %d minutes old" %
+                       (timestr, args.olderthan))
+                return 1
+            else:
+                if not args.quiet:
+                    print ("Image less than %d minutes old" % (args.olderthan))
+                return 0
 
 
 def main():