root/setup.py

#!/usr/bin/env python
# Copyright (C) 2006, University of California Regents
# All rights reserved
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
#  * Redistributions of source code must retain the above copyright notice,
#    this list of conditions and the following disclaimer.
#  * Redistributions in binary form must reproduce the above copyright notice,
#    this list of conditions and the following disclaimer in the documentation
#    and/or other materials provided with the distribution.
#  * Neither the name of the University of California, San Diego nor the
#    names of its contributors may be used to endorse or promote products
#    derived from this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
#
# AUTHORS:  Marvin McNett <mmcnett@cs.ucsd.edu>
#           Diwaker Gupta <dgupta@cs.ucsd.edu>
#           Roman Kurakin <rik@inse.ru>
#
# PROJECT URL: http://usher.ucsd.edu/

import sys
import os
import time

from distutils.core import setup
from distutils.command.install_data import install_data
from distutils.util import change_root, convert_path
from types import StringType


pyver = '%d.%d'%sys.version_info[0:2]

class smart_install_data(install_data):

    def initialize_options (self):
        self.install_dir = None
        self.outfiles = []
        self.root = None
        self.force = 0

        self.data_files = self.distribution.data_files
        self.script_args = self.distribution.script_args
        self.warn_dir = 1

    def run(self):
        # Use revers order for safe removal
        for i in range(len(self.data_files)-1, -1, -1):
            f = self.data_files[i]
            if type(f) is StringType:
                continue
            if len(f) <= 2:
               continue
            # Ok, we have additional value, do some magick according it.
            if f[2] != 'preserve':
                # Nope, we do not know it. Just ignore for now.
                # Should we scream about it?
                continue
            # Check a tuple with path to install to and a list of files
            dir = convert_path(f[0])
            if not os.path.isabs(dir):
                dir = os.path.join(self.install_dir, dir)
            elif self.root:
                dir = change_root(self.root, dir)

            if f[1] == []:
                # If there are no files listed, the user must be
                # trying to create an empty directory, so just ignore
                # it.
                # Should we scream in this case?
                continue

            # Check files one by one.
            # Use revers order for safe removal
            for j in range(len(f[1])-1, -1, -1):
                data=f[1][j]
                data = convert_path(data)
                if not os.path.isfile(data):
                    # Again skip dirs.
                    continue
                dst = os.path.join(dir, os.path.basename(data))
                if not os.path.exists(dst):
                    continue
                # just move it to dst_<time>
                os.rename(dst, dst+'_'+time.strftime('%Y-%m-%d_%H:%M:%S',
                    time.gmtime()))
            #    del f[1][j]
            #if len(f[1]) == 0:
            #    del self.data_files[i]

        return install_data.run(self) 


setup(name = 'usher',
      version = '0.3',
      description = 'Usher Virtual Machine Management System',
      author = 'Marvin McNett',
      author_email = 'mmcnett@cs.ucsd.edu',
      url = 'http://usher.ucsd.edu/',
      download_url = 'http://usher.ucsd.edu/UsherDownloads',
      license = 'New BSD',
      packages = ['usher', 'usher.ctrl', 'usher.cli', 'usher.lnm',
          'usher.utils', 'usher.plugins', 'usher.plugins.sample2'],
      cmdclass = {'install_data':smart_install_data},
      data_files = [ 
          ('/var/usher', []),
          ('/etc/init.d', ['initscripts/usherctrl', 'initscripts/usherlnm'],
              'preserve'),
          ('/etc/usher', ['configs/ctrl.config.orig',
              'configs/lnm.config.orig', 'configs/cli.config.orig']),
          ('/usr/lib/python%s/site-packages/usher/ctrl' % pyver,
              ['usher/ctrl/app.tac']),
          ('/usr/lib/python%s/site-packages/usher/lnm' % pyver,
              ['usher/lnm/app.tac'])
      ]
     )