Get Python `flatten` goody to have a configurable depth

[miorel]

See the discussion in https://github.com/code-chronicles-code/leetcode-curriculum/pull/370/files#r1730590840