#!/usr/bin/env python3

# Copyright (C) 2016, the Pyzo development team

#

# Pyzo is distributed under the terms of the (new) BSD License.

# The full license can be found in 'license.txt'.

""" FREEZING Pyzo WITH CX_FREEZE

This script can be run as a script (no need to do distutils stuff...)

Pyzo is frozen in such a way that it still uses the plain source code.

This is achieved by putting the Pyzo package in a subdirectory called

"source". This source directory is added to sys.path by __main__.py.

For distribution:

* Write release notes

* Update __version__

* Build binaries for Windows, Linux and Mac

* Upload binaries to pyzo website

* Upload source to pypi

* Announce

* Add tag to released commit

* Incease version number to dev

"""

import sys, os, stat, shutil, struct

import subprocess

import cx_Freeze

from cx_Freeze import Executable, Freezer, setup

# Define app name and such

name = "pyzo"

baseDir = os.path.abspath('') + '/'

srcDir = baseDir + 'pyzo/'

distDir = baseDir + 'frozen/'

scriptFiles = [srcDir + '__main__.py']

iconFile = srcDir + 'resources/appicons/pyzologo.ico'

# On MAC, build an application bundle

if sys.platform=='darwin':

contentsDir=distDir+name+'.app/Contents/'

resourcesDir=contentsDir+'Resources/'

appDir=distDir

distDir=contentsDir+'MacOS/'

applicationBundle=True

createDmg=True # for dev

else:

applicationBundle=False

sys.path.append('')

## Includes and excludes

# We do not need these

excludes = ['_ssl', 'pyreadline', 'pdb',

]

# Excludes for tk

tk_excludes = [ "pywin", "pywin.debugger", "pywin.debugger.dbgcon",

"pywin.dialogs", "pywin.dialogs.list",

"Tkconstants", "Tkinter", "tcl" ]

excludes.extend(tk_excludes)

excludes.append('numpy')

excludes.append('PySide.QtNetwork')

excludes.append('PyQt4.QtNetwork')

# For qt to work

PyQtModules = ['PyQt4', 'PyQt4.QtCore', 'PyQt4.QtGui', 'PyQt4.QtHelp']

PySideModules = ['PySide', 'PySide.QtCore', 'PySide.QtGui', 'PySide.QtHelp']

#

# try:

# import PyQt4

# except ImportError:

# use_pyqt = False

# else:

# use_pyqt = True

use_pyqt = False

#

if use_pyqt: # and sys.platform == 'darwin':

excludes.extend(PySideModules)

includes = PyQtModules

else:

excludes.extend(PyQtModules)

includes = PySideModules

# # For Pyzolib dependencies

includes.extend(['pyzolib.paths', 'pyzolib.path', 'pyzolib.ssdf', 'pyzolib.interpreters'])

## Freeze

# Clear first

if os.path.isdir(baseDir+'frozen'):

shutil.rmtree(baseDir+'frozen')

os.makedirs(distDir)

executables = {}

for scriptFile in scriptFiles:

if sys.platform.startswith('win'):

ex = Executable( scriptFile,

targetName = 'pyzo.exe',

icon=iconFile,

appendScriptToExe = True,

base = 'Win32GUI', # this is what hides the console

)

else:

ex = Executable( scriptFile,

targetName = 'pyzo',

)

executables[ex] = True

f = Freezer( executables,

)

f.Freeze()

## Process source code and other resources

MANIFEST_TEMPLATE = """

""".lstrip().replace('\r\n', '\n').replace('\n', '\r\n')

def install_c_runtime(targetDir):

open(manifest_filename, 'wb').write(t.encode('utf-8'))

# taken from pyzo_build:

def copydir_smart(path1, path2):

return count

# Install MS C runtime, cx_freeze does not seem to find mscvp100.dll

if sys.platform.startswith('win'):

install_c_runtime(distDir)

# Copy the whole Pyzo package

copydir_smart(os.path.join(srcDir), os.path.join(distDir, 'source', 'pyzo'))

# Remove dummy executable

tmp1 = os.path.join(distDir,'pyzo_.exe')

tmp2 = os.path.join(distDir,'pyzo_')

for fname in [tmp1, tmp2]:

if os.path.isfile(fname):

os.remove(fname)

# Create settings folder and put in a file

os.mkdir(os.path.join(distDir, '_settings'))

SETTINGS_TEXT = """

""".lstrip()

with open(os.path.join(distDir, '_settings', 'README.txt'), 'wb') as file:

file.write(SETTINGS_TEXT.encode('utf-8'))

## Post processing

# Set search path of dynamic libraries

from pyzolib import dllutils

if sys.platform.startswith('linux'):

# Exe

dllutils.set_search_path(os.path.join(distDir, 'pyzo'), '', 'lib')

# Libs

os.mkdir(os.path.join(distDir, 'lib'))

for entry in os.listdir(distDir):

filename = os.path.join(distDir, entry)

if not os.path.isfile(filename):

continue

if not (entry.endswith('.so') or '.so.' in entry):

continue

#

rpaths = '', 'lib'

if entry.startswith('lib') and not 'python' in entry:

filename = os.path.join(distDir, 'lib', entry)

shutil.move(os.path.join(distDir, entry), filename)

rpaths = '', '..'

try:

dllutils.set_search_path(filename, *rpaths)

except Exception as err:

print('Cannot set search path of %s:\n%s' %

(os.path.basename(entry), str(err)))

if sys.platform.startswith('linux'):

# Set qt.conf

# Prevent loading plugins form the system plugin dir, which

# may cause incompatibility conflicts. This complements the call

# QApplication.setLibraryPaths([]), it does not replace it.

# See issue 138 and issue 198.

with open(os.path.join(distDir, 'qt.conf'), 'wb') as file:

import pyzolib.qt

file.write(pyzolib.qt.DEFAULT_QT_CONF_TEXT.encode('utf-8'))

#file.write("[Paths]\nPlugins = '.'\n".encode('utf-8'))

# Write about experimental feature

fname = 'CHECK QT.CONF TO USE NATIVE STYLING'

with open(os.path.join(distDir, fname), 'wb') as file:

pass

# Remove imageforma dir. These libs hook into the original

# Qt libs, giving rise to these nasty mixed binaries errors.

# This happened on OSX, but Linux may also be affected

imformatsdir = os.path.join(distDir, 'imageformats')

if os.path.isdir(imformatsdir):

shutil.rmtree(imformatsdir)

# todo: this is now in cx_Freeze right?

if applicationBundle:

#Change the absolute paths in all library files to relative paths

#This should be a cx_freeze task, but cx_freeze doesn't do it

# Move PyQt4 libs into subdir

# This gets us a similar dir structure as installed so

# we dont have to fix paths ...

if True: # IF FREEZING FROM CONDA

os.makedirs(distDir+'source/more/PyQt4')

open(distDir+'source/more/PyQt4/__init__.py', 'wb').close()

for fname in os.listdir(distDir):

if fname.startswith('PyQt4'):

filename = distDir+'source/more/PyQt4/' + fname[6:]

elif fname.startswith('sip'):

filename = distDir+'source/more/' + fname

else:

continue

shutil.move(distDir+fname, filename)

shippedfiles=os.listdir(distDir)

for file in shippedfiles:

#Do the processing for any found file or dir, the tools will just

#fail for files for which it does not apply

filepath=os.path.join(distDir,file)

#Ensure write permissions

mode = os.stat(filepath).st_mode

if not (mode & stat.S_IWUSR):

os.chmod(filepath, mode | stat.S_IWUSR)

#Let the library itself know its place

subprocess.call(('install_name_tool','-id','@executable_path/'+file,filepath))

#Find the references

otool=subprocess.Popen(('otool','-L', filepath),stdout=subprocess.PIPE)

libs=otool.stdout.readlines()

for lib in libs:

#For each referenced library, chech if it is in the set of

#files that we ship. If so, change the reference to a path

#relative to the executable path

lib=lib.decode()

filename,_,_=lib.strip().partition(' ')

prefix,name=os.path.split(filename)

if name in shippedfiles:

newfilename='@executable_path/'+name

print ('%s => %s' % (name,newfilename))

subprocess.call(('install_name_tool','-change',filename,newfilename,filepath))

#Copy the icons

if not os.path.isdir(resourcesDir):

os.mkdir(resourcesDir)

shutil.copy(srcDir+'resources/appicons/pyzologo.icns',resourcesDir+'pyzo.icns')

shutil.copy(srcDir+'resources/appicons/py.icns',resourcesDir+'py.icns')

#Write qt.conf in the Resources dir

with open(os.path.join(resourcesDir, 'qt.conf'), 'wb') as file:

import pyzolib.qt

file.write(pyzolib.qt.DEFAULT_QT_CONF_TEXT.encode('utf-8'))

#Copy the Info.plist file

shutil.copy(baseDir+'Info.plist',contentsDir+'Info.plist')

#Copy the qt_menu.nib directory (TODO: is this the place to look for it?)

shutil.copytree('/opt/local/Library/Frameworks/QtGui.framework/Versions/4/Resources/qt_menu.nib',resourcesDir+'qt_menu.nib')

#Package in a dmg

dmgFile=appDir+'pyzo.dmg'

# Create the dmg

if createDmg:

if os.spawnlp(os.P_WAIT,'hdiutil','hdiutil','create','-fs','HFSX',

'-format','UDZO',dmgFile, '-imagekey', 'zlib-level=9',

'-srcfolder',appDir,'-volname', 'pyzo')!=0:

raise OSError('creation of the dmg failed')