# SPDX-License-Identifier: LGPL-2.1-or-later

# Copyright (C) 2020, 2021 igo95862

# This file is part of python-sdbus

# This library is free software; you can redistribute it and/or

# modify it under the terms of the GNU Lesser General Public

# License as published by the Free Software Foundation; either

# version 2.1 of the License, or (at your option) any later version.

# This library is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU

# Lesser General Public License for more details.

# You should have received a copy of the GNU Lesser General Public

# License along with this library; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

from __future__ import annotations

from argparse import ArgumentParser

from os import environ

from pathlib import Path

from subprocess import SubprocessError, run

from typing import List

source_root = Path(environ['MESON_SOURCE_ROOT'])

build_dir = Path(environ['MESON_BUILD_ROOT'])

tools_dir = source_root / 'tools'

src_dir = source_root / 'src'

test_dir = source_root / 'test'

wheel_build_dir = source_root / 'wheel-build'

examples_dir = source_root / 'examples'

all_python_modules = [

tools_dir, test_dir, wheel_build_dir,

src_dir / 'sdbus',

src_dir / 'sdbus_async/dbus_daemon',

src_dir / 'sdbus_block/dbus_daemon',

source_root / 'setup.py',

]

mypy_cache_dir = build_dir / '.mypy_cache'

def run_mypy() -> None:

print('Running mypy on all modules')

run(

args=(

'mypy', '--strict', '--pretty',

'--cache-dir', mypy_cache_dir,

'--python-version', '3.8',

'--namespace-packages',

'--explicit-package-bases',

*all_python_modules,

),

check=True,

env={'MYPYPATH': str(src_dir.absolute()), **environ},

)

def run_flake8() -> None:

run(

args=(

'flake8',

*all_python_modules,

),

check=True,

)

def linter_main() -> None:

is_success = True

try:

run_flake8()

except SubprocessError:

is_success = False

try:

run_mypy()

except SubprocessError:

is_success = False

if not is_success:

raise SystemExit(1)

def get_all_python_files() -> List[Path]:

python_files: List[Path] = [source_root / 'setup.py']

for python_module in all_python_modules:

if python_module.is_dir():

for a_file in python_module.iterdir():

if a_file.suffix == '.py':

python_files.append(a_file)

else:

python_files.append(python_module)

return python_files

def formater_main() -> None:

run(

args=('autopep8', '--recursive', '--in-place', *all_python_modules),

check=True,

)

run(

args=(

'isort',

'-m', 'VERTICAL_HANGING_INDENT',

'--trailing-comma',

*all_python_modules,

),

check=True,

)

def main() -> None:

parser = ArgumentParser()

parser.add_argument(

'mode',

choices=('lint', 'format'),

)

args = parser.parse_args()

mode = args.mode

if mode == 'lint':

linter_main()

elif mode == 'format':

formater_main()

else:

raise ValueError('Unknown mode', mode)

if __name__ == '__main__':

main()