setup.py

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

try:
    from setuptools import setup
except ImportError:
    print "WARNING: Failing back to distutils"
    from distutils.core import setup
import sys
import os.path
import shutil
import glob
import zipfile

APP_NAME = 'metalinkc'
VERSION = '6.2'
LICENSE = 'GPL'
DESC = 'A metalink checker and download client.'
AUTHOR_NAME = 'Neil McNab'
EMAIL = 'nabber00@gmail.com'
URL = 'https://github.com/metalink-dev/pymetalink'

PYMETALINK_URL = "https://github.com/metalink-dev/pymetalink/releases/download/v6.2/pymetalink-6.2.zip"


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(APP_NAME.replace('-','_') + ".egg-info")
    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):
    start = os.path.dirname(__file__)
    if not start:
        start = '.'
    mylist = []
    for root, dirs, files in os.walk(start):
        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):], filename))
                    
    return mylist



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

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

elif sys.argv[1] == 'zip':
    #print "Zipping up..."
    create_zip("dist/", APP_NAME + "-" + VERSION + "-win32.zip")
    

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

    #localegen()
    #localecompile()
    
    setup(console = ["metalinkc.py"],
      zipfile = None,
      packages = ['metalinkc'],
      dependency_links=[PYMETALINK_URL],
      install_requires=['pymetalink'],        
      name = APP_NAME,
      version = VERSION,
      license = LICENSE,
      description = DESC,
      author = AUTHOR_NAME,
      author_email = EMAIL,
      url = URL
      )
    setup(windows = ["metalinkcw.py"],
      zipfile = None,
      packages = ['metalinkc'],
      dependency_links=[PYMETALINK_URL],
      install_requires=['pymetalink'],        
      name = APP_NAME,
      version = VERSION,
      license = LICENSE,
      description = DESC,
      author = AUTHOR_NAME,
      author_email = EMAIL,
      url = URL
      )    
      
else:      
    #scripts = rec_search(".py")

    #localegen()

    scripts = ['metalinkc.py', 'metalinkcw.py']
    setup(scripts = scripts,
      dependency_links=[PYMETALINK_URL],
      install_requires=['pymetalink'],
      packages = ['metalinkc'],
      #data_files = data,
      name = APP_NAME,
      version = VERSION,
      license = LICENSE,
      description = DESC,
      author = AUTHOR_NAME,
      author_email = EMAIL,
      url = URL
      )