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

Revised the accounting_sample command to try to use the user with the log in ID as the current system user when there are more than one users in the accounting application.

Browse Source
This commit is contained in:
依瑪貓 2020-08-31 23:04:03 +08:00
parent 731d867c56
commit 03ebe62eb8
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -19,6 +19,7 @@
""" """
import datetime import datetime
import getpass
import random import random
from typing import Optional from typing import Optional
@ -75,8 +76,14 @@ class Command(BaseCommand):
elif user_model.objects.count() == 1: elif user_model.objects.count() == 1:
user = user_model.objects.first() user = user_model.objects.first()
else: else:
error = "Please specify the user with -u." try:
raise CommandError(error, returncode=1) user = user_model.objects.get(**{
user_model.USERNAME_FIELD: getpass.getuser()
})
except ObjectDoesNotExist:
error = "Please specify the user with -u."
raise CommandError(error, returncode=1)
self.stdout.write(F"Filling sample data as \"{user}\"")
with transaction.atomic(): with transaction.atomic():
self._filler = DataFiller(user) self._filler = DataFiller(user)