Metalinks Code

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

from distutils.core import setup
import sys
import os.path
import shutil
import glob
import zipfile

APP_NAME = 'GGet'
VERSION = '0.0.4.0'
LICENSE = 'GPL'
DESC = ''
AUTHOR_NAME = 'Neil McNab'
EMAIL = 'webmaster@nabber.org'
URL = 'http://www.nabber.org/projects/metalink'

def clean():
    ignore = []
    
    filelist = []
    #filelist.extend(glob.glob("*metalink*.txt"))
    filelist.extend(rec_search(".exe"))
    filelist.extend(rec_search(".zip"))
    filelist.extend(rec_search(".pyc"))
    filelist.extend(rec_search(".pyo"))
    filelist.extend(rec_search(".mo"))
    filelist.extend(rec_search(".pot"))
    
    try:
        shutil.rmtree("build")
    except: pass
    try:
        shutil.rmtree("dist")
    except: pass
    # try:
        # shutil.rmtree("buildMetalink")
    # except: pass
    # try:
        # shutil.rmtree("buildmetalinkw")
    # except: pass
    
    try:
        shutil.rmtree("tests_temp")
    except: pass
    
    for filename in filelist:
        if filename not in ignore:
            try:
                os.remove(filename)
            except: pass

def create_zip(rootpath, zipname, mode="w"):
    print zipname
    myzip = zipfile.ZipFile(zipname, mode, zipfile.ZIP_DEFLATED)
    for root, dirs, files in os.walk(rootpath):
        for filename in files:
            filepath = os.path.join(root, filename)
            filehandle = open(filepath, "rb")
            filepath = filepath[len(rootpath):]
            text = filehandle.read()
            #print filepath, len(text)
            myzip.writestr(filepath, text)
            filehandle.close()
    myzip.close()

def localegen():
    localedir = "locale"
    ignore = ("setup.py", "test.py")

    files = rec_search(".py")
    for pyfile in files:
        if os.path.basename(pyfile) not in ignore:
            potdir = os.path.join(os.path.dirname(pyfile), localedir, os.path.basename(pyfile))[:-3]
            print potdir
            try:
                os.makedirs(os.path.join(os.path.dirname(pyfile), localedir))
            except: pass

            command = os.path.join(sys.prefix, "Tools/i18n/pygettext.py") + " --no-location -d \"%s\" \"%s\"" % (potdir, pyfile)
            print(command)
            result = os.system(command)
            if result != 0:
                raise AssertionError, "Generation of .pot file failed for %s." % pyfile


def localecompile():
    files = rec_search(".po")

    for pofile in files:
        command = os.path.join(sys.prefix, "Tools/i18n/msgfmt.py") + " \"%s\"" % pofile
        print(command)
        result = os.system(command)
        if result != 0:
            raise AssertionError, "Generation of .mo file failed for %s." % pofile
            
def rec_search(end, abspath = True, ignoredirs = []):
    start = os.path.abspath(os.path.dirname(__file__))
    mylist = []
    for root, dirs, files in os.walk(start):
        if not check_ignore(root[len(start):].strip("\\"), ignoredirs):
            for filename in files:
                if filename.endswith(end):
                    if abspath:
                        mylist.append(os.path.join(root, filename))
                    else:
                        mylist.append(os.path.join(root[len(start):].strip("\\"), filename))
                    
    return mylist
    
def check_ignore(root, ignoredirs):
    for dirname in ignoredirs:
        if root.startswith(dirname):
            return True
    return False
    
def copy_directory(source, target):
    if not os.path.exists(target):
        os.mkdir(target)
    for root, dirs, files in os.walk(source):
        if '.svn' in dirs:
            dirs.remove('.svn')  # don't visit .svn directories           
        for file in files:
            from_ = os.path.join(root, file)           
            to_ = from_.replace(source, target, 1)
            to_directory = os.path.split(to_)[0]
            if not os.path.exists(to_directory):
                os.makedirs(to_directory)
            shutil.copyfile(from_, to_)

if sys.argv[1] == 'sdist':
    scripts = rec_search(".py")

    localegen()

    setup(scripts = scripts,
	#packages = packages,
      #data_files = data,
      name = APP_NAME,
      version = VERSION,
      license = LICENSE,
      description = DESC,
      author = AUTHOR_NAME,
      author_email = EMAIL,
      url = URL
      )

elif sys.argv[1] == 'translate':
    localegen()
    localecompile()

elif sys.argv[1] == 'clean':
    clean()

elif sys.argv[1] == 'py2exe':
    import py2exe

    # DLL files to exclude from distribution
    dlllist = [ "DNSAPI.DLL", "MSIMG32.DLL", "NSI.DLL", "USP10.DLL"]
    
    #localegen()
    #localecompile()
    
    skipdirs = ["build", "dist", "gtk"]
    
    temp = ["AUTHORS"]
    temp.extend(rec_search(".ini", False, skipdirs))
    temp.extend(rec_search(".txt", False, skipdirs))
    temp.extend(rec_search(".mo", False, skipdirs))
    temp.extend(rec_search(".qm", False, skipdirs))
    temp.extend(rec_search(".png", False, skipdirs))
    temp.extend(rec_search(".svg", False, skipdirs))
    temp.extend(rec_search(".glade", False, skipdirs))
    temp.extend(rec_search(".in", False, skipdirs))

    data = []
    for item in temp:
        data.append((os.path.dirname(item), [item]))
    
    setup(windows = ["gget.py"],
      zipfile = None,
      data_files = data,
      options={"py2exe" : {'dll_excludes': dlllist, "packages": 'encodings', "includes" : "cairo, pango, pangocairo, atk, gobject, gio", "optimize": 2}},
      name = APP_NAME,
      version = VERSION,
      license = LICENSE,
      description = DESC,
      author = AUTHOR_NAME,
      author_email = EMAIL,
      url = URL
      )
        
    copyfiles = [os.path.join(sys.exec_prefix,"Lib\\site-packages\\gtk-2.0\\runtime\\bin\\libxml2-2.dll")]
    for filename in copyfiles:
        shutil.copy(filename, "dist")
        
elif sys.argv[1] == 'zip':
    #print "Zipping up..."
    create_zip("dist/", APP_NAME + "-" + VERSION + "-win32.zip")