pyfastblur/setup.py at main · sertraline/pyfastblur · 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
from setuptools.command.build_py import build_py as _build_py
from distutils.core import setup, Extension
from glob import glob
import shutil
import os
from os.path import isdir, isfile, join

this_directory = os.path.abspath(os.path.dirname(__file__))
with open(os.path.join(this_directory, 'README.md'), encoding='utf-8') as f:
    long_description = f.read()


class build_py(_build_py):
    def run(self):
        self.run_command("build_ext")
        if os.name == 'nt':
            pyd = glob('pyfastblur/*.pyd')
        else:
            pyd = glob('pyfastblur/*.so')
        build_dir = join('pyfastblur', 'libpyfastblur')
        if not isdir(build_dir):
            os.mkdir(build_dir)

        if os.name == 'nt':
            build_file = join(build_dir, 'pyfastblur_cpp.pyd')
        else:
            build_file = join(build_dir, 'pyfastblur_cpp.so')
        if not isfile(build_file):
            shutil.move(pyd[0], build_file)
        else:
            os.replace(pyd[0], build_file)
        with open(join(build_dir, '__init__.py'), 'w') as file:
            file.write('from .pyfastblur_cpp import *')

        if os.name == 'nt':
            dll_path = join('pyfastblur', 'src', 'win')
            ext = '.dll'
        else:
            dll_path = join('pyfastblur', 'src', 'unix')
            ext = '.so'

        lpng = join(dll_path, 'libpng16'+ext)
        dest_lpng = join(build_dir, 'libpng16'+ext)

        lz = join(dll_path, 'zlib'+ext)
        dest_lz = join(build_dir, 'zlib'+ext)

        if not isfile(dest_lpng):
            shutil.copy(lpng, dest_lpng)
        if not isfile(dest_lz):
            shutil.copy(lz, dest_lz)
        return super().run()


VERSION = [('MAJOR_VERSION', '1'),
           ('MINOR_VERSION', '3')]

sources = ['pyfastblur/src/blur.cpp']
include = ['pyfastblur/src',
           'pyfastblur/src/include',
           ]

if os.name == 'nt':
    bmodule = Extension('pyfastblur_cpp',
                        define_macros=VERSION,
                        include_dirs=include,
                        library_dirs=['pyfastblur/src/win'],
                        libraries=['libpng16', 'zlib'],
                        sources=sources
                        )
    packages = {"pyfastblur/libpyfastblur": ['__init__.py',
                                             'pyfastblur_cpp.pyd',
                                             'libpng16.dll',
                                             'zlib.dll'
                                             ]}

else:
    bmodule = Extension('pyfastblur_cpp',
                        define_macros=VERSION,
                        include_dirs=include,
                        library_dirs=['pyfastblur/src/unix', '/usr/local/lib'],
                        libraries=['png16', 'z'],
                        sources=sources
                        )
    packages = {"pyfastblur/libpyfastblur": ['__init__.py',
                                             'pyfastblur_cpp.so',
                                             'png16.so',
                                             'z.so'
                                             ]}

setup(name='pyfastblur',
      version='1.3',
      description='Small Python library with a single purpose to apply fast blur to PNG images (libpng backend)',
      author='Toshiro Iwa',
      author_email='iwa@acid.im',
      long_description=long_description,
      long_description_content_type='text/markdown',
      url='https://github.com/sertraline/pyfastblur',
      packages=["pyfastblur", "pyfastblur/libpyfastblur"],
      package_data=packages,
      include_package_data=True,
      ext_modules=[bmodule],
      cmdclass={'build_py': build_py},
      )