~hemanth-hm/rfc/rfc-finder

#!/usr/bin/env python

# -*- coding: utf-8 -*-

### BEGIN LICENSE

# Copyright (C) 2010 hemanth.hm hemanth.hm@gmail.com

# This program is free software: you can redistribute it and/or modify it 

# under the terms of the GNU General Public License version 3, as published 

# by the Free Software Foundation.

# 

# This program is distributed in the hope that it will be useful, but 

# WITHOUT ANY WARRANTY; without even the implied warranties of 

# MERCHANTABILITY, SATISFACTORY QUALITY, or FITNESS FOR A PARTICULAR 

# PURPOSE.  See the GNU General Public License for more details.

# 

# You should have received a copy of the GNU General Public License along 

# with this program.  If not, see <http://www.gnu.org/licenses/>.

### END LICENSE

###################### DO NOT TOUCH THIS (HEAD TO THE SECOND PART) ######################

import os

import sys

try:

    import DistUtilsExtra.auto

except ImportError:

    print >> sys.stderr, 'To build rfc-search you need https://launchpad.net/python-distutils-extra'

    sys.exit(1)

assert DistUtilsExtra.auto.__version__ >= '2.18', 'needs DistUtilsExtra.auto >= 2.18'

def update_data_path(prefix, oldvalue=None):

    try:

        fin = file('rfc_search/rfc_searchconfig.py', 'r')

        fout = file(fin.name + '.new', 'w')

        for line in fin:            

            fields = line.split(' = ') # Separate variable from value

            if fields[0] == '__rfc_search_data_directory__':

                # update to prefix, store oldvalue

                if not oldvalue:

                    oldvalue = fields[1]

                    line = "%s = '%s'\n" % (fields[0], prefix)

                else: # restore oldvalue

                    line = "%s = %s" % (fields[0], oldvalue)

            fout.write(line)

        fout.flush()

        fout.close()

        fin.close()

        os.rename(fout.name, fin.name)

    except (OSError, IOError), e:

        print ("ERROR: Can't find rfc_search/rfc_searchconfig.py")

        sys.exit(1)

    return oldvalue

def update_desktop_file(datadir):

    try:

        fin = file('rfc-search.desktop.in', 'r')

        fout = file(fin.name + '.new', 'w')

        for line in fin:            

            if 'Icon=' in line:

                line = "Icon=%s\n" % (datadir + 'media/icon.png')

            fout.write(line)

        fout.flush()

        fout.close()

        fin.close()

        os.rename(fout.name, fin.name)

    except (OSError, IOError), e:

        print ("ERROR: Can't find rfc-search.desktop.in")

        sys.exit(1)

class InstallAndUpdateDataDirectory(DistUtilsExtra.auto.install_auto):

    def run(self):

        previous_value = update_data_path(self.prefix + '/share/rfc-search/')

        update_desktop_file(self.prefix + '/share/rfc-search/')

        DistUtilsExtra.auto.install_auto.run(self)

        update_data_path(self.prefix, previous_value)

##################################################################################

###################### YOU SHOULD MODIFY ONLY WHAT IS BELOW ######################

##################################################################################

DistUtilsExtra.auto.setup(

    name='rfc-search',

    version='1.0',

    license='GPL-3',

    author='Hemanth',

    author_email='hemanth.hm@gmail.com',

    #description='UI for managing …',

    #long_description='Here a longer description',

    url='https://launchpad.net/rfc',

    cmdclass={'install': InstallAndUpdateDataDirectory}

    )