imacat/mia-accounting-django
1
0
Fork 0
Code Issues Pull Requests Projects Releases 4 Wiki Activity

Revised the accounting_accounts and accounting_sample commands to call the createsuperuser command when there is no user yet, instead of throwing an error in the accounting application.

Browse Source
This commit is contained in:
依瑪貓 2020-09-02 09:58:02 +08:00
parent 4e411bdc43
commit bc5f940e0a
2 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
accounting/management/commands/accounting_accounts.py
View File

@ -23,7 +23,8 @@ from typing import Optional, Dict
from django.contrib.auth import get_user_model
from django.core.exceptions import ObjectDoesNotExist
from django.core.management import BaseCommand, CommandParser, CommandError
from django.core.management import BaseCommand, CommandParser, CommandError, \
call_command
from django.db import transaction
from accounting.models import Account
@ -875,8 +876,8 @@ class Command(BaseCommand):
error = F"User \"{username_option}\" does not exist."
raise CommandError(error, returncode=1)
if user_model.objects.count() == 0:
error = "Please run the \"createsuperuser\" command first."
raise CommandError(error, returncode=1)
call_command("createsuperuser")
return user_model.objects.first()
if user_model.objects.count() == 1:
return user_model.objects.first()
try:

4
accounting/management/commands/accounting_sample.py
View File

@ -133,8 +133,8 @@ class Command(BaseCommand):
error = F"User \"{username_option}\" does not exist."
raise CommandError(error, returncode=1)
if user_model.objects.count() == 0:
error = "Please run the \"createsuperuser\" command first."
raise CommandError(error, returncode=1)
call_command("createsuperuser")
return user_model.objects.first()
if user_model.objects.count() == 1:
return user_model.objects.first()
try: