sawyer/chromedriver-http-proxy
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(config): allow configuration with environment variables

Browse source
This commit is contained in:
Sawyer 2025-08-05 18:47:06 -05:00
parent 997ab17dd8
commit 244aea41cc
1 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

11
config.py
View file

@ -1,3 +1,4 @@
import os
import argparse import argparse
from dataclasses import dataclass from dataclasses import dataclass
@ -24,7 +25,7 @@ def get_configs():
help="Port the proxy runs on.", help="Port the proxy runs on.",
required=False, required=False,
type=int, type=int,
default=32323 default=os.getenv("PROXY_PORT", 32323)
) )
parser.add_argument( parser.add_argument(
@ -32,7 +33,7 @@ def get_configs():
help="Host the proxy to runs on.", help="Host the proxy to runs on.",
required=False, required=False,
type=str, type=str,
default="0.0.0.0" default=os.getenv("PROXY_HOST", "0.0.0.0")
) )
parser.add_argument( parser.add_argument(
@ -40,7 +41,7 @@ def get_configs():
help="Seconds to wait before returning content.", help="Seconds to wait before returning content.",
required=False, required=False,
type=float, type=float,
default=10 default=os.getenv("SCRAPER_WAIT_TIME", 10)
) )
parser.add_argument( parser.add_argument(
@ -48,7 +49,7 @@ def get_configs():
help="Whether or not to run Chrome headless.", help="Whether or not to run Chrome headless.",
required=False, required=False,
type=bool, type=bool,
default=True default=os.getenv("SCRAPER_HEADLESS", True)
) )
parser.add_argument( parser.add_argument(
@ -56,7 +57,7 @@ def get_configs():
help="Chrome user agent. Changing with the current ChromeDriver version recommended.", help="Chrome user agent. Changing with the current ChromeDriver version recommended.",
required=False, required=False,
type=str, type=str,
default="Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Safari/537.36" default=os.getenv("SCRAPER_USER_AGENT", "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Safari/537.36")
) )
args = parser.parse_args() args = parser.parse_args()