From 512e438ca04a016d8973a80ffd9368ca9074ad7d Mon Sep 17 00:00:00 2001 From: Azalea Gui Date: Sat, 4 Feb 2023 02:34:36 -0500 Subject: [PATCH] [O] Use toml instead of environ to store user and pass --- auth.toml | 2 ++ mobile.py | 10 +++++++++- requirements.txt | 1 + 3 files changed, 12 insertions(+), 1 deletion(-) create mode 100644 auth.toml diff --git a/auth.toml b/auth.toml new file mode 100644 index 0000000..f05411b --- /dev/null +++ b/auth.toml @@ -0,0 +1,2 @@ +login = "meow@example.com" +passwd = "qwq" diff --git a/mobile.py b/mobile.py index a1ea4a3..8ca79b8 100644 --- a/mobile.py +++ b/mobile.py @@ -1,3 +1,4 @@ +import argparse import os import sys import threading @@ -9,6 +10,7 @@ from typing import Literal from urllib.parse import urlparse, parse_qs import requests +import toml from hypy_utils import json_stringify from hypy_utils.serializer import pickle_decode, pickle_encode @@ -219,7 +221,13 @@ class Handler(BaseHTTPRequestHandler): if __name__ == '__main__': - print(login(os.environ['user'], os.environ['pass'])) + par = argparse.ArgumentParser("Superbuy Order Helper") + par.add_argument("-a", "--auth", help="Auth config path", default="auth.toml") + args = par.parse_args() + + auth = toml.loads(Path(args.auth).read_text()) + + print(login(auth['login'], auth['passwd'])) # Start HTTP server asyncronously server = HTTPServer(("127.0.0.1", 12842), Handler) diff --git a/requirements.txt b/requirements.txt index c355c6a..dcb1579 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,3 +3,4 @@ fastapi hypy_utils uvicorn pysocks +toml