# -*- python -*-
#
# OpenAlea.StdLib
#
# Copyright 2006-2023 INRIA - CIRAD - INRA
#
# Distributed under the Cecill-C License.
# See accompanying file LICENSE.txt or copy at
# http://www.cecill.info/licences/Licence_CeCILL-C_V1-en.html
#
# OpenAlea WebSite : http://openalea.gforge.inria.fr
#
################################################################################
""" Python Nodes """
__license__ = "Cecill-C"
__revision__ = " $Id$ "
import os
import operator
from openalea.core import *
[docs]
def py_ifelse(c=True, a=None, b=None):
""" Return a if c is true else b """
return a if bool(c) else b
[docs]
def keys(obj):
""" call keys() on obj """
ret = list(obj.keys())
return (ret,)
[docs]
def values(obj):
""" call values() on obj """
ret = list(obj.values())
return (ret,)
[docs]
def items(obj):
""" call items() on obj """
ret = list(obj.items())
return (ret,)
[docs]
def pyrange(start=0, stop=0, step=1):
""" range(start, stop, step) """
return (list(range(start, stop, step)),)
[docs]
def pyenumerate(obj):
""" enumerate(iterable) -> iterator for index, value of iterable """
return (list(enumerate(obj)),)
[docs]
def pylen(obj):
""" len(obj) """
f = len(obj)
return (f, )
# def py_getitem(obj, index):
# """ obj.__getitem__ """
# return operator.getitem(obj, index)
[docs]
def py_setitem(obj, index, value):
""" obj.__setitem__ """
operator.setitem(obj, index, value)
return (obj, )
[docs]
def py_delitem(obj, key):
""" call __delitem__ on obj"""
operator.delitem(obj, key)
return (obj,)
[docs]
def py_print(x):
""" Print to the console """
print(x)
return (x,)
[docs]
def py_method(obj=None, member_name="", args=()):
""" call obj.name(\*args) """
m = getattr(obj, member_name)
m(*args)
return obj
[docs]
def py_getattr(items, member_name):
""" getattr """
return getattr(items, member_name)
[docs]
def py_setattr(obj, member_name, value_str):
setattr(obj, member_name, eval(value_str))
return obj,
[docs]
def py_eval(str):
""" Python eval """
return (eval(str),)
[docs]
def py_exec(str):
""" Python exec """
l = {}
exec(str, globals(), l)
return l,
[docs]
def py_zip(s1=(), s2=()):
__doc__ = zip.__doc__
return (list(zip(s1, s2)),)
[docs]
def py_zip2(args):
__doc__ = zip.__doc__
return (list(zip(*args)),)
[docs]
def py_flatten(obj=[]):
""" Flatten nested list """
tobeflatten = False
for v in iter(obj):
if hasattr(v, '__iter__'):
tobeflatten = True
break
if not tobeflatten:
return (obj,)
else:
nl = []
for v in iter(obj):
for x in iter(py_flatten(v)[0]):
nl.append(x)
return (nl,)
[docs]
class PyGetItem(Node):
def __call__(self, inputs):
""" Python __getitem__ method like obj[key] """
obj, key = inputs[0:2]
if isinstance(obj, dict):
if obj and (key not in obj):
key = next(iter(obj.keys()))
# send an event to the node to set the 2nd inputs
self.set_input(1, key)
return obj[key],
else:
return obj[key],
[docs]
def pysum(sequence):
return (sum(sequence), )
[docs]
def pymean(sequence):
return (sum(sequence) / float(len(sequence)), )
# DEPRECATED
[docs]
def py_fwrite(x="", filename="", mode="w"):
""" Write to a file """
print(("This node is DEPRECATED. Use %s instead" % ("Catalog.File.write")))
f = open(filename, mode)
f.write(x)
f.close()
[docs]
class FileRead(object):
""" Read a file """
def __init__(self):
self.mtime = 0 # modification time
self.filename = ''
self.s = ''
def __call__(self, filename=""):
print(("This node is DEPRECATED. Use %s instead" % ("Catalog.File.read")))
try:
mtime = os.stat(filename).st_mtime
except:
mtime = 0
if(filename != self.filename or
mtime != self.mtime):
self.filename = filename
self.mtime = mtime
f = open(filename, 'r')
self.s = f.read()
f.close()
return self.s
