[gentoo-catalyst] [PATCH] catalyst: Switch to tomli

[gentoo-catalyst] [PATCH] catalyst: Switch to tomli

[Matt Turner]

The Python community is coalescing around tomli, and tomli is likely to
be integrated into the standard library per PEP680.

Signed-off-by: Matt Turner <mattst88@gentoo.org>
---
 catalyst/base/stagebase.py         | 6 +++---
 catalyst/main.py                   | 5 +++--
 doc/make_subarch_table_guidexml.py | 5 +++--
 3 files changed, 9 insertions(+), 7 deletions(-)

diff --git a/catalyst/base/stagebase.py b/catalyst/base/stagebase.py
index 4a1b4eb6..de1e30ef 100644
--- a/catalyst/base/stagebase.py
+++ b/catalyst/base/stagebase.py
@@ -10,7 +10,7 @@ from pathlib import Path
 
 import fasteners
 import libmount
-import toml
+import tomli
 
 from snakeoil import fileutils
 from snakeoil.osutils import pjoin
@@ -123,8 +123,8 @@ class StageBase(TargetBase, ClearBase, GenBase):
             log.debug("\tTrying %s", x)
             name = x[:-len('.toml')]
 
-            with open(arch_dir + x) as file:
-                arch_config = toml.load(file)
+            with open(arch_dir + x, 'rb') as file:
+                arch_config = tomli.load(file)
 
                 # Search for a subarchitecture in each arch in the arch_config
                 for arch in [x for x in arch_config if x.startswith(name) and host in arch_config[x]]:
diff --git a/catalyst/main.py b/catalyst/main.py
index 0de1040f..6e9a2d3e 100644
--- a/catalyst/main.py
+++ b/catalyst/main.py
@@ -6,7 +6,7 @@ import os
 import sys
 import textwrap
 
-import toml
+import tomli
 
 from DeComp.definitions import (COMPRESS_DEFINITIONS, DECOMPRESS_DEFINITIONS,
                                 CONTENTS_DEFINITIONS)
@@ -35,7 +35,8 @@ def parse_config(config_files):
     for config_file in config_files:
         log.notice('Loading configuration file: %s', config_file)
         try:
-            config = toml.load(config_file)
+            with open(config_file, 'rb') as f:
+                config = tomli.load(f)
             for key in config:
                 if key not in valid_config_file_values:
                     log.critical("Unknown option '%s' in config file %s",
diff --git a/doc/make_subarch_table_guidexml.py b/doc/make_subarch_table_guidexml.py
index 67ed3ccc..3c03f90c 100755
--- a/doc/make_subarch_table_guidexml.py
+++ b/doc/make_subarch_table_guidexml.py
@@ -5,7 +5,7 @@
 import pathlib
 import sys
 import textwrap
-import toml
+import tomli
 
 
 def write_guidexml(arch_to_subarch):
@@ -40,7 +40,8 @@ def main(_argv):
     p = pathlib.Path('arch')
 
     for file in p.glob('*.toml'):
-        data = toml.load(file)
+        with file.open('rb') as f:
+            data = tomli.load(f)
 
         for arch in [x for x in data if x != 'setarch']:
             arch_to_subarch.update({arch: list(data[arch].keys())})
-- 
2.34.1