collect: move vin, interval, host, port, log_dir to credentials file
Remove --vin, --interval/-i, --output/-o, --host, --port from argparse. All are now read from the credentials JSON (interval_s, host, port, log_dir with sensible defaults). CLI flags -u/-p/-d still override the file; log_dir replaces the old --output/auto-logs logic — logs_dir is always used, no fixed output path option. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
+43
-76
@@ -8,42 +8,39 @@ the WeConnect domain hierarchy: domain → status-object → field.
|
|||||||
|
|
||||||
Usage examples
|
Usage examples
|
||||||
--------------
|
--------------
|
||||||
# Credentials from env vars, all defaults:
|
# Credentials file with all settings:
|
||||||
|
python collect.py -c credentials/alex.json
|
||||||
|
|
||||||
|
# Override username/password via env vars:
|
||||||
export WC_USER=me@example.com WC_PASS=secret
|
export WC_USER=me@example.com WC_PASS=secret
|
||||||
python collect.py
|
python collect.py -c credentials/alex.json
|
||||||
|
|
||||||
# Credentials from a JSON file (output auto-named under ./logs/):
|
# Override domains on the CLI:
|
||||||
python collect.py -c creds.json -i 600
|
python collect.py -c credentials/alex.json -d charging,measurements
|
||||||
|
|
||||||
# Credentials from a JSON file, explicit output path:
|
|
||||||
python collect.py -c creds.json -i 600 -o id3.json
|
|
||||||
|
|
||||||
# Custom domains, 10-minute interval:
|
|
||||||
python collect.py -u me@example.com -p secret \\
|
|
||||||
-d charging,measurements,readiness -i 600 -o id3.json
|
|
||||||
|
|
||||||
# Everything, verbose, keep last 1 000 records:
|
|
||||||
python collect.py -d all -v --max-records 1000
|
|
||||||
|
|
||||||
Available domains: charging, climatisation, measurements, readiness, parking, access
|
Available domains: charging, climatisation, measurements, readiness, parking, access
|
||||||
|
|
||||||
Push server
|
|
||||||
-----------
|
|
||||||
The collector listens on --host/--port (default 0.0.0.0:9999).
|
|
||||||
Each connected TCP client receives every new snapshot as a single
|
|
||||||
line of JSON (newline-delimited) the moment it is collected.
|
|
||||||
Connect with: nc localhost 9999 or any TCP client that reads lines.
|
|
||||||
|
|
||||||
Credentials file format (JSON)
|
Credentials file format (JSON)
|
||||||
-------------------------------
|
-------------------------------
|
||||||
{
|
{
|
||||||
|
"credentials": {
|
||||||
"username": "me@example.com",
|
"username": "me@example.com",
|
||||||
"password": "secret",
|
"password": "secret"
|
||||||
|
},
|
||||||
"vin": "WVWZZZE1ZMP123456",
|
"vin": "WVWZZZE1ZMP123456",
|
||||||
"domains": ["charging", "measurements", "readiness"]
|
"domains": ["charging", "measurements", "readiness"],
|
||||||
|
"interval_s": 300,
|
||||||
|
"host": "0.0.0.0",
|
||||||
|
"port": 9999,
|
||||||
|
"log_dir": "./logs"
|
||||||
}
|
}
|
||||||
All four keys are optional; domains may also be a comma-separated string.
|
|
||||||
Explicit CLI flags and env vars take precedence.
|
Push server
|
||||||
|
-----------
|
||||||
|
The collector listens on host/port from the credentials file (default 0.0.0.0:9999).
|
||||||
|
Each connected TCP client receives every new snapshot as a single
|
||||||
|
line of JSON (newline-delimited) the moment it is collected.
|
||||||
|
Connect with: nc localhost 9999 or any TCP client that reads lines.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
import argparse
|
import argparse
|
||||||
@@ -85,16 +82,11 @@ def run(args: argparse.Namespace) -> None:
|
|||||||
vin = vehicle.vin.value
|
vin = vehicle.vin.value
|
||||||
log.info("Vehicle: %s", vin)
|
log.info("Vehicle: %s", vin)
|
||||||
|
|
||||||
if args.output:
|
logs_dir = Path(args.log_dir)
|
||||||
out = Path(args.output)
|
logs_dir.mkdir(parents=True, exist_ok=True)
|
||||||
logs_dir = None
|
|
||||||
else:
|
|
||||||
logs_dir = Path("logs")
|
|
||||||
logs_dir.mkdir(exist_ok=True)
|
|
||||||
out = auto_out_path(logs_dir, vin)
|
out = auto_out_path(logs_dir, vin)
|
||||||
|
|
||||||
push_clients, push_lock, push_store_ref = network.start_push_server(args.host, args.port)
|
push_clients, push_lock, push_store_ref = network.start_push_server(args.host, args.port)
|
||||||
if logs_dir is not None:
|
|
||||||
push_store_ref["preloaded"] = load_last_24h_records(logs_dir, vin)
|
push_store_ref["preloaded"] = load_last_24h_records(logs_dir, vin)
|
||||||
if push_store_ref["preloaded"]:
|
if push_store_ref["preloaded"]:
|
||||||
log.info("Pre-loaded %d record(s) from the last 24 h", len(push_store_ref["preloaded"]))
|
log.info("Pre-loaded %d record(s) from the last 24 h", len(push_store_ref["preloaded"]))
|
||||||
@@ -111,7 +103,6 @@ def run(args: argparse.Namespace) -> None:
|
|||||||
|
|
||||||
try:
|
try:
|
||||||
while True:
|
while True:
|
||||||
if logs_dir is not None:
|
|
||||||
today = datetime.now(timezone.utc).date()
|
today = datetime.now(timezone.utc).date()
|
||||||
if today != current_day:
|
if today != current_day:
|
||||||
out = auto_out_path(logs_dir, vin)
|
out = auto_out_path(logs_dir, vin)
|
||||||
@@ -152,67 +143,36 @@ def main() -> None:
|
|||||||
creds.add_argument(
|
creds.add_argument(
|
||||||
"-c", "--credentials",
|
"-c", "--credentials",
|
||||||
metavar="FILE",
|
metavar="FILE",
|
||||||
help=(
|
help="JSON file with connection settings (see format above).",
|
||||||
"JSON file with keys: username, password, vin, domains. "
|
|
||||||
"Values are overridden by explicit CLI flags and env vars."
|
|
||||||
),
|
|
||||||
)
|
)
|
||||||
creds.add_argument(
|
creds.add_argument(
|
||||||
"-u", "--username",
|
"-u", "--username",
|
||||||
default=os.environ.get("WC_USER"),
|
default=os.environ.get("WC_USER"),
|
||||||
help="WeConnect / MyVolkswagen email",
|
help="WeConnect / MyVolkswagen email (overrides credentials file)",
|
||||||
)
|
)
|
||||||
creds.add_argument(
|
creds.add_argument(
|
||||||
"-p", "--password",
|
"-p", "--password",
|
||||||
default=os.environ.get("WC_PASS"),
|
default=os.environ.get("WC_PASS"),
|
||||||
help="WeConnect password",
|
help="WeConnect password (overrides credentials file)",
|
||||||
)
|
)
|
||||||
|
|
||||||
p.add_argument("--vin", help="Vehicle VIN (uses first vehicle when omitted)")
|
|
||||||
p.add_argument(
|
p.add_argument(
|
||||||
"-d", "--domains",
|
"-d", "--domains",
|
||||||
default=None,
|
default=None,
|
||||||
metavar="DOMAIN[,DOMAIN…]",
|
metavar="DOMAIN[,DOMAIN…]",
|
||||||
help=(
|
help=(
|
||||||
"Comma-separated domains to collect, or 'all'. "
|
"Comma-separated domains to collect, or 'all' (overrides credentials file). "
|
||||||
f"Available: {', '.join(ALL_DOMAINS)}. "
|
f"Available: {', '.join(ALL_DOMAINS)}. "
|
||||||
"Default: charging,measurements,readiness"
|
"Default: charging,measurements,readiness"
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
p.add_argument(
|
|
||||||
"-i", "--interval",
|
|
||||||
type=int,
|
|
||||||
default=300,
|
|
||||||
metavar="SECONDS",
|
|
||||||
help="Collection interval in seconds (default: 300)",
|
|
||||||
)
|
|
||||||
p.add_argument(
|
|
||||||
"-o", "--output",
|
|
||||||
default=None,
|
|
||||||
metavar="FILE",
|
|
||||||
help="Output file (default: logs/YYYY_MM_DD_HH_MM_SS_<VIN>.json)",
|
|
||||||
)
|
|
||||||
p.add_argument(
|
p.add_argument(
|
||||||
"--max-records",
|
"--max-records",
|
||||||
type=int,
|
type=int,
|
||||||
default=None,
|
default=None,
|
||||||
metavar="N",
|
metavar="N",
|
||||||
help="Keep only the last N records in the file (default: unlimited)",
|
help="Keep only the last N records per file (default: unlimited)",
|
||||||
)
|
)
|
||||||
|
|
||||||
srv = p.add_argument_group("push server")
|
|
||||||
srv.add_argument(
|
|
||||||
"--host",
|
|
||||||
default="0.0.0.0",
|
|
||||||
help="Bind address for the push server (default: 0.0.0.0)",
|
|
||||||
)
|
|
||||||
srv.add_argument(
|
|
||||||
"--port",
|
|
||||||
type=int,
|
|
||||||
default=9999,
|
|
||||||
help="TCP port for the push server (default: 9999)",
|
|
||||||
)
|
|
||||||
|
|
||||||
p.add_argument("-v", "--verbose", action="store_true", help="Enable debug logging")
|
p.add_argument("-v", "--verbose", action="store_true", help="Enable debug logging")
|
||||||
p.add_argument(
|
p.add_argument(
|
||||||
"--list-domains",
|
"--list-domains",
|
||||||
@@ -226,6 +186,7 @@ def main() -> None:
|
|||||||
print("\n".join(ALL_DOMAINS))
|
print("\n".join(ALL_DOMAINS))
|
||||||
sys.exit(0)
|
sys.exit(0)
|
||||||
|
|
||||||
|
creds_data: dict = {}
|
||||||
if args.credentials:
|
if args.credentials:
|
||||||
creds_path = Path(args.credentials)
|
creds_path = Path(args.credentials)
|
||||||
if not creds_path.exists():
|
if not creds_path.exists():
|
||||||
@@ -234,26 +195,32 @@ def main() -> None:
|
|||||||
creds_data = json.loads(creds_path.read_text())
|
creds_data = json.loads(creds_path.read_text())
|
||||||
except json.JSONDecodeError as exc:
|
except json.JSONDecodeError as exc:
|
||||||
p.error(f"Invalid JSON in credentials file: {exc}")
|
p.error(f"Invalid JSON in credentials file: {exc}")
|
||||||
|
|
||||||
|
# username / password: env / CLI flag > credentials.username > credentials file root
|
||||||
|
nested = creds_data.get("credentials", {})
|
||||||
if args.username is None:
|
if args.username is None:
|
||||||
args.username = creds_data.get("username")
|
args.username = nested.get("username") or creds_data.get("username")
|
||||||
if args.password is None:
|
if args.password is None:
|
||||||
args.password = creds_data.get("password")
|
args.password = nested.get("password") or creds_data.get("password")
|
||||||
if args.vin is None:
|
|
||||||
args.vin = creds_data.get("vin")
|
args.vin = creds_data.get("vin")
|
||||||
|
|
||||||
if args.domains is None:
|
if args.domains is None:
|
||||||
raw_domains = creds_data.get("domains")
|
raw_domains = creds_data.get("domains")
|
||||||
if isinstance(raw_domains, list):
|
if isinstance(raw_domains, list):
|
||||||
args.domains = ",".join(raw_domains)
|
args.domains = ",".join(raw_domains)
|
||||||
else:
|
else:
|
||||||
args.domains = raw_domains
|
args.domains = raw_domains or "charging,measurements,readiness"
|
||||||
|
|
||||||
if args.domains is None:
|
args.interval = int(creds_data.get("interval_s", 300))
|
||||||
args.domains = "charging,measurements,readiness"
|
args.host = creds_data.get("host", "0.0.0.0")
|
||||||
|
args.port = int(creds_data.get("port", 9999))
|
||||||
|
args.log_dir = creds_data.get("log_dir", "./logs")
|
||||||
|
|
||||||
if not args.username or not args.password:
|
if not args.username or not args.password:
|
||||||
p.error(
|
p.error(
|
||||||
"Username and password are required. "
|
"Username and password are required. "
|
||||||
"Pass them with -u/-p or set WC_USER / WC_PASS environment variables."
|
"Provide them in the credentials file or via -u/-p / WC_USER / WC_PASS."
|
||||||
)
|
)
|
||||||
|
|
||||||
run(args)
|
run(args)
|
||||||
|
|||||||
Reference in New Issue
Block a user