setup.py

#!/usr/bin/python2

#   groupbutton.py
#
#	Copyright 2010 Matias Sars
#
#	DockbarX is free software: you can redistribute it and/or modify
#	it under the terms of the GNU General Public License as published by
#	the Free Software Foundation, either version 3 of the License, or
#	(at your option) any later version.
#
#	DockbarX 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 General Public License
#	along with dockbar.  If not, see <http://www.gnu.org/licenses/>.

from distutils.core import setup
from distutils.core import setup
from distutils import cmd
from distutils.command.install_data import install_data as _install_data
from distutils.command.build import build as _build

import msgfmt
import os
import sys

VERSION = "0.93"

class build_trans(cmd.Command):
    description = "Compile .po files into .mo files"
    def initialize_options(self):
        pass

    def finalize_options(self):
        pass

    def run(self):
        po_dict = {
                    "dockbarx": os.path.join(os.path.dirname(os.curdir), "po"),
                    "dockbarx-themes": os.path.join(os.path.dirname(os.curdir), "po-themes")
                  }
        for (mo_file, po_dir) in po_dict.items():
            for path, names, filenames in os.walk(po_dir):
                for f in filenames:
                    if f.endswith(".po"):
                        lang = f[:len(f) - 3]
                        src = os.path.join(path, f)
                        dest_path = os.path.join("build", "locale", lang, "LC_MESSAGES")
                        dest = os.path.join(dest_path, "%s.mo"%mo_file)
                        if not os.path.exists(dest_path):
                            os.makedirs(dest_path)
                        if not os.path.exists(dest):
                            print "Compiling %s for %s" % (src, mo_file)
                            msgfmt.make(src, dest)
                        else:
                            src_mtime = os.stat(src)[8]
                            dest_mtime = os.stat(dest)[8]
                            if src_mtime > dest_mtime:
                                print "Compiling %s for %s" % (src, mo_file)
                                msgfmt.make(src, dest)

class build(_build):
    sub_commands = _build.sub_commands + [("build_trans", None)]
    def run(self):
        _build.run(self)

class install_data(_install_data):

    def run(self):
        for lang in os.listdir("build/locale/"):
            lang_dir = os.path.join("/", "usr", "share",
                                    "locale", lang, "LC_MESSAGES")
            lang_files = []
            d_file = os.path.join("build", "locale", lang,
                                  "LC_MESSAGES", "dockbarx.mo")
            dt_file = os.path.join("build", "locale", lang,
                                   "LC_MESSAGES", "dockbarx-themes.mo")
            if os.path.exists(d_file):
                lang_files.append(d_file)
            if os.path.exists(dt_file):
                lang_files.append(dt_file)
            self.data_files.append( (lang_dir, lang_files) )
        # Scan folders for the right files
        self.scan_path("/usr/share/dockbarx/themes", "themes", ext=".tar.gz")
        self.scan_path("share/icons/", "icons", ext=".png")
        self.scan_path("/usr/share/namebar/themes",
                       "dockx_applets/namebar_themes",
                       ext=".tar.gz")
        self.scan_path("/usr/share/dockbarx/applets/vc-themes",
                       "dockx_applets/vc-themes")
        _install_data.run(self)

    def scan_path(self, install_path, base_path, path="", ext=""):
        files = []
        for f in os.listdir(os.path.join(base_path, path)):
            fpath = os.path.join(base_path, path, f)
            if os.path.isdir(fpath):
                self.scan_path(install_path, base_path,
                               os.path.join(path, f), ext)
            elif os.path.isfile(fpath) and fpath.endswith(ext):
                files.append(fpath)
        if files:
            self.data_files.append((os.path.join(install_path, path), files))


cmdclass = {
    "build": build,
    "build_trans": build_trans,
    "install_data": install_data,
}

data_files=[
            ("/usr/share/dockbarx/applets", ["dockx_applets/cardapio_dbx.py",
                                             "dockx_applets/cardapio.applet",
                                             "dockx_applets/clock.py",
                                             "dockx_applets/clock.applet",
                                             "dockx_applets/namebar.py",
                                             "dockx_applets/namebar.applet",
                                             "dockx_applets/appindicator.py",
                                             "dockx_applets/appindicator.applet",
                                             "dockx_applets/hello_world.py",
                                             "dockx_applets/hello_world.applet",
                                             "dockx_applets/battery_status.py",
                                             "dockx_applets/battery_status.applet",
                                             "dockx_applets/volume-control.py",
                                             "dockx_applets/volume-control.applet",
                                             "dockx_applets/volume-control.ui",
                                             "dockx_applets/namebar_window_buttons.applet",
                                             "dockx_applets/namebar_window_buttons.py",
                                             "dockx_applets/namebar_window_title.applet",
                                             "dockx_applets/namebar_window_title.py"]),
            ("/usr/bin", ["dockbarx_factory", "dbx_preference", "dockx"]),
            ("/usr/lib/bonobo/servers", ["GNOME_DockBarXApplet.server"]),
            ("/usr/share/applications/", ["dbx_preference.desktop"]),
            ("/usr/share/applications/", ["DockX.desktop"]),
         ]

setup(name="Dockbarx",
      version=VERSION,
      description="A dock-ish gnome-applet",
      author="Aleksey Shaferov and Matias Sars",
      url="http://launchpad.net/dockbar/",
      packages=["dockbarx"],
      data_files=data_files,
      cmdclass=cmdclass
     )



if len(sys.argv) == 2 and sys.argv[1] == "install":
    if os.path.exists("/usr/bin/dockbarx.py"):
        # Remove old dockbarx.py so that it isn't imported
        # instead of the package dockbarx when dockbarx is run.
        print
        print "There is a dockbarx.py in /usr/bin. " + \
              "This has to be removed to make DockbarX run correctly."
        remove = raw_input("Remove /usr/bin/dockbarx.py? (Y/n)")
        if remove == "" or remove[0].lower() == "y":
            os.remove("/usr/bin/dockbarx.py")
        else:
            print "/usr/bin/dockbarx.py is not removed. " + \
                  "Please remove it or rename it manually."

    if os.path.exists("/usr/bin/dockbarx_factory.py"):
        print
        print "There is a dockbarx_factory.py in /usr/bin. " + \
              "This file is no longer used."
        remove = raw_input("Remove /usr/bin/dockbarx_factory.py? (Y/n)")
        if remove == "" or remove[0].lower() == "y":
            os.remove("/usr/bin/dockbarx_factory.py")
            print "/usr/bin/dockbarx_factory.py is removed. "
        else:
            print "/usr/bin/dockbarx_factory.py is not removed. "

    if os.path.exists("/usr/bin/dbx_preference.py"):
        print
        print "There is a dbx_preference.py in /usr/bin. " + \
              "This file is no longer used."
        remove = raw_input("Remove /usr/bin/dbx_preference.py? (Y/n)")
        if remove == "" or remove[0].lower() == "y":
            os.remove("/usr/bin/dbx_preference.py")
            print "/usr/bin/dbx_preference.py is removed. "
        else:
            print "/usr/bin/dbx_preference.py is not removed. "