setup.py

import setuptools
from setuptools.command.install import install
from pathlib import Path
from pycritty import __version__
from pycritty.resources import pycritty_dir, alacritty_dir, config_file


class PostInstallHook(install):
    user_options = install.user_options + [
        ('themes=', None, 'Themes to be installed')
    ]

    def initialize_options(self):
        super().initialize_options()
        self.themes = None

    def finalize_options(self):
        super().finalize_options()

    def run(self):
        super().run()
        self.hook()

    def hook(self):
        installation_resources = Path(__file__).parent / 'config'
        pycritty_dir.create()
        alacritty_dir.create()
        config_file.create()

        theme_files = list((installation_resources / 'themes').iterdir())
        # Don't install any theme by default
        self.exclude = set(theme_files)

        if self.themes is not None:
            # Parse options
            include = set(self.themes.replace(' ', '').split(','))
            if 'all' in include:
                include = set([f.stem for f in theme_files])
            for f in theme_files:
                if f.stem in include:
                    self.exclude.remove(f)

        self.cp(installation_resources, pycritty_dir.path)

    def cp(self, src_dir: Path, dest_dir: Path):
        for src_child in src_dir.iterdir():
            if src_child in self.exclude:
                continue
            dest_child = dest_dir / src_child.name
            if src_child.is_dir():
                dest_child.mkdir(exist_ok=True)
                self.cp(src_child, dest_child)
            elif not dest_child.is_file():
                dest_child.touch()
                with open(src_child, 'r') as s, open(dest_child, 'w') as d:
                    d.write(s.read())


with open("README.md", "r", encoding="utf-8") as f:
    README = f.read()

setuptools.setup(
    name="pycritty",
    version=__version__,
    author="Antonio Sarosi",
    author_email="sarosiantonio@gmail.com",
    description="CLI program that allows you to change your alacritty config file with one command.",  # noqa: E501
    long_description=README,
    long_description_content_type="text/markdown",
    url="https://github.com/antoniosarosi/pycritty",
    packages=setuptools.find_packages(),
    classifiers=[
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: MIT License",
        "Operating System :: POSIX :: Linux",
        "Topic :: Terminals :: Terminal Emulators/X Terminals"
    ],
    keywords="alacritty",
    python_requires='>=3.6',
    install_requires=["PyYAML"],
    include_package_data=True,
    cmdclass={
        'install': PostInstallHook,  # type: ignore
    },
    entry_points={
        "console_scripts": ["pycritty = pycritty.main:main"]
    },
)