Semi-aquatic and ball python don't really go together. What you are describing is great for a green tree python. Yellow anaconda gets too big.

Are you set on ball python? If so, then you can still achieve naturalistic like this:



Not as natural as what you're describing but still gives a nice look.