bug fixes (#118)
This commit is contained in:
@@ -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
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user