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

Changed the safe-guard condition of the accounting_sample command in the accounting application.

Browse Source
This commit is contained in:
依瑪貓 2020-08-02 18:29:10 +08:00
parent 2ae1e825f6
commit 65a950a28a
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -19,6 +19,7 @@
""" """
import random import random
import sys
from django.core.management import BaseCommand, CommandParser from django.core.management import BaseCommand, CommandParser
from django.db import transaction from django.db import transaction
@ -47,8 +48,9 @@ class Command(BaseCommand):
*args (list[str]): The command line arguments. *args (list[str]): The command line arguments.
**options (dict[str,str]): The command line switches. **options (dict[str,str]): The command line switches.
""" """
print("This may mess up your data. Continue? [Y/N] ", end="") if User.objects.first() is not None:
if input().lower() not in ("y", "yes"): print("Refused to fill in sample data with existing data.",
file=sys.stderr)
return return
with transaction.atomic(): with transaction.atomic():