A 添加命令行系统设置

spug_api/apps/account/management/commands/set.py

@@ -0,0 +1,37 @@
+# Copyright: (c) OpenSpug Organization. https://github.com/openspug/spug
+# Copyright: (c) <spug.dev@gmail.com>
+# Released under the AGPL-3.0 License.
+from django.core.management.base import BaseCommand
+from apps.setting.utils import AppSetting
+
+
+class Command(BaseCommand):
+    help = '系统设置'
+
+    def add_arguments(self, parser):
+        parser.add_argument('target', type=str, help='设置对象')
+        parser.add_argument('value', type=str, help='设置值')
+
+    def echo_success(self, msg):
+        self.stdout.write(self.style.SUCCESS(msg))
+
+    def echo_error(self, msg):
+        self.stderr.write(self.style.ERROR(msg))
+
+    def print_help(self, *args):
+        message = '''
+        系统设置命令用法：
+            set mfa disable     禁用登录MFA
+        '''
+        self.stdout.write(message)
+
+    def handle(self, *args, **options):
+        target = options['target']
+        if target == 'mfa':
+            if options['value'] != 'disable':
+                return self.echo_error(f'mfa设置，不支持的值【{options["value"]}】')
+            AppSetting.set('MFA', {'enable': False})
+            self.echo_success('MFA已禁用')
+        else:
+            self.echo_error('未识别的操作')
+            self.print_help()