Python Wrap Function In Timeout