31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
from django.core.management.base import BaseCommand
|
|
|
|
from apps.user.models import Institution
|
|
from apps.user.auth import TRIAL_INSTITUTION_NAME, TRIAL_INSTITUTION_CODE
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = '初始化预留试用机构「北大医学部（实验室）试用」'
|
|
|
|
def handle(self, *args, **options):
|
|
inst, created = Institution.objects.get_or_create(
|
|
code=TRIAL_INSTITUTION_CODE,
|
|
defaults={'name': TRIAL_INSTITUTION_NAME, 'type': 'hospital'},
|
|
)
|
|
|
|
if created:
|
|
self.stdout.write(self.style.SUCCESS(
|
|
f'[创建] 试用机构: {inst.code} / {inst.name}'
|
|
))
|
|
elif inst.name != TRIAL_INSTITUTION_NAME:
|
|
old_name = inst.name
|
|
inst.name = TRIAL_INSTITUTION_NAME
|
|
inst.save(update_fields=['name'])
|
|
self.stdout.write(self.style.WARNING(
|
|
f'[更新] 试用机构名称: {old_name} -> {inst.name}'
|
|
))
|
|
else:
|
|
self.stdout.write(self.style.WARNING(
|
|
f'[已存在] 试用机构: {inst.code} / {inst.name}'
|
|
))
|