Coverage for app/venv/lib/python3.14/site-packages/weblate/utils/management/__init__.py: 71%

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

2# 

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

4 

5import sys 

6 

7from django.core.management import ManagementUtility 

8 

9RESTRICTED_COMMANDS = {"squashmigrations", "makemigrations"} 

10 

11 

12class WeblateManagementUtility(ManagementUtility): 

13 def __init__(self, argv=None, developer_mode: bool = False) -> None: 

14 super().__init__(argv) 

15 self.developer_mode = developer_mode 

16 

17 def fetch_command(self, subcommand): 

18 # Block usage of some commands 

19 if not self.developer_mode and subcommand in RESTRICTED_COMMANDS: 19 ↛ 20line 19 didn't jump to line 20 because the condition on line 19 was never true

20 sys.stderr.write(f"Blocked command: {subcommand!r}\n") 

21 sys.stderr.write("This command is restricted for developers only.\n") 

22 sys.stderr.write( 

23 "In case you really want to do this, please execute " 

24 "using manage.py from the Weblate source code.\n" 

25 ) 

26 sys.exit(1) 

27 

28 # Fetch command class 

29 command = super().fetch_command(subcommand) 

30 

31 # Monkey patch it's output 

32 original_notice = command.style.NOTICE 

33 

34 def patched_notice(txt): 

35 return original_notice( 

36 txt.replace("python manage.py migrate", "weblate migrate") 

37 ) 

38 

39 command.style.NOTICE = patched_notice # type: ignore[method-assign] 

40 

41 return command