augmented-sim/setup.py at master · Recovida/augmented-sim · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
#!/usr/bin/env python3
# coding: utf-8
# flake8: noqa

from setuptools import setup, find_packages

import os
import pathlib
import platform


YEAR = 2020
GIT_URL = 'https://gitlab.com/projeto-fm-usp-mortalidade-sp/augmented-sim'


here = pathlib.Path(__file__).parent.resolve()
LONG_DESCRIPTION = (here / 'README.md').read_text(encoding='utf-8')
DEPENDENCIES = (here / 'requirements.txt') \
    .read_text(encoding='utf-8').strip().split('\n')


windows = platform.system() == 'Windows'

s = setup(
    long_description=LONG_DESCRIPTION,
    long_description_content_type='text/markdown',
    url=GIT_URL,

    # author='FM-USP',
    # author_email='palotufo@usp.br',

    classifiers=[
        #  3 - Alpha; 4 - Beta; 5 - Production/Stable
        'Development Status :: 3 - Alpha',
        'Intended Audience :: Healthcare Industry',
        'Intended Audience :: Science/Research',
        'Topic :: Scientific/Engineering :: Information Analysis',
        'License :: Other/Proprietary License',
        'Programming Language :: Python :: 3',
        'Programming Language :: Python :: 3.6',
        'Programming Language :: Python :: 3.7',
        'Programming Language :: Python :: 3.8',
        'Programming Language :: Python :: 3 :: Only',
    ],
    keywords='SIM, mortality, healthcare, ICD',

    packages=find_packages(),
    python_requires='>=3.6, <4',
    install_requires=DEPENDENCIES,
    package_data={'augmented_sim':
                    ['../README.md', '../LICENCE.txt', '.licences/*',
                     'i18n/*.qm']},
    include_package_data=True,

    entry_points={
        'console_scripts': [
            'augmentedsim_cli=augmented_sim.augmented_sim_cli:main',
        ],
        'gui_scripts': [
            'augmentedsim_gui=augmented_sim.augmented_sim_gui:main',
        ],
    },

    project_urls={
        'Bug Reports': GIT_URL + '/-/issues',
        'Source': GIT_URL,
    },

)


def create_shortcut(name: str, description: str, file_name: str) -> None:
    import pyshortcuts

    # On Linux, add to some categories
    df = pyshortcuts.linux.DESKTOP_FORM
    add = 'Categories=Office;Spreadsheet;Database;MedicalSoftware;'
    pyshortcuts.linux.DESKTOP_FORM = df.rstrip() + '\n' + add

    cmd = file_name
    kwargs = {
        'name': name,
        'description': description,
        'terminal': False
    }
    pyshortcuts.make_shortcut(cmd, desktop=True, startmenu=True, **kwargs)


def _create_custom_entry_point(file_name: str) -> None:
    # The file generated by setuptools has an invalid encoding
    contents = """import subprocess, sys;""" \
        """process = subprocess.Popen([sys.executable, '-m', """ \
        """'augmented_sim.augmented_sim_gui',*sys.argv[1:]],shell=False);""" \
        """process.wait(); exit(process.returncode)"""
    with open(file_name, 'w', encoding='utf-8') as fd:
        print(contents, file=fd, end='')


def _post_install(setup):
    def _post_install_actions():
        if 'install' not in setup.command_obj:
            return
        directory = setup.command_obj['install'].install_scripts
        gui_script = os.path.join(directory, 'augmentedsim_gui')
        import sys
        if 'install' in sys.argv[1:]:
            if windows:
                if os.path.isfile(sys.executable[:-4] + 'w.exe'):
                    gui_script += '_custom.py'
                    _create_custom_entry_point(gui_script)
            name = setup.metadata.name
            description = setup.metadata.description
            create_shortcut(name, description, gui_script)
        print('Ok')
    _post_install_actions()
    return setup


setup = _post_install(s)