support.py

   1 # Copyright (C) 2007, Thomas Leonard
   2 # See the README file for details, or visit http://0install.net.
   3
   4 import os, subprocess, shutil
   5 from zeroinstall import SafeException
   6 from zeroinstall.injector import model
   7 from logging import info
   8
   9 def check_call(*args, **kwargs):
  10         exitstatus = subprocess.call(*args, **kwargs)
  11         if exitstatus != 0:
  12                 raise SafeException("Command %s failed with exit code %d" % (' '.join(args), exitstatus))
  13
  14 def show_and_run(cmd, args):
  15         print "Executing: %s %s" % (cmd, ' '.join("[%s]" % x for x in args))
  16         check_call(['sh', '-c', cmd, '-'] + args)
  17
  18 def suggest_release_version(snapshot_version):
  19         """Given a snapshot version, suggest a suitable release version.
  20         >>> suggest_release_version('1.0-pre')
  21         '1.0'
  22         >>> suggest_release_version('0.9-post')
  23         '0.10'
  24         >>> suggest_release_version('3')
  25         Traceback (most recent call last):
  26                 ...
  27         SafeException: Version '3' is not a snapshot version (should end in -pre or -post)
  28         """
  29         version = model.parse_version(snapshot_version)
  30         mod = version[-1]
  31         if mod == 0:
  32                 raise SafeException("Version '%s' is not a snapshot version (should end in -pre or -post)" % snapshot_version)
  33         if mod > 0:
  34                 # -post, so increment the number
  35                 version[-2][-1] += 1
  36         version[-1] = 0 # Remove the modifier
  37         return model.format_version(version)
  38
  39 def publish(iface, **kwargs):
  40         args = [os.environ['0PUBLISH']]
  41         for k in kwargs:
  42                 value = kwargs[k]
  43                 if value is True:
  44                         args += ['--' + k.replace('_', '-')]
  45                 elif value is not None:
  46                         args += ['--' + k.replace('_', '-'), value]
  47         args.append(iface)
  48         info("Executing %s", args)
  49         check_call(args)
  50
  51 def get_singleton_impl(iface):
  52         impls = iface.implementations
  53         if len(impls) != 1:
  54                 raise SafeException("Local feed '%s' contains %d versions! I need exactly one!" % (iface.uri, len(impls)))
  55         return impls.values()[0]
  56
  57 def backup_if_exists(name):
  58         if not os.path.exists(name):
  59                 return
  60         backup = name + '~'
  61         if os.path.exists(backup):
  62                 print "(deleting old backup %s)" % backup
  63                 if os.path.isdir(backup):
  64                         shutil.rmtree(backup)
  65                 else:
  66                         os.unlink(backup)
  67         os.rename(name, backup)
  68         print "(renamed old %s as %s; will delete on next run)" % (name, backup)
  69
  70 def get_choice(*options):
  71         while True:
  72                 choice = raw_input('/'.join(options) + ': ').lower()
  73                 if not choice: continue
  74                 for o in options:
  75                         if o.lower().startswith(choice):
  76                                 return o
  77
  78 class Status(object):
  79         __slots__ = ['old_snapshot_version', 'release_version', 'head_before_release', 'new_snapshot_version', 'head_at_release', 'created_archive', 'tagged']
  80         def __init__(self):
  81                 for name in self.__slots__:
  82                         setattr(self, name, None)
  83
  84                 if os.path.isfile(release_status_file):
  85                         for line in file(release_status_file):
  86                                 assert line.endswith('\n')
  87                                 line = line[:-1]
  88                                 name, value = line.split('=')
  89                                 setattr(self, name, value)
  90                                 info("Loaded status %s=%s", name, value)
  91
  92         def save(self):
  93                 tmp_name = release_status_file + '.new'
  94                 tmp = file(tmp_name, 'w')
  95                 try:
  96                         lines = ["%s=%s\n" % (name, getattr(self, name)) for name in self.__slots__ if getattr(self, name)]
  97                         tmp.write(''.join(lines))
  98                         tmp.close()
  99                         os.rename(tmp_name, release_status_file)
 100                         info("Wrote status to %s", release_status_file)
 101                 except:
 102                         os.unlink(tmp_name)
 103                         raise
 104