bug fixes (#118)

This commit is contained in:
Masoko
2024-05-24 18:12:37 +03:00
committed by GitHub
parent b25293d419
commit ad6b240c14

View File

@@ -3,11 +3,17 @@ import os
import subprocess import subprocess
import config import config
def safe_literal_eval(node):
try:
return ast.literal_eval(node)
except ValueError:
return None
def get_assignments(filename): def get_assignments(filename):
with open(filename) as f: with open(filename) as f:
tree = ast.parse(f.read(), filename) 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 return assignments