PyExpression (@Nullable PyStatement def) { if (def == null) return null; if (def instanceof PyAssignmentStatement) { return ((PyAssignmentStatement)def).getAssignedValue(); } return ((PyAugAssignmentStatement)def).getValue(); }