heck
5 years ago
2 changed files with 289 additions and 4 deletions
@ -0,0 +1,285 @@ |
|||
# -*- coding: utf-8 -*- |
|||
|
|||
# This file is under GNU Affero General Public License 3.0 |
|||
# see LICENSE.txt |
|||
|
|||
from __future__ import print_function |
|||
|
|||
import sys |
|||
from sys import argv |
|||
|
|||
import os |
|||
from os import environ, uname |
|||
from os.path import dirname, exists, join |
|||
|
|||
from setuptools import setup, Extension |
|||
from glob import glob |
|||
import sysconfig |
|||
|
|||
if sys.platform == 'winnt': |
|||
if sys.version_info[0] >= 3: |
|||
import winreg |
|||
else: |
|||
import _winreg as winreg |
|||
|
|||
|
|||
from setuptools.command.build_ext import build_ext |
|||
|
|||
|
|||
if sys.version_info[0] < 3: |
|||
FileNotFoundError = EnvironmentError |
|||
|
|||
def find(fname, pathlist): |
|||
for path in pathlist: |
|||
cand_fname = join(path, fname) |
|||
if exists(cand_fname): |
|||
return path |
|||
raise FileNotFoundError(fname) |
|||
|
|||
def append_once(lst, element): |
|||
if element not in lst: |
|||
lst.append(element) |
|||
|
|||
def extend_once(lst, elements): |
|||
for element in elements: |
|||
if element not in lst: |
|||
lst.append(element) |
|||
|
|||
def getPythonLibver(): |
|||
g = sysconfig.get_config_vars() |
|||
sent = [] |
|||
for template in ( |
|||
'{cmd}{py_version_nodot}{abiflags}', |
|||
'{cmd}{py_version_nodot}', |
|||
'{cmd}{py_version_major}', |
|||
'{cmd}{py_version_major}{abiflags}', |
|||
'{cmd}' |
|||
): |
|||
val = template.format( |
|||
cmd = g['PYTHON'], |
|||
py_version_nodot = g['py_version_nodot'], |
|||
py_version_major = sys.version_info[0], |
|||
abiflags = g.get('ABIFLAGS', ''), |
|||
) |
|||
if val not in sent: |
|||
yield val |
|||
sent.append(val) |
|||
|
|||
|
|||
class BuildExtCommand(build_ext): |
|||
|
|||
# default_pyver = next( getPythonLibver() ) |
|||
user_options = build_ext.user_options + [ |
|||
# ('boost-python=', None, 'Boost Python version to use (e.g. \'python34\')'), |
|||
('local', None, 'Use local pEp install in HOME/USERPROFILE'), |
|||
('with-pEp-engine=', None, 'Path to pEp Engine source'), |
|||
('with-pEp-libadapter=', None, 'Path to pEp C++ Adapter Library source'), |
|||
('with-boost=', None, 'Path to Boost install prefix'), |
|||
('with-asn1c-share=', None, 'Path to installed ASN1C share directory'), |
|||
] |
|||
|
|||
def windowsGetInstallLocation(): |
|||
# Note: should be installed to 'C:\Program Files (x86)' while a 32-bit distro |
|||
# TODO: Try desktop adapter location first, then COM server |
|||
# FIXME: This is wrong, we should chase the COM server, not the Outlook Plugin (even if they're in the same place) |
|||
REG_PATH = "Software\\Microsoft\\Office\\Outlook\\Addins\\pEp" |
|||
regKey = None |
|||
try: |
|||
regKey = winreg.OpenKey( |
|||
winreg.HKEY_LOCAL_MACHINE, REG_PATH, 0, winreg.KEY_READ) |
|||
# Keys: Description, FileName, FriendlyName, LoadBehavior |
|||
com_server, regtype = winreg.QueryValueEx(regKey, 'FileName') |
|||
winreg.CloseKey(regKey) |
|||
except WindowsError: |
|||
com_server = None |
|||
finally: |
|||
if winreg: |
|||
winreg.CloseKey(regKey) |
|||
# <install-base>\\bin\\COM_Server.exe |
|||
dirname = os.path.dirname |
|||
return dirname( dirname( com_server ) ) |
|||
|
|||
def initialize_options(self): |
|||
build_ext.initialize_options(self) |
|||
# self.boost_python = BuildExtCommand.default_pyver |
|||
self.local = None != environ.get('PER_USER_DIRECTORY') |
|||
self.with_pEp_engine = None |
|||
self.with_pEp_libadapter = None |
|||
self.with_boost = None |
|||
self.with_asn1c_share = None |
|||
|
|||
def finalize_options(self): |
|||
build_ext.finalize_options(self) |
|||
|
|||
# normalize paths given by user (expand tilde etc.) |
|||
for with_option in ('pEp_engine', 'pEp_libadapter', 'boost', 'asn1c_share'): |
|||
w_opt = 'with_' + with_option |
|||
w_val = getattr(self, w_opt) |
|||
if w_val != None: |
|||
setattr(self, w_opt, os.path.normpath(w_val)) |
|||
|
|||
if sys.platform == 'darwin': |
|||
HOME = environ.get('PER_USER_DIRECTORY') or environ.get('HOME') |
|||
PEPLIBNAME = 'libpEpEngine.dylib' |
|||
LIBPEPA = 'libpEpAdapter.a' |
|||
BOOSTLIBNAME = 'libboost_{boost_python:s}-mt.dylib' |
|||
SYS_INCLUDES = [ |
|||
'/opt/local/include', # we prefer MacPorts over Homebrew (but support both) |
|||
'/usr/local/include', |
|||
'/Library/Frameworks/PrettyEasyPrivacy.framework/Versions/A/include', |
|||
join(environ["HOME"], 'include'), |
|||
'/usr/include', |
|||
] |
|||
SYS_SHARES = [ |
|||
'/opt/local/share', |
|||
'/usr/local/share', |
|||
'/Library/Frameworks/PrettyEasyPrivacy.framework/Versions/A/share', |
|||
join(environ["HOME"], 'share'), |
|||
'/usr/share', |
|||
] |
|||
SYS_LIB_PREFIXES = [ |
|||
'/opt/local/lib', |
|||
'/usr/local/lib', |
|||
'/Library/Frameworks/PrettyEasyPrivacy.framework/Versions/A/lib', |
|||
join(environ["HOME"], 'lib'), |
|||
'/usr/lib', |
|||
] |
|||
elif sys.platform == 'winnt': |
|||
HOME = environ.get('PER_USER_DIRECTORY') or environ.get('USERPROFILE') |
|||
SYS_ROOT = environ.get('SystemRoot') |
|||
PROFILE_ROOT = environ.get('AppData') |
|||
LOCAL_ROOT = environ.get('LocalAppData') |
|||
INST_PREFIX = windowsGetInstallLocation() |
|||
PEPLIBNAME = 'pEpEngine.dll' |
|||
LIBPEPA = 'libpEpAdapter.a' |
|||
BOOSTLIBNAME = 'boost_{boost_python:s}-mt.dll' |
|||
SYS_INCLUDES = [ |
|||
join(INST_PREFIX, 'include'), |
|||
join(PROFILE_ROOT, 'pEp', 'include'), |
|||
join(LOCAL_ROOT, 'pEp', 'include'), |
|||
join(SYS_ROOT, 'pEp', 'include'), |
|||
] |
|||
SYS_SHARES = [ |
|||
join(INST_PREFIX, 'share'), |
|||
join(PROFILE_ROOT, 'pEp', 'share'), |
|||
join(LOCAL_ROOT, 'pEp', 'share'), |
|||
join(SYS_ROOT, 'pEp', 'share'), |
|||
] |
|||
SYS_LIB_PREFIXES = [ |
|||
join(INST_PREFIX, 'bin'), |
|||
join(PROFILE_ROOT, 'pEp', 'bin'), |
|||
join(LOCAL_ROOT, 'pEp', 'bin'), |
|||
join(SYS_ROOT, 'pEp', 'bin'), |
|||
] |
|||
else: |
|||
HOME = environ.get('PER_USER_DIRECTORY') or environ.get('HOME') |
|||
PEPLIBNAME = 'libpEpEngine.so' |
|||
LIBPEPA = 'libpEpAdapter.a' |
|||
BOOSTLIBNAME = 'libboost_python37.so' |
|||
SYS_INCLUDES = ['/usr/local/pEp/include', '/usr/local/include', '/usr/include'] |
|||
SYS_SHARES = ['/usr/local/pEp/share', '/usr/local/share', '/usr/share'] |
|||
SYS_LIB_PREFIXES = ['/usr/local/pEp/bin', '/usr/local/bin', '/usr/bin', '/usr/lib/x86_64-linux-gnu/'] |
|||
|
|||
use_local_incl = (self.local or os.path.isfile( |
|||
join(HOME, 'include', 'pEp', 'pEpEngine.h')) ) |
|||
use_local_lib = (self.local or os.path.isfile( |
|||
join(HOME, 'lib', PEPLIBNAME)) ) |
|||
|
|||
INCLUDES = [ join(HOME, 'include') ] if use_local_incl else [] |
|||
INCLUDES.extend(SYS_INCLUDES) |
|||
SHARES = [ join(HOME, 'share') ] if use_local_incl else [] |
|||
SHARES.extend(SYS_SHARES) |
|||
LIBS = [ join(HOME, 'lib') ] if use_local_lib else [] |
|||
LIBS.extend(SYS_LIB_PREFIXES) |
|||
|
|||
if not self.with_pEp_engine: |
|||
ENGINE_INC = find( join('pEp', 'pEpEngine.h'), INCLUDES ) |
|||
ENGINE_LIB = find( PEPLIBNAME, LIBS ) |
|||
else: |
|||
if exists( join(self.with_pEp_engine, 'include', 'pEp') ): |
|||
ENGINE_INC = find( join('pEp', 'pEpEngine.h'), (join(self.with_pEp_engine, 'include'),) ) |
|||
ENGINE_LIB = find( PEPLIBNAME, (join(self.with_pEp_engine, 'lib'),) ) |
|||
else: |
|||
ENGINE_INC = find( 'pEpEngine.h', (join(self.with_pEp_engine, 'src'),) ) |
|||
ENGINE_LIB = find( PEPLIBNAME, (join(self.with_pEp_engine, 'src'),) ) |
|||
|
|||
if not self.with_pEp_libadapter: |
|||
LIBPEPA_INC = find( join('pEp', 'Adapter.hh'), INCLUDES ) |
|||
LIBPEPA_LIB = find( LIBPEPA, LIBS ) |
|||
else: |
|||
if exists( join(self.with_pEp_libadapter, 'include', 'pEp') ): |
|||
LIBPEPA_INC = find( join('pEp', 'Adapter.hh'), (join(self.with_pEp_libadapter, 'include'),) ) |
|||
LIBPEPA_LIB = find( LIBPEPA, (join(self.with_pEp_libadapter, 'lib'),) ) |
|||
else: |
|||
LIBPEPA_INC = find( 'Adapter.hh', (join(self.with_pEp_libadapter, 'src'),) ) |
|||
LIBPEPA_LIB = find( LIBPEPA, (join(self.with_pEp_libadapter, 'src'),) ) |
|||
|
|||
if not self.with_boost: |
|||
BOOST_INC = find( join('boost', 'system', 'config.hpp'), INCLUDES ) |
|||
BOOST_LIB = None |
|||
e_ = None |
|||
for py in getPythonLibver(): |
|||
f = BOOSTLIBNAME.format(boost_python=py) |
|||
try: |
|||
BOOST_LIB = find( f, LIBS ) |
|||
break |
|||
except Exception as e: |
|||
if self.verbose: |
|||
print("Not found: " + f) |
|||
e_ = e if e_ is None else e_ |
|||
if BOOST_LIB is None: |
|||
raise e_ |
|||
else: |
|||
raise NotImplementedError("Building from Boost source not implemented yet") # FIXME |
|||
# BOOST_INC = find( join('boost', 'system', 'config.hpp'), (self.with_boost,) ) |
|||
# BOOST_LIB = find( BOOSTLIBNAME, (self.with_boost,) ) |
|||
|
|||
if not self.with_asn1c_share: |
|||
ASN1C_INC = find( join('asn1c', 'asn_system.h'), SHARES ) |
|||
else: |
|||
ASN1C_INC = find( join('asn1c', 'asn_system.h'), (self.with_asn1c_share,) ) |
|||
|
|||
if self.verbose: |
|||
print('ENGINE_INC=%s' % ENGINE_INC) |
|||
print('ENGINE_LIB=%s' % ENGINE_LIB) |
|||
print('LIBPEPA_INC=%s' % LIBPEPA_INC) |
|||
print('LIBPEPA_LIB=%s' % LIBPEPA_LIB) |
|||
print('BOOST_INC=%s' % BOOST_INC) |
|||
print('BOOST_LIB=%s' % BOOST_LIB) |
|||
print('ASN1C_INC=%s' % ASN1C_INC) |
|||
|
|||
global module_pEp |
|||
extend_once( module_pEp.include_dirs, [ENGINE_INC, LIBPEPA_INC, BOOST_INC, ASN1C_INC] ) |
|||
extend_once( module_pEp.library_dirs, [ENGINE_LIB, LIBPEPA_LIB, BOOST_LIB] ) |
|||
extend_once( module_pEp.libraries, ['pEpEngine', 'boost_python3', 'boost_locale'] ) |
|||
|
|||
if self.debug: |
|||
module_pEp.extra_compile_args = ['-O0', '-g', '-UNDEBUG', '-std=c++14'] |
|||
else: |
|||
module_pEp.extra_compile_args = ['-std=c++14'] # FIXME |
|||
|
|||
def run(self): |
|||
build_ext.run(self) |
|||
|
|||
|
|||
# module_pEp global is referenced in BuildExtCommand |
|||
module_pEp = Extension('pEp', |
|||
sources = glob('src/*.cc'), |
|||
libraries = ['pEpEngine'], |
|||
# extra_compile_args = compile_args, |
|||
# include_dirs = [ENGINE_INC, BOOST_INC, ASN1C_INC], |
|||
# library_dirs = [ENGINE_LIB, BOOST_LIB], |
|||
) |
|||
|
|||
|
|||
setup( |
|||
name='pEp', |
|||
version='2.0', |
|||
description='p≡p for Python', |
|||
author="Volker Birk", |
|||
author_email="vb@pep-project.org", |
|||
ext_modules=[module_pEp,], |
|||
cmdclass={ |
|||
'build_ext': BuildExtCommand, |
|||
}, |
|||
) |
|||
Loading…
Reference in new issue