knotctl 0.1.0__tar.gz → 0.1.1__tar.gz

knotctl-0.1.1/.gitignore

@@ -0,0 +1,162 @@
+# ---> Python
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/#use-with-ide
+.pdm.toml
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+#.idea/
+

{knotctl-0.1.0 → knotctl-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: knotctl
-Version: 0.1.0
+Version: 0.1.1
 Summary: A CLI for knotapi.
 Author-email: Micke Nordin <hej@mic.ke>
 Requires-Python: >=3.9
@@ -208,9 +208,9 @@ options:
 ### LIST
 
 ```
-usage: knotctl list [-h] [-d DATA] [-n NAME] [-r RTYPE] -z ZONE
+usage: knotctl list [-h] [-d DATA] [-n NAME] [-r RTYPE] [-z ZONE]
 
-List records in the zone.
+List records.
 
 options:
   -h, --help            show this help message and exit
@@ -249,3 +249,26 @@ Available arguments are:
     ttl: New record time to live (TTL).
 ```
 
+### USER
+```
+usage: knotctl user [-h] [-u USERNAME]
+
+View user information.
+
+options:
+  -h, --help            show this help message and exit
+  -u USERNAME, --username USERNAME
+```
+
+### ZONE
+
+```
+usage: knotctl zone 
+
+List zones.
+
+options:
+  -h, --help            show this help message and exit
+```
+
+

{knotctl-0.1.0 → knotctl-0.1.1}/README.md

@@ -191,9 +191,9 @@ options:
 ### LIST
 
 ```
-usage: knotctl list [-h] [-d DATA] [-n NAME] [-r RTYPE] -z ZONE
+usage: knotctl list [-h] [-d DATA] [-n NAME] [-r RTYPE] [-z ZONE]
 
-List records in the zone.
+List records.
 
 options:
   -h, --help            show this help message and exit
@@ -231,3 +231,26 @@ Available arguments are:
     rtype: New record type.
     ttl: New record time to live (TTL).
 ```
+
+### USER
+```
+usage: knotctl user [-h] [-u USERNAME]
+
+View user information.
+
+options:
+  -h, --help            show this help message and exit
+  -u USERNAME, --username USERNAME
+```
+
+### ZONE
+
+```
+usage: knotctl zone 
+
+List zones.
+
+options:
+  -h, --help            show this help message and exit
+```
+

{knotctl-0.1.0 → knotctl-0.1.1}/knotctl/__init__.py

@@ -7,7 +7,7 @@ import os
 import sys
 import urllib.parse
 from os import environ, mkdir
-from os.path import isdir, isfile, islink, join, split
+from os.path import isdir, isfile, join
 from typing import Union
 from urllib.parse import urlparse
 
@@ -36,6 +36,9 @@ def error(description: str, error: str) -> list[dict]:
 
 
 def get_config(config_filename: str):
+    if not isfile(config_filename):
+        print("You need to configure knotctl before proceeding")
+        run_config(config_filename)
     with open(config_filename, "r") as fh:
         return yaml.safe_load(fh.read())
 
@@ -46,7 +49,8 @@ def nested_out(input, tabs="") -> str:
         string += "{}\n".format(input)
     elif isinstance(input, dict):
         for key, value in input.items():
-            string += "{}{}: {}".format(tabs, key, nested_out(value, tabs + " "))
+            string += "{}{}: {}".format(tabs, key,
+                                        nested_out(value, tabs + " "))
     elif isinstance(input, list):
         for entry in input:
             string += "{}\n{}".format(tabs, nested_out(entry, tabs + " "))
@@ -70,11 +74,9 @@ def run_add(url: str, jsonout: bool, headers: dict):
     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"]
-            ):
+            if (record["data"] == parsed["data"]
+                    and record["name"] == parsed["name"]
+                    and record["rtype"] == parsed["rtype"]):
                 output(record, jsonout)
                 break
     else:
@@ -94,29 +96,28 @@ def run_log(url: str, jsonout: bool, headers: dict):
             line = lines[index]
             index += 1
             cur_has_timestamp = line.startswith("[")
-            next_has_timestamp = index < len(lines) and lines[index].startswith(
-                "["
-            )
+            next_has_timestamp = index < len(
+                lines) and lines[index].startswith("[")
             # Simple case, just one line with timestamp
             if cur_has_timestamp and next_has_timestamp:
-                timestamp = line.split(']')[0].split('[')[1]
-                text = line.split(']')[1].lstrip(':').strip()
-                out.append({'timestamp': timestamp, 'text': text})
+                timestamp = line.split("]")[0].split("[")[1]
+                text = line.split("]")[1].lstrip(":").strip()
+                out.append({"timestamp": timestamp, "text": text})
                 text = ""
                 timestamp = ""
             # Start of multiline
             elif cur_has_timestamp:
-                timestamp = line.split(']')[0].split('[')[1]
-                text = line.split(']')[1].lstrip(':').strip()
+                timestamp = line.split("]")[0].split("[")[1]
+                text = line.split("]")[1].lstrip(":").strip()
             # End of multiline
             elif next_has_timestamp:
-                text += f'\n{line.strip()}'
-                out.append({'timestamp': timestamp, 'text': text})
+                text += f"\n{line.strip()}"
+                out.append({"timestamp": timestamp, "text": text})
                 text = ""
                 timestamp = ""
             # Middle of multiline
             else:
-                text += f'\n{line.strip()}'
+                text += f"\n{line.strip()}"
 
     else:
         out = string
@@ -144,7 +145,7 @@ def run_config(
     if current:
         if os.path.islink(config_filename):
             actual_path = os.readlink(config_filename)
-            print(actual_path.split('-')[-1])
+            print(actual_path.split("-")[-1])
         else:
             print("none")
         return
@@ -171,8 +172,7 @@ def run_config(
                 error(
                     "Can not configure without {}".format(need),
                     "No {}".format(need),
-                )
-            )
+                ))
             sys.exit(1)
         config[need] = input("Enter {}: ".format(need))
 
@@ -196,7 +196,10 @@ def run_delete(url: str, jsonout: bool, headers: dict):
     output(reply, jsonout)
 
 
-def run_list(url: str, jsonout: bool, headers: dict, ret=False) -> Union[None, str]:
+def run_list(url: str,
+             jsonout: bool,
+             headers: dict,
+             ret=False) -> Union[None, str]:
     response = requests.get(url, headers=headers)
     string = response.json()
     if ret:
@@ -210,6 +213,22 @@ def run_update(url: str, jsonout: bool, headers: dict):
     output(response.json(), jsonout)
 
 
+def run_zone(url: str,
+             jsonout: bool,
+             headers: dict,
+             ret=False) -> Union[None, str]:
+    response = requests.get(url, headers=headers)
+    zones = response.json()
+    for zone in zones:
+        del zone["records"]
+    string = zones
+
+    if ret:
+        return string
+    else:
+        output(string, jsonout)
+
+
 # Set up the url
 def setup_url(
     baseurl: str,
@@ -248,16 +267,14 @@ def setup_url(
             error(
                 "ttl only makes sense with rtype, name and zone",
                 "Missing parameter",
-            )
-        )
+            ))
         sys.exit(1)
     if rtype and (not name or not zone):
         output(
             error(
                 "rtype only makes sense with name and zone",
                 "Missing parameter",
-            )
-        )
+            ))
         sys.exit(1)
     if name and not zone:
         output(error("name only makes sense with a zone", "Missing parameter"))
@@ -297,8 +314,7 @@ def split_url(url: str) -> dict:
     }
 
 
-# Entry point to program
-def main() -> int:
+def get_parser() -> dict:
     description = """Manage DNS records with knot dns rest api:
         * https://gitlab.nic.cz/knot/knot-dns-rest"""
 
@@ -366,18 +382,22 @@ def main() -> int:
     subparsers.add_parser("auditlog", description=auditlog_description)
 
     changelog_description = "View the changelog of a zone."
-    changelogcmd = subparsers.add_parser("changelog", description=changelog_description)
+    changelogcmd = subparsers.add_parser("changelog",
+                                         description=changelog_description)
     changelogcmd.add_argument("-z", "--zone", required=True)
 
     complete_description = "Generate shell completion script."
-    completecmd = subparsers.add_parser("completion", description=complete_description)
+    completecmd = subparsers.add_parser("completion",
+                                        description=complete_description)
     completecmd.add_argument("-s", "--shell")
 
     config_description = "Configure access to knot-dns-rest-api."
     configcmd = subparsers.add_parser("config", description=config_description)
     configcmd.add_argument("-b", "--baseurl")
     configcmd.add_argument("-c", "--context")
-    configcmd.add_argument("-C", "--current", action=argparse.BooleanOptionalAction)
+    configcmd.add_argument("-C",
+                           "--current",
+                           action=argparse.BooleanOptionalAction)
     configcmd.add_argument("-p", "--password")
     configcmd.add_argument("-u", "--username")
 
@@ -388,22 +408,23 @@ def main() -> int:
     deletecmd.add_argument("-r", "--rtype")
     deletecmd.add_argument("-z", "--zone", required=True)
 
-    list_description = "List records in the zone."
+    list_description = "List records."
     listcmd = subparsers.add_parser("list", description=list_description)
     listcmd.add_argument("-d", "--data")
     listcmd.add_argument("-n", "--name")
     listcmd.add_argument("-r", "--rtype")
-    listcmd.add_argument("-z", "--zone", required=True)
+    listcmd.add_argument("-z", "--zone", required=False)
+
+    user_description = "View user information."
+    usercmd = subparsers.add_parser("user", description=user_description)
+    usercmd.add_argument("-u", "--username", default=None)
 
     update_description = (
-        "Update a record in the zone. The record must exist in the zone.\n"
-    )
+        "Update a record in the zone. The record must exist in the zone.\n")
     update_description += (
-        "In this case --data, --name, --rtype and --ttl switches are used\n"
-    )
+        "In this case --data, --name, --rtype and --ttl switches are used\n")
     update_description += (
-        "for searching for the appropriate record, while the --argument\n"
-    )
+        "for searching for the appropriate record, while the --argument\n")
     update_description += "switches are used for updating the record."
     update_epilog = """Available arguments are:
     data: New record data.
@@ -421,7 +442,8 @@ def main() -> int:
         "--argument",
         action="append",
         metavar="KEY=VALUE",
-        help="Specify key - value pairs to be updated: name=dns1.example.com. or data=127.0.0.1 for example. --argument can be repeated",
+        help="Specify key - value pairs to be updated: name=dns1.example.com."
+        + " or data=127.0.0.1 for example. --argument can be repeated",
         required=True,
     )
     updatecmd.add_argument("-d", "--data", required=True)
@@ -430,7 +452,68 @@ def main() -> int:
     updatecmd.add_argument("-t", "--ttl")
     updatecmd.add_argument("-z", "--zone", required=True)
 
+    zone_description = "View zones."
+    subparsers.add_parser("zone", description=zone_description)
+
     argcomplete.autocomplete(parser)
+
+    return parser
+
+
+def get_token(config) -> str:
+    # Authenticate
+    baseurl = config["baseurl"]
+    username = config["username"]
+    password = config["password"]
+    basic = HTTPBasicAuth(username, password)
+    response = requests.get(baseurl + "/user/login", auth=basic)
+    token = ""
+    try:
+        token = response.json()["token"]
+    except KeyError:
+        output(response.json())
+    except requests.exceptions.JSONDecodeError:
+        output(
+            error("Could not decode api response as JSON", "Could not decode"))
+    return token
+
+
+def run(url, args, headers, baseurl, parser, username):
+    try:
+        if args.command == "add":
+            run_add(url, args.json, headers)
+        elif args.command == "delete":
+            run_delete(url, args.json, headers)
+        elif args.command == "list":
+            run_list(url, args.json, headers)
+        elif args.command == "update":
+            run_update(url, args.json, headers)
+        elif args.command == "user":
+            url = baseurl + f"/user/info/{username}"
+            run_list(url, args.json, headers)
+        elif args.command == "auditlog":
+            url = baseurl + "/user/auditlog"
+            run_log(url, args.json, headers)
+        elif args.command == "changelog":
+            url = baseurl + f"/zones/changelog/{args.zone.rstrip('.')}"
+            run_log(url, args.json, headers)
+        elif args.command == "zone":
+            url = baseurl + "/zones"
+            run_zone(url, args.json, headers)
+        else:
+            parser.print_help(sys.stderr)
+            return 2
+    except requests.exceptions.RequestException as e:
+        output(error(e, "Could not connect to server"))
+    except (RequestsJSONDecodeError, SimplejsonJSONDecodeError):
+        output(
+            error("Could not decode api response as JSON", "Could not decode"))
+    return 0
+
+
+# Entry point to program
+def main() -> int:
+    parser = get_parser()
     args = parser.parse_args()
     if args.command == "completion":
         run_complete(args.shell)
@@ -445,35 +528,27 @@ def main() -> int:
 
     if args.command == "config":
         run_config(
-            config_filename, args.context, args.baseurl, args.username, args.password, args.current
+            config_filename,
+            args.context,
+            args.baseurl,
+            args.username,
+            args.password,
+            args.current,
         )
         return 0
 
-    if not isfile(config_filename):
-        print("You need to configure knotctl before proceeding")
-        run_config(config_filename)
-
     config = get_config(config_filename)
     baseurl = config["baseurl"]
-    username = config["username"]
-    password = config["password"]
-
-    # Authenticate
-    basic = HTTPBasicAuth(username, password)
-    response = requests.get(baseurl + "/user/login", auth=basic)
-    try:
-        token = response.json()["token"]
-    except KeyError:
-        output(response.json())
-        return 1
-    except requests.exceptions.JSONDecodeError:
-        output(error("Could not decode api response as JSON", "Could not decode"))
+    token = get_token(config)
+    if token == "":
+        print("Could not get token, exiting")
         return 1
     headers = {"Authorization": "Bearer {}".format(token)}
 
     # Route based on command
     url = ""
     ttl = None
+    user = config["username"]
     if "ttl" in args:
         ttl = args.ttl
     if args.command != "update":
@@ -486,7 +561,10 @@ def main() -> int:
         soa_url = setup_url(baseurl, None, None, zname, "SOA", None, args.zone)
         soa_json = run_list(soa_url, True, headers, ret=True)
         ttl = soa_json[0]["ttl"]
-    if args.command in ["auditlog", "changelog"]:
+    if args.command == "user":
+        if args.username:
+            user = args.username
+    if args.command in ["auditlog", "changelog", "user", "zone"]:
         pass
     else:
         try:
@@ -503,29 +581,7 @@ def main() -> int:
             parser.print_help(sys.stderr)
             return 1
 
-    try:
-        if args.command == "add":
-            run_add(url, args.json, headers)
-        elif args.command == "delete":
-            run_delete(url, args.json, headers)
-        elif args.command == "list":
-            run_list(url, args.json, headers)
-        elif args.command == "update":
-            run_update(url, args.json, headers)
-        elif args.command == "auditlog":
-            url = baseurl + "/user/auditlog"
-            run_log(url, args.json, headers)
-        elif args.command == "changelog":
-            url = baseurl + f"/zones/changelog/{args.zone.rstrip('.')}"
-            run_log(url, args.json, headers)
-        else:
-            parser.print_help(sys.stderr)
-            return 2
-    except requests.exceptions.RequestException as e:
-        output(error(e, "Could not connect to server"))
-    except (RequestsJSONDecodeError, SimplejsonJSONDecodeError):
-        output(error("Could not decode api response as JSON", "Could not decode"))
-    return 0
+    return run(url, args, headers, baseurl, parser, user)
 
 
 if __name__ == "__main__":

{knotctl-0.1.0 → knotctl-0.1.1}/pyproject.toml

@@ -16,7 +16,7 @@ classifiers=[
         "Operating System :: OS Independent",
         ]
 requires-python= ">=3.9"
-version = "0.1.0"
+version = "0.1.1"
 
 dependencies = [
   "argcomplete==2.0.0",

knotctl-0.1.1/requirements.txt

@@ -0,0 +1,4 @@
+argcomplete==2.0.0
+pyyaml==6.0.1
+requests==2.27.1
+simplejson==3.17.6

knotctl-0.1.1/stdeb.cfg

@@ -0,0 +1,4 @@
+[DEFAULT]
+Depends3: python3-argcomplete, python3-requests, python3-simplejson, python3-yaml
+Debian-Version: 1
+Package3: knotctl