mirror of
https://github.com/davidhalter/django-stubs.git
synced 2025-12-07 04:34:29 +08:00
31 lines
1.4 KiB
Python
31 lines
1.4 KiB
Python
from pytest_mypy.collect import File
|
|
from pytest_mypy.item import YamlTestItem
|
|
|
|
|
|
def django_plugin_hook(test_item: YamlTestItem) -> None:
|
|
custom_settings = test_item.parsed_test_data.get('custom_settings', '')
|
|
installed_apps = test_item.parsed_test_data.get('installed_apps', None)
|
|
|
|
if installed_apps and custom_settings:
|
|
raise ValueError('"installed_apps" and "custom_settings" are not compatible, please use one or the other')
|
|
|
|
if installed_apps is not None:
|
|
# custom_settings is empty, add INSTALLED_APPS
|
|
installed_apps += ['django.contrib.contenttypes']
|
|
installed_apps_as_str = '(' + ','.join([repr(app) for app in installed_apps]) + ',)'
|
|
custom_settings += 'INSTALLED_APPS = ' + installed_apps_as_str
|
|
|
|
if 'SECRET_KEY' not in custom_settings:
|
|
custom_settings = 'SECRET_KEY = "1"\n' + custom_settings
|
|
|
|
django_settings_section = "\n[mypy.plugins.django-stubs]\n" \
|
|
"django_settings_module = mysettings"
|
|
if not test_item.additional_mypy_config:
|
|
test_item.additional_mypy_config = django_settings_section
|
|
else:
|
|
if '[mypy.plugins.django-stubs]' not in test_item.additional_mypy_config:
|
|
test_item.additional_mypy_config += django_settings_section
|
|
|
|
mysettings_file = File(path='mysettings.py', content=custom_settings)
|
|
test_item.files.append(mysettings_file)
|