From 588cb1e44dce0e9043bb1acd0875f0ee737f4ffe Mon Sep 17 00:00:00 2001 From: Akhil Narang Date: Mon, 22 Dec 2025 17:10:02 +0530 Subject: [PATCH] refactor: tomli -> tomllib Signed-off-by: Akhil Narang --- frappe/testing/environment.py | 7 +++---- frappe/tests/utils/generators.py | 5 ++--- frappe/utils/change_log.py | 2 +- frappe/www/attribution.py | 5 ++--- 4 files changed, 8 insertions(+), 11 deletions(-) diff --git a/frappe/testing/environment.py b/frappe/testing/environment.py index 079daa588f..4faa8f2bd2 100644 --- a/frappe/testing/environment.py +++ b/frappe/testing/environment.py @@ -23,10 +23,9 @@ import functools import inspect import logging import pkgutil +import tomllib import unittest -import tomli - import frappe import frappe.utils.scheduler from frappe.tests.utils import make_test_records, toggle_test_mode @@ -91,7 +90,7 @@ def _decorate_all_methods_and_functions_with_type_checker(): def _get_config_from_pyproject(app_path): try: with open(f"{app_path}/pyproject.toml", "rb") as f: - config = tomli.load(f) + config = tomllib.load(f) return ( config.get("tool", {}) .get("frappe", {}) @@ -100,7 +99,7 @@ def _decorate_all_methods_and_functions_with_type_checker(): ) except FileNotFoundError: return {} - except tomli.TOMLDecodeError: + except tomllib.TOMLDecodeError: logger.warning(f"Failed to parse pyproject.toml for app {app_path}") return {} diff --git a/frappe/tests/utils/generators.py b/frappe/tests/utils/generators.py index 93b72edc2d..e22fc7d2ed 100644 --- a/frappe/tests/utils/generators.py +++ b/frappe/tests/utils/generators.py @@ -2,6 +2,7 @@ import datetime import json import logging import os +import tomllib from collections import defaultdict from collections.abc import Generator from functools import cache @@ -10,8 +11,6 @@ from pathlib import Path from types import MappingProxyType, ModuleType from typing import TYPE_CHECKING, Any -import tomli - import frappe from frappe.model.naming import revert_series_if_last from frappe.modules import get_doctype_module, get_module_path, load_doctype_module @@ -129,7 +128,7 @@ def load_test_records_for(index_doctype) -> dict[str, Any]: toml_path = os.path.join(module_path, "test_records.toml") if os.path.exists(toml_path): with open(toml_path, "rb") as f: - return tomli.load(f) + return tomllib.load(f) else: return {} diff --git a/frappe/utils/change_log.py b/frappe/utils/change_log.py index 83310567a6..67b1615b1e 100644 --- a/frappe/utils/change_log.py +++ b/frappe/utils/change_log.py @@ -394,7 +394,7 @@ def show_update_popup(): def get_pyproject(app: str) -> dict | None: - from tomli import load + from tomllib import load pyproject_path = frappe.get_app_path(app, "..", "pyproject.toml") diff --git a/frappe/www/attribution.py b/frappe/www/attribution.py index 2024aadbd2..35f539dc9e 100644 --- a/frappe/www/attribution.py +++ b/frappe/www/attribution.py @@ -2,10 +2,9 @@ import contextlib import importlib.metadata import json import re +import tomllib from pathlib import Path -import tomli - import frappe from frappe import _ from frappe.permissions import is_system_user @@ -134,7 +133,7 @@ def get_pyproject_info(app: str) -> dict: return {} with open(pyproject_toml, "rb") as f: - pyproject = tomli.load(f) + pyproject = tomllib.load(f) return pyproject.get("project", {})