mirror of
https://github.com/davidhalter/django-stubs.git
synced 2025-12-10 05:51:53 +08:00
44151c485d
* updated package setup * updated to use python 3.9 * fixed test runner * fixed typecheck tests * fixed discrepencies * added override to runner * updated travis * updated pre-commit hooks * updated dep
30 lines
1.4 KiB
Python
30 lines
1.4 KiB
Python
from pytest_mypy_plugins.collect import File
|
|
from pytest_mypy_plugins.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)
|