setup.py

   1 '''e.g. python setup.py bdist_apps --cores=1 --no-windows=1'''
   2
   3
   4 from ya2.utils.log import LogManager
   5 LogManager.before_init_setup('pmachines')
   6 import sys, os
   7 from sys import argv
   8 from setuptools import setup
   9 from subprocess import Popen
  10 from setuptools import Command
  11 from setuptools.command.develop import develop
  12 from multiprocessing import cpu_count
  13 from textwrap import dedent
  14 from direct.dist.commands import bdist_apps
  15 from p3d_appimage import AppImageBuilder
  16 from ya2.build.build import _branch, find_file_names, _version, FindFileNamesArgs
  17 from ya2.build.lang import LanguageBuilder
  18 from ya2.build.screenshots import ScreenshotsBuilder
  19 from ya2.build.images import ImagesBuilder
  20 from ya2.build.models import ModelsBuilder
  21 from pmachines.application.application import Pmachines
  22
  23
  24 app_name = long_name = 'pmachines'
  25
  26
  27 class SetupDevelopmentCommand(develop):
  28
  29     def run(self):
  30         super().run(self)
  31         for argument in ['lang', 'models', 'images']:
  32             p = Popen([sys.executable, __file__, argument])
  33             p.communicate()
  34
  35
  36 class BaseCommand(Command):
  37
  38     user_options = [('cores=', None, '#cores')]
  39     cores = cpu_count()
  40
  41     def initialize_options(self):
  42         BaseCommand.cores = cpu_count()
  43         for a in argv[:]: self.__process_argument(a)
  44
  45     def __process_argument(self, argument):
  46         if argument.startswith('--cores='):
  47             BaseCommand.cores = int(argument.split('=')[1])
  48
  49     def finalize_options(self): pass  # must override
  50
  51
  52 class ModelsCommand(BaseCommand):
  53
  54     def run(self):
  55         m = ModelsBuilder('assets/models', int(BaseCommand.cores))
  56         m.build()
  57
  58
  59 class ImagesCommand(BaseCommand):
  60
  61     def run(self):
  62         s = ScreenshotsBuilder(Pmachines.scenes())
  63         s.build()
  64         find_info = FindFileNamesArgs(
  65             ['jpg', 'png'],
  66             ['models', 'gltf', 'bam', 'build'],
  67             ['_png.png'])
  68         images = find_file_names(find_info)
  69         i = ImagesBuilder(images, int(BaseCommand.cores))
  70         i.build()
  71
  72
  73 class L10nCommand(BaseCommand):
  74
  75     def run(self):
  76         l = LanguageBuilder(
  77             app_name, 'assets/locale/po/', 'assets/scenes/', 'assets/locale/')
  78         l.build()
  79         l = LanguageBuilder(
  80             'test', 'tests/assets/locale/po/', 'tests/assets/scenes/', 'tests/assets/locale/')
  81         l.build()
  82
  83
  84 class BDistAppsCommand(bdist_apps):
  85
  86     user_options = bdist_apps.user_options + [
  87         ('cores', None, '#cores'),
  88         ('nowindows=', None, "don't build for windows"),  # letters, numbers and hypens only
  89         ('nolinux=', None, "don't build for linux")]
  90
  91     def initialize_options(self):
  92         super().initialize_options()
  93         self.nowindows, self.nolinux = None, None
  94         for a in argv[:]: self.__process_argument(a)
  95
  96     def __process_argument(self, argument):
  97         if argument.startswith('--no-windows='):
  98             self.nowindows = int(argument.split('=')[1])
  99         if argument.startswith('--no-linux='):
 100             self.nolinux = int(argument.split('=')[1])
 101
 102     def run(self):
 103         assets_creation_message = dedent('''NOTE: please be sure that you've already created the assets with:
 104             * python setup.py images models lang''')
 105         print(assets_creation_message)
 106         self.__patch_commands_py()
 107         super().run()
 108         self.__eventually_build_appimage()
 109
 110     def __patch_commands_py(self):
 111         command = 'patch --forward ' + \
 112               '../venv/lib/python3.7/site-packages/direct/dist/commands.py' + \
 113               ' ya2/build/commands.patch'
 114         os.system(command)
 115
 116     def __eventually_build_appimage(self):
 117         if not self.nolinux:
 118             a = AppImageBuilder(self)
 119             filename_branch = {'master': 'alpha', 'rc': 'rc', 'stable': ''}
 120             branch = filename_branch[_branch()]
 121             a.build(long_name, branch, 'https://www.ya2.it/downloads/')
 122
 123
 124 def _build_setup_arguments():
 125     d = {
 126         'name': app_name,
 127         'version': _version(),
 128         'packages': ['pmachines', 'ya2', 'assets', 'licenses'],
 129         'cmdclass': {
 130             'develop': SetupDevelopmentCommand,
 131             'models': ModelsCommand,
 132             'images': ImagesCommand,
 133             'lang': L10nCommand,
 134             'bdist_apps': BDistAppsCommand},
 135         'install_requires': ['panda3d'],
 136         'options': {
 137             'build_apps': {
 138                 'exclude_patterns': [
 139                     'build/*', 'built/*', 'setup.py', 'requirements.txt',
 140                     'venv/*', '.git*', '*.pyc', 'options.ini', '__pycache__',
 141                     'assets/models/gltf/*', 'assets/models/**/*.blend',
 142                     'assets/models/**/models/*.png',
 143                     'assets/models/**/models/*.jpg'],
 144                 'log_filename_strftime': True,
 145                 'log_filename': '$USER_APPDATA/pmachines/logs/%Y/%B/%d/%H_%M_%S.log',
 146                 'plugins': ['pandagl', 'p3openal_audio', 'pandadx9'],
 147                 'gui_apps': {app_name: 'main.py'},
 148                 'package_data_dirs': {'simplepbr': [('simplepbr/shaders*', '', {})]},
 149                 'icons': {
 150                     app_name: [
 151                         'assets/images/icon/icon256.png',
 152                         'assets/images/icon/icon128.png',
 153                         'assets/images/icon/icon48.png',
 154                         'assets/images/icon/icon32.png',
 155                         'assets/images/icon/icon16.png']},
 156                 'include_patterns': [
 157                     '**/ya2/licenses/*',
 158                     '**/licenses/*',
 159                     '**/*.bam',
 160                     '**/*.dds',
 161                     '**/*.bin',
 162                     #'**/*.png',
 163                     #'**/*.jpg',
 164                     '**/*.ico',
 165                     '**/*.json',
 166                     '**/track_tr.py',
 167                     '**/*.txt',
 168                     '**/*.ttf',
 169                     '**/*.vert',
 170                     '**/*.frag',
 171                     '**/*.ogg',
 172                     '**/*.wav',
 173                     '**/*.mo'],
 174                 'platforms': __platform_list(),
 175                 'include_modules': {'*': ['encodings.hex_codec']},
 176                 'prefer_discrete_gpu': True},
 177             'bdist_apps': {'installers': __installers_dictionary()}}}
 178     return d
 179
 180 def __platform_list():
 181     platforms = []
 182     if all('--no-windows' not in a for a in argv):
 183         platforms += ['win_amd64']
 184     if all('--no-linux' not in a for a in argv):
 185         platforms += ['manylinux2010_x86_64']
 186     return platforms
 187
 188 def __installers_dictionary():
 189     installers = {}
 190     if all('--no-windows' not in a for a in argv):
 191         installers['win_amd64'] = ['nsis']
 192     if all('--no-linux' not in a for a in argv):
 193         installers['manylinux2010_x86_64'] = []
 194     return installers
 195
 196 if __name__ == '__main__':
 197     setup(**_build_setup_arguments())