| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| */ |
| |
| package freemarker.ext.jython; |
| |
| import java.util.Iterator; |
| import java.util.List; |
| |
| import org.python.core.Py; |
| import org.python.core.PyException; |
| import org.python.core.PyObject; |
| |
| import freemarker.ext.util.ModelFactory; |
| import freemarker.ext.util.WrapperTemplateModel; |
| import freemarker.template.AdapterTemplateModel; |
| import freemarker.template.ObjectWrapper; |
| import freemarker.template.TemplateBooleanModel; |
| import freemarker.template.TemplateHashModel; |
| import freemarker.template.TemplateMethodModelEx; |
| import freemarker.template.TemplateModel; |
| import freemarker.template.TemplateModelException; |
| import freemarker.template.TemplateScalarModel; |
| |
| /** |
| * Generic model for arbitrary Jython objects. |
| */ |
| public class JythonModel |
| implements TemplateBooleanModel, TemplateScalarModel, TemplateHashModel, |
| TemplateMethodModelEx, AdapterTemplateModel, WrapperTemplateModel { |
| protected final PyObject object; |
| protected final JythonWrapper wrapper; |
| |
| static final ModelFactory FACTORY = |
| new ModelFactory() |
| { |
| @Override |
| public TemplateModel create(Object object, ObjectWrapper wrapper) { |
| return new JythonModel((PyObject) object, (JythonWrapper) wrapper); |
| } |
| }; |
| |
| public JythonModel(PyObject object, JythonWrapper wrapper) { |
| this.object = object; |
| this.wrapper = wrapper; |
| } |
| |
| /** |
| * Returns the value of {@link PyObject#__nonzero__()}. |
| */ |
| @Override |
| public boolean getAsBoolean() throws TemplateModelException { |
| try { |
| return object.__nonzero__(); |
| } catch (PyException e) { |
| throw new TemplateModelException(e); |
| } |
| } |
| |
| /** |
| * Returns the value of {@link Object#toString()}. |
| */ |
| @Override |
| public String getAsString() throws TemplateModelException { |
| try { |
| return object.toString(); |
| } catch (PyException e) { |
| throw new TemplateModelException(e); |
| } |
| } |
| |
| /** |
| * Calls {@link PyObject#__findattr__(java.lang.String)}, then if it |
| * returns null calls {@link PyObject#__finditem__(java.lang.String)}. |
| * If {@link JythonWrapper#setAttributesShadowItems(boolean)} was called |
| * with <code>false</code>, the order of calls is reversed (that is, item |
| * lookup takes precedence over attribute lookup). |
| */ |
| @Override |
| public TemplateModel get(String key) |
| throws TemplateModelException { |
| if (key != null) { |
| key = key.intern(); |
| } |
| |
| PyObject obj = null; |
| |
| try { |
| if (wrapper.isAttributesShadowItems()) { |
| obj = object.__findattr__(key); |
| if (obj == null) { |
| obj = object.__finditem__(key); |
| } |
| } else { |
| obj = object.__finditem__(key); |
| if (obj == null) { |
| obj = object.__findattr__(key); |
| } |
| } |
| } catch (PyException e) { |
| throw new TemplateModelException(e); |
| } |
| |
| return wrapper.wrap(obj); |
| } |
| |
| /** |
| * Returns {@link PyObject#__len__()}<code> == 0</code>. |
| */ |
| @Override |
| public boolean isEmpty() throws TemplateModelException { |
| try { |
| return object.__len__() == 0; |
| } catch (PyException e) { |
| throw new TemplateModelException(e); |
| } |
| } |
| |
| /** |
| * @see freemarker.template.TemplateMethodModel#exec(List) |
| */ |
| @Override |
| public Object exec(List arguments) throws TemplateModelException { |
| int size = arguments.size(); |
| try { |
| switch(size) |
| { |
| case 0: |
| { |
| return wrapper.wrap(object.__call__()); |
| } |
| case 1: |
| { |
| return wrapper.wrap(object.__call__(wrapper.unwrap( |
| (TemplateModel) arguments.get(0)))); |
| } |
| default: |
| { |
| PyObject[] pyargs = new PyObject[size]; |
| int i = 0; |
| for (Iterator arg = arguments.iterator(); arg.hasNext(); ) { |
| pyargs[i++] = wrapper.unwrap( |
| (TemplateModel) arg.next()); |
| } |
| return wrapper.wrap(object.__call__(pyargs)); |
| } |
| } |
| } catch (PyException e) { |
| throw new TemplateModelException(e); |
| } |
| } |
| |
| @Override |
| public Object getAdaptedObject(Class hint) { |
| if (object == null) { |
| return null; |
| } |
| Object view = object.__tojava__(hint); |
| if (view == Py.NoConversion) { |
| view = object.__tojava__(Object.class); |
| } |
| return view; |
| } |
| |
| @Override |
| public Object getWrappedObject() { |
| return object == null ? null : object.__tojava__(Object.class); |
| } |
| } |