Merge pull request #1768 from mhils/organize-examples

Organize Examples

1 files changed, 24 insertions, 0 deletions

diff --git a/examples/complex/change_upstream_proxy.py b/examples/complex/change_upstream_proxy.py
new file mode 100644
index 00000000..49d5379f
--- /dev/null
+++ b/examples/complex/change_upstream_proxy.py
@@ -0,0 +1,24 @@
+# This scripts demonstrates how mitmproxy can switch to a second/different upstream proxy
+# in upstream proxy mode.
+#
+# Usage: mitmdump -U http://default-upstream-proxy.local:8080/ -s change_upstream_proxy.py
+#
+# If you want to change the target server, you should modify flow.request.host and flow.request.port
+
+
+def proxy_address(flow):
+    # Poor man's loadbalancing: route every second domain through the alternative proxy.
+    if hash(flow.request.host) % 2 == 1:
+        return ("localhost", 8082)
+    else:
+        return ("localhost", 8081)
+
+
+def request(flow):
+    if flow.request.method == "CONNECT":
+        # If the decision is done by domain, one could also modify the server address here.
+        # We do it after CONNECT here to have the request data available as well.
+        return
+    address = proxy_address(flow)
+    if flow.live:
+        flow.live.change_upstream_proxy_server(address)