Utility.py

import json

from dataclasses import dataclass, asdict, field
from typing import Dict

DATA_FILE = 'data.json'

@dataclass
class RequestOptions:
    data: list[str] = field(default_factory=lambda: ["", ""])
    connection_message: str = ""
    cookies: str = ""
    origin: str = ""
    url: str = ""
    variables: str = ""

    def to_dict(self):
        return asdict(self)

    @staticmethod
    def from_dict(d: dict):
        return RequestOptions(**d)
    
def fetch_options(gate_name):
    dict = json.loads(open(DATA_FILE).read())

    if(dict.get(gate_name)):
        return RequestOptions.from_dict(dict[gate_name])

    return RequestOptions()

def save_options(gate_name, options, print_value=True):
    dict = json.loads(open(DATA_FILE).read())
    options_dict = options.to_dict()

    dict[gate_name] = options_dict

    if(print_value == True):
        print(options_dict)

    with open(DATA_FILE, "w") as file:
        file.write(json.dumps(dict, indent=2))

def map_to_options(gate_name, args):
    options = fetch_options(gate_name)

    if args.connection_message != "":
        options.connection_message = args.connection_message
    if args.cookies != "":
        options.cookies = args.cookies
    if args.origin != "":
        options.origin = args.origin
    if args.url != "":
        options.url = args.url
    if args.data and len(args.data) > 0:
        options.data[0] = args.data[0][0]
    if args.data and len(args.data) > 1:
        options.data[1] = args.data[1][0]

    save_options(gate_name, options)

def variables_to_dict(variables):
    variables_list = str.split(variables, " ")

    variables_dict = {}

    for key_value_pair in variables_list:
        key = str.split(key_value_pair, ':')[0]
        value = str.split(key_value_pair, ':')[1]

        variables_dict[key] = value

    return variables_dict