setup.py

#!/usr/bin/env python
#
#   Copyright (C) 2021 Sean D'Epagnier
#
# This Program 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.  

import sys, time
import os, os.path

if sys.version_info[0] < 3:
    print('pypilot requires python version 3.  python version is', sys.version)
    exit(1)

if os.system('which apt'):
    print('system does not support apt, you can try running dependencies script and/or manually install needed packages')
else:
    if not os.path.exists('deps'):
        import dependencies

try:
    from setuptools import setup, Extension
    
except ImportError:
    print('ERROR, requires python setuptools package!')
    exit(0)
    #from distutils.core import setup, Extension

# because basically, a bug in swig extensions this is a workaround to ensure the python scripts
# are generated before they are scanned
from setuptools.command.install import install
class build_ext_first(install):
    def run(self):
        self.run_command('build_ext')
        return install.run(self)

linebuffer_module = Extension('pypilot/linebuffer/_linebuffer',
                        sources=['pypilot/linebuffer/linebuffer.cpp', 'pypilot/linebuffer/linebuffer.i'],
                        extra_compile_args=['-Wno-unused-result'],
                        swig_opts=['-c++']
)

arduino_servo_module = Extension('pypilot/arduino_servo/_arduino_servo',
                        sources=['pypilot/arduino_servo/arduino_servo.cpp', 'pypilot/arduino_servo/arduino_servo_eeprom.cpp', 'pypilot/arduino_servo/arduino_servo.i'],
                        extra_compile_args=['-Wno-unused-result'],
                        swig_opts=['-c++']
)

ugfx_defs = ['-DWIRINGPI']
try:
    import RPi.GPIO
    ugfx_libraries=['wiringPi']
except:
    try:
        import OPi.GPIO
        ugfx_libraries=['wiringPi']
    except:
        print('no RPi.GPIO library for ugfx')
        ugfx_libraries=[]
        ugfx_defs = []

ugfx_module = Extension('pypilot/hat/ugfx/_ugfx',
                        sources=['hat/ugfx/ugfx.cpp',
                                 'hat/ugfx/ugfx.i'],
                        extra_compile_args=['-Wno-unused-result'] + ugfx_defs,
                        libraries=ugfx_libraries,
                        swig_opts=['-c++'] + ugfx_defs)

if ugfx_libraries:
    spireader_module = Extension('pypilot/hat/spireader/_spireader',
                        sources=['hat/spireader/spireader.cpp',
                                 'hat/spireader/spireader.i'],
                        extra_compile_args=['-Wno-unused-result'],
                        libraries=ugfx_libraries,
                        swig_opts=['-c++'])

else:
    spireader_module = None

os.system('cd hat/locale;./translate.sh')
os.system('cd hat; pybabel compile -d translations')
os.system('cd pypilot/locale;./translate.sh')
os.system('cd web; pybabel compile -d translations')

def find_locales(name, dir = 'locale'):
    locale_files = []
    for walk in os.walk('./' + name + '/' + dir):
        path, dirs, files = walk
        path = path[len(name) + 3:]
        for file in files:
            if file[-3:] == '.mo':
                locale_files.append(os.path.join(path, file))

    return locale_files

from pypilot import version

packages = ['pypilot', 'pypilot/pilots', 'pypilot/arduino_servo', 'ui', 'hat', 'web', 'pypilot/linebuffer', 'hat/ugfx', 'hat/spireader']
try:
    from setuptools import find_packages
    packages = find_packages()
except:
    pass


# ensure all packages are under pypilot
package_dirs = {}
for package in list(packages):
    if not package.startswith('pypilot'):
        packages.remove(package)
        packages.append('pypilot.'+package)
        package_dirs['pypilot.'+package] = package.replace('.', '/')

package_data = {'pypilot': find_locales('pypilot'),
                'pypilot.hat': ['font.ttf', 'static/*', 'templates/*'] + find_locales('hat')  + find_locales('hat', 'translations'),
                'pypilot.ui': ['*.png', '*.mtl', '*.obj'],
                'pypilot.web': ['static/*', 'templates/*'] + ['pypilot_web.pot'] + find_locales('web', 'translations')}
        

ext_modules = [arduino_servo_module, linebuffer_module, ugfx_module]
if spireader_module:
    ext_modules.append(spireader_module)    

    
setup (name = 'pypilot',
       version = version.strversion,
       description = 'pypilot sailboat autopilot',
       license = 'GPLv3',
       author="Sean D'Epagnier",
       url='http://pypilot.org/',
       packages=packages,
       package_dir=package_dirs,
       ext_modules = ext_modules,
       package_data=package_data,
       cmdclass={'install': build_ext_first},
       entry_points={
           'console_scripts': [
               'pypilot=pypilot.autopilot:main',
               'pypilot_boatimu=pypilot.boatimu:main',
               'pypilot_servo=pypilot.servo:main',
               'pypilot_web=pypilot.web.web:main',
               'pypilot_hat=pypilot.hat.hat:main',
               'pypilot_control=pypilot.ui.autopilot_control:main',
               'pypilot_calibration=pypilot.ui.autopilot_calibration:main',
               'pypilot_client=pypilot.client:main',
               'pypilot_scope=pypilot.ui.scope_wx:main',
               'pypilot_client_wx=pypilot.ui.client_wx:main'
               ]
        }
       )