Python AST read/write