From ad6b240c144271ce1bfbc2de50dd76b87576a09a Mon Sep 17 00:00:00 2001 From: Masoko Date: Fri, 24 May 2024 18:12:37 +0300 Subject: [PATCH] bug fixes (#118) --- src/update.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/update.py b/src/update.py index 1014406..1377a1d 100644 --- a/src/update.py +++ b/src/update.py @@ -3,11 +3,17 @@ import os import subprocess import config +def safe_literal_eval(node): + try: + return ast.literal_eval(node) + except ValueError: + return None + def get_assignments(filename): with open(filename) as f: tree = ast.parse(f.read(), filename) - assignments = {node.targets[0].id: ast.literal_eval(node.value) for node in ast.walk(tree) if isinstance(node, ast.Assign)} + assignments = {node.targets[0].id: safe_literal_eval(node.value) for node in ast.walk(tree) if isinstance(node, ast.Assign)} return assignments