simplify load in state manager

Author: dmytrotkk

credit-distributor/src/state.py

@@ -17,7 +17,6 @@
 #   You should have received a copy of the GNU Affero General Public License
 #   along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
-import json
 import logging
 from pathlib import Path
 
@@ -35,18 +34,16 @@ def __init__(self, state_file: str | Path):
         self.state_file = Path(state_file)
 
     def load(self, initial_from_blocks: dict[str, int]) -> State:
+        from_blocks = dict(initial_from_blocks)
         if self.state_file.exists():
-            try:
-                stored = State(**json.loads(self.state_file.read_text())).from_blocks
-                logger.info(f'Loaded state from {self.state_file}')
-                from_blocks = {n: stored.get(n, b) for n, b in initial_from_blocks.items()}
-                return State(from_blocks=from_blocks)
-            except Exception as e:
-                logger.warning(
-                    f'Failed to load state from {self.state_file}: {e}. Using initial values.'
-                )
-        logger.info(f'Creating new state with from_blocks={initial_from_blocks}')
-        return State(from_blocks=dict(initial_from_blocks))
+            stored = State.model_validate_json(self.state_file.read_text()).from_blocks
+            logger.info(f'Loaded state from {self.state_file}')
+            for name in from_blocks:
+                if name in stored:
+                    from_blocks[name] = stored[name]
+        else:
+            logger.info(f'Creating new state with from_blocks={from_blocks}')
+        return State(from_blocks=from_blocks)
 
     def save(self, state: State) -> None:
         self.state_file.write_text(state.model_dump_json(indent=2))