parent
7caed48746
commit
665942653e
1 changed files with 46 additions and 2 deletions
|
@ -9,6 +9,7 @@ import urllib.parse
|
|||
from os import environ, mkdir
|
||||
from os.path import isdir, isfile, join
|
||||
from typing import Union
|
||||
from urllib.parse import urlparse
|
||||
|
||||
import argcomplete
|
||||
import requests
|
||||
|
@ -62,9 +63,20 @@ def output(response: list[dict], jsonout: bool = False):
|
|||
|
||||
# Define the runner for each command
|
||||
def run_add(url: str, jsonout: bool, headers: dict):
|
||||
print(url)
|
||||
parsed = split_url(url)
|
||||
response = requests.put(url, headers=headers)
|
||||
output(response.json(), jsonout)
|
||||
out = response.json()
|
||||
if isinstance(out, list):
|
||||
for record in out:
|
||||
if (
|
||||
record["data"] == parsed["data"]
|
||||
and record["name"] == parsed["name"]
|
||||
and record["rtype"] == parsed["rtype"]
|
||||
):
|
||||
output(record, jsonout)
|
||||
break
|
||||
else:
|
||||
output(out, jsonout)
|
||||
|
||||
|
||||
def run_complete(shell: Union[None, str]):
|
||||
|
@ -188,6 +200,38 @@ def setup_url(
|
|||
return url
|
||||
|
||||
|
||||
def split_url(url: str) -> dict:
|
||||
parsed = urlparse(url, allow_fragments=False)
|
||||
path = parsed.path
|
||||
query = parsed.query
|
||||
arguments: Union[None, list[str]] = query.split("&")
|
||||
path_arr = path.split("/")
|
||||
data: Union[None, str] = None
|
||||
name: Union[None, str] = None
|
||||
rtype: Union[None, str] = None
|
||||
ttl: Union[None, str] = None
|
||||
zone: Union[None, str] = None
|
||||
if len(path_arr) > 2:
|
||||
zone = path_arr[2]
|
||||
if len(path_arr) > 4:
|
||||
name = path_arr[4]
|
||||
if len(path_arr) > 5:
|
||||
rtype = path_arr[5]
|
||||
if len(path_arr) > 6:
|
||||
data = path_arr[6]
|
||||
if len(path_arr) > 7:
|
||||
ttl = path_arr[7]
|
||||
|
||||
return {
|
||||
"arguments": arguments,
|
||||
"data": data,
|
||||
"name": name,
|
||||
"rtype": rtype,
|
||||
"ttl": ttl,
|
||||
"zone": zone,
|
||||
}
|
||||
|
||||
|
||||
# Entry point to program
|
||||
def main() -> int:
|
||||
# Grab user input
|
||||
|
|
Loading…
Add table
Reference in a new issue