Coverage for app/venv/lib/python3.14/site-packages/weblate/machinery/management/commands/install_machinery.py: 0%

1# Copyright © Michal Čihař <michal@weblate.org> 

2# 

3# SPDX-License-Identifier: GPL-3.0-or-later 

4from __future__ import annotations 

5 

6from typing import TYPE_CHECKING 

7 

8from django.core.management.base import CommandError 

9 

10from weblate.configuration.models import Setting, SettingCategory 

11from weblate.machinery.models import validate_service_configuration 

12from weblate.utils.management.base import BaseCommand 

13 

14if TYPE_CHECKING: 

15 from django.core.management.base import CommandParser 

16 

17 

18class Command(BaseCommand): 

19 help = "installs site-wide automatic suggestion service" 

20 

21 def add_arguments(self, parser: CommandParser) -> None: 

22 super().add_arguments(parser) 

23 parser.add_argument("--service", required=True, help="Service name") 

24 parser.add_argument( 

25 "--configuration", default="{}", help="Service configuration in JSON" 

26 ) 

27 parser.add_argument( 

28 "--update", 

29 action="store_true", 

30 help="Update existing service configuration", 

31 ) 

32 

33 def handle(self, *args, **options) -> None: 

34 service, configuration, errors = validate_service_configuration( 

35 options["service"], options["configuration"] 

36 ) 

37 

38 if service is None or errors: 

39 for error in errors: 

40 self.stderr.write(error) 

41 msg = "Invalid add-on configuration!" 

42 raise CommandError(msg) 

43 

44 setting, created = Setting.objects.get_or_create( 

45 category=SettingCategory.MT, 

46 name=options["service"], 

47 defaults={"value": configuration}, 

48 ) 

49 if not created and options["update"]: 

50 setting.value = configuration 

51 setting.save() 

52 self.stdout.write(f"Service installed: {service.name}")