Cleaning up the imports and fixing a few pythonic issues

This commit is contained in:
Daniel W. Anner 2023-08-04 15:48:16 +00:00 committed by GitHub
parent cdd3bc31ba
commit 71983ad84e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 13 deletions

View File

@ -558,4 +558,4 @@
} }
} }
} }
} }

View File

@ -1,19 +1,15 @@
from os import path as os_path
from json import dumps as json_dumps from json import dumps as json_dumps
from json import loads as json_loads from json import loads as json_loads
from jinja2 import FileSystemLoader, Environment from jinja2 import FileSystemLoader, Environment
from django.core.management.base import BaseCommand
from django.conf import settings from django.conf import settings
from django.core.management.base import BaseCommand
from dcim.choices import DeviceAirflowChoices from dcim.choices import (
from dcim.choices import SubdeviceRoleChoices DeviceAirflowChoices, SubdeviceRoleChoices, ConsolePortTypeChoices, PowerPortTypeChoices,
from dcim.choices import ConsolePortTypeChoices PowerOutletTypeChoices, PowerOutletFeedLegChoices, InterfaceTypeChoices, InterfacePoEModeChoices,
from dcim.choices import PowerPortTypeChoices InterfacePoETypeChoices, PortTypeChoices, WeightUnitChoices
from dcim.choices import PowerOutletTypeChoices, PowerOutletFeedLegChoices )
from dcim.choices import InterfaceTypeChoices, InterfacePoEModeChoices, InterfacePoETypeChoices
from dcim.choices import PortTypeChoices
from dcim.choices import WeightUnitChoices
class Command(BaseCommand): class Command(BaseCommand):
@ -55,7 +51,7 @@ class Command(BaseCommand):
print(json_dumps(json_loads(outputText), indent=4)) print(json_dumps(json_loads(outputText), indent=4))
if kwargs['file']: if kwargs['file']:
print() with open(f'{settings.BASE_DIR}/../contrib/generated_schema.json', mode='w', encoding='UTF-8') as generated_json_file:
with open(f'{settings.BASE_DIR}/../contrib/generated_schema.json', 'w') as generated_json_file:
generated_json_file.write(json_dumps(json_loads(outputText), indent=4)) generated_json_file.write(json_dumps(json_loads(outputText), indent=4))
generated_json_file.write('\n')
generated_json_file.close() generated_json_file.close()