diff --git a/script/setup/Layout.py b/script/setup/Layout.py index 73b6f7c..23f11d0 100644 --- a/script/setup/Layout.py +++ b/script/setup/Layout.py @@ -59,6 +59,12 @@ for p in(puck_layout): if puck.block not in (_block_ids): _block_ids.append(puck.block) +def get_puck(number): + for p in _puck_list: + if number==p.number: + return p + return None + def get_pucks(block = None): ret = [] for p in _puck_list: