CGM_Public/pretix_cgo
1
0
Fork 0
forked from CGM_Public/pretix_original
Code Pull Requests Actions Activity

shell_scoped: Fall back to shell if shell_plus is not installed

Browse Source
This commit is contained in:
Raphael Michel
2020-03-06 14:03:06 +01:00
parent d8c2c82da7
commit 389884d191
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/pretix/base/management/commands/shell_scoped.py
View File

@@ -13,11 +13,17 @@ class Command(BaseCommand):
return parser return parser
def handle(self, *args, **options): def handle(self, *args, **options):
try:
from django_extensions.management.commands import shell_plus # noqa
cmd = 'shell_plus'
except ImportError:
cmd = 'shell'
parser = self.create_parser(sys.argv[0], sys.argv[1]) parser = self.create_parser(sys.argv[0], sys.argv[1])
flags = parser.parse_known_args(sys.argv[2:])[1] flags = parser.parse_known_args(sys.argv[2:])[1]
if "--override" in flags: if "--override" in flags:
with scopes_disabled(): with scopes_disabled():
return call_command("shell_plus", *args, **options) return call_command(cmd, *args, **options)
lookups = {} lookups = {}
for flag in flags: for flag in flags:
@@ -36,4 +42,4 @@ class Command(BaseCommand):
for app_name, app_value in lookups.items() for app_name, app_value in lookups.items()
} }
with scope(**scope_options): with scope(**scope_options):
return call_command("shell_plus", *args, **options) return call_command(cmd, *args, **options)