Coverage for hdl_registers/parser/toml.py: 85%

1# -------------------------------------------------------------------------------------------------- 

2# Copyright (c) Lukas Vik. All rights reserved. 

3# 

4# This file is part of the hdl-registers project, an HDL register generator fast enough to run 

5# in real time. 

6# https://hdl-registers.com 

7# https://github.com/hdl-registers/hdl-registers 

8# -------------------------------------------------------------------------------------------------- 

9 

10from __future__ import annotations 

11 

12from typing import TYPE_CHECKING, Any 

13 

14import rtoml 

15 

16from .parser import RegisterParser 

17 

18if TYPE_CHECKING: 

19 from pathlib import Path 

20 

21 from hdl_registers.register import Register 

22 from hdl_registers.register_list import RegisterList 

23 

24 

25def from_toml( 

26 name: str, toml_file: Path, default_registers: list[Register] | None = None 

27) -> RegisterList: 

28 """ 

29 Parse a TOML file with register data. 

30 

31 Arguments: 

32 name: The name of the register list. 

33 toml_file: The TOML file path. 

34 default_registers: List of default registers. 

35 

36 Return: 

37 The resulting register list. 

38 """ 

39 parser = RegisterParser( 

40 name=name, source_definition_file=toml_file, default_registers=default_registers 

41 ) 

42 toml_data = _load_toml_file(file_path=toml_file) 

43 

44 return parser.parse(register_data=toml_data) 

45 

46 

47def _load_toml_file(file_path: Path) -> dict[str, Any]: 

48 """ 

49 Load and parse the TOML data into a dictionary. Raise exceptions if things dont work. 

50 """ 

51 if not file_path.exists(): 

52 raise FileNotFoundError(f"Requested TOML file does not exist: {file_path}") 

53 

54 try: 

55 return rtoml.load(file_path) 

56 except Exception as exception_info: 

57 message = f"Error while parsing TOML file {file_path}:\n{exception_info}" 

58 raise ValueError(message) from exception_info