В таком контексте сравнительный оборот играет роль несогласованного определения: конструкцию (какую?) как на рисунке. Такие случаи не описаны в справочниках по пунктуации. Уместно будет не обособлять этот оборот, поскольку существительное конструкция не имеет других (согласованных) определений.
Правильно: по возвращении, по прибытии. Предлог по в значении 'после чего-либо' употребляется с предложным падежом.
Пример относится к случаям, когда «перечисляемые разновидности предметов или явлений внутренне связаны» (см. пункт 4 параграфа 194 «Справочника по правописанию, произношению, литературному редактированию»). Существительное здесь употребляется в единственном числе: Нанесите тени на верхнее и нижнее веко.
В значении по «после» верно: по окончании, по истечении.
Если имеется в виду день недели, оба варианта верны.
