Hello,
A bit late to the discussion, but if anyone is still interested, we managed to come up with a lightweight focus stacking algorithm based on Python. I am aware that the discussion was also about stitching the images first, which our code is not capable of doing. But if someone is only interested in focus stacking, you might want to check out our code.
Cheers
Silvan