WIP slapos.cache.source

setup.py

@@ -80,6 +80,7 @@ setup(name=name,
         'slapos.cli': [
           # Utilities
           'cache lookup = slapos.cli.cache:CacheLookupCommand',
+          'cache source = slapos.cli.cache_source:CacheLookupCommand',
           # SlapOS Node commands
           'node bang = slapos.cli.bang:BangCommand',
           'node format = slapos.cli.format:FormatCommand',

slapos/cli/cache_source.py

+# -*- coding: utf-8 -*-
+##############################################################################
+#
+# Copyright (c) 2010-2014 Vifib SARL and Contributors.
+# All Rights Reserved.
+#
+# WARNING: This program as such is intended to be used by professional
+# programmers who take the whole responsibility of assessing all potential
+# consequences resulting from its eventual inadequacies and bugs
+# End users who are looking for a ready-to-use solution with commercial
+# guarantees and support are strongly adviced to contract a Free Software
+# Service Company
+#
+# This program is Free Software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public License
+# as published by the Free Software Foundation; either version 2.1
+# of the License, or (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
+#
+##############################################################################
+
+import ast
+import hashlib
+import json
+import re
+import requests
+import sys
+
+import prettytable
+
+from slapos.grid import networkcache
+from slapos.cli.config import ConfigCommand
+from slapos.cli.list import resetLogger
+
+class CacheLookupCommand(ConfigCommand):
+    """
+    perform a query to the networkcache
+    You can provide either a complete URL to the software release,
+    or a corresponding MD5 hash value.
+
+    The command will report which OS distribution/version have a binary
+    cache of the software release, and which ones are compatible
+    with the OS you are currently running.
+    """
+
+    def get_parser(self, prog_name):
+        ap = super(CacheLookupCommand, self).get_parser(prog_name)
+        ap.add_argument('url',
+                        help='Wanted url for testing')
+        return ap
+
+    def take_action(self, args):
+        configp = self.fetch_config(args)
+        cache_dir = configp.get('networkcache', 'download-binary-dir-url')
+        do_lookup(self.app.log, cache_dir, args.url)
+
+
+def do_lookup(logger, cache_dir, url):
+    md5 = hashlib.md5(url).hexdigest()
+
+    try:
+        cached_url = '%s/slapos-buildout-%s' % (cache_dir, md5)
+        logger.debug('Connecting to %s', url)
+        req = requests.get(cached_url, timeout=5)
+    except (requests.Timeout, requests.ConnectionError):
+        logger.critical('Cannot connect to cache server at %s', cached_url)
+        sys.exit(10)
+
+    if not req.ok:
+        if req.status_code == 404:
+            logger.critical('Object not in cache: %s', url)
+        else:
+            logger.critical('Error while looking object %s: %s', url, req.reason)
+        sys.exit(10)
+
+    entries = req.json()
+
+    if not entries:
+        logger.info('Object found in cache, but has no entries.')
+        return
+
+    import pdb;pdb.set_trace()
+
+    pt = prettytable.PrettyTable(['file', 'sha512'])
+
+    entry_list = sorted(json.loads(entry[0]) for entry in entries)
+
+    for entry in entry_list:
+        pt.add_row([entry["file"], entry["sha512"]])
+
+    meta = json.loads(entries[0][0])
+    logger.info('Software URL: %s', url)
+    logger.info('SHADIR URL: %s', cached_url)
+
+    resetLogger(logger)
+    for line in pt.get_string(border=True, padding_width=0, vrules=prettytable.NONE).split('\n'):
+        logger.info(line)