feat: Allow customization of socket options

[dustinbyrne]

Adds enable_keep_alive() and set_socket_options() to configure TCP keepalive on HTTP connections, preventing idle connections from being dropped by network infrastructure.

Additionally adds disable_connection_reuse() to prevent connection pooling altogether.

Usage

import posthog

posthog.enable_keep_alive()
posthog.api_key = '...'

Or with custom socket options:

import posthog

posthog.set_socket_options([...])
posthog.api_key = '...'

Testing

If you'd like to try this branch, install it to your project.

requirements.txt

posthog @ git+https://github.com/PostHog/posthog-python.git@feat/custom-socket-options

pyproject.toml (Poetry):

[tool.poetry.dependencies]
posthog = { git = "https://github.com/PostHog/posthog-python.git", branch = "feat/custom-socket-options" }

pyproject.toml (pip/setuptools):

[project]
dependencies = [
	"posthog @ git+https://github.com/PostHog/posthog-python.git@feat/custom-socket-options",
]

[greptile-apps]

_{3 files reviewed, 3 comments}

_{Edit Code Review Agent Settings | Greptile}

[dmarticus]

Great job finding this; I'm impressed that you tracked this down.

In addition to my comments, 2 more questions:

1. Given that the specific use case around this is AWS NAT timeouts (and I imagine this might also apply to azure/gcp load balancers as well), can we document this in the options as well?
2. Should enable_keep_alive() be idempotent? Currently calling it twice rebuilds the session unnecessarily.