pyntcli 0.1.46__py3-none-any.whl → 0.1.48__py3-none-any.whl

build/lib/build/lib/build/lib/build/lib/build/lib/pyntcli/commands/root.py

@@ -1,58 +0,0 @@
-import argparse
-import sys
-
-from pyntcli.auth import login
-from pyntcli.ui import ui_thread
-from pyntcli.analytics import send as analytics
-
-def root_usage():
-    return ui_thread.PrinterText("Usage:",style=ui_thread.PrinterText.HEADER) \
-        .with_line("\tpynt [COMMAND] [OPTIONS]") \
-        .with_line("") \
-        .with_line("Commands:",style=ui_thread.PrinterText.HEADER) \
-        .with_line("\tpostman - integration with postman, run scan from pynt postman collection") \
-        .with_line("\tnewman -  run postman collection from the CLI") \
-        .with_line("\tpynt-id - view your pynt-id to use when running pynt in CI pipeline") \
-        .with_line("") \
-        .with_line("Run pynt [COMMAND] -h to get help on a specific command",style=ui_thread.PrinterText.INFO)
-
-def usage(*args):
-    ui_thread.print(root_usage())
-
-class BaseCommand():
-    def __init__(self) -> None:
-        self.base: argparse.ArgumentParser = None
-        self.subparser: argparse._SubParsersAction = None
-
-    def cmd(self):
-        if self.base:
-            return self.base
-        
-        self.base = argparse.ArgumentParser(prog="pynt")
-        self.base.print_usage = usage
-        self.base.print_help = usage
-        return self.base 
-   
-    def add_base_arguments(self, parser):
-        parser.add_argument("--insecure", default=False, required=False, action='store_true',help="use when target uses self signed certificates")
-        parser.add_argument("--dev-flags", type=str,default="",  help=argparse.SUPPRESS)
-        parser.add_argument("--host-ca", type=str, default="")
-
-    def get_subparser(self) -> argparse._SubParsersAction:
-        if self.subparser is None:
-            self.subparser = self.base.add_subparsers(help="", dest="command")
-        
-        return self.subparser
-
-    def run_cmd(self, args: argparse.Namespace):
-        analytics.emit(analytics.LOGIN_START)
-
-        if login.should_login():
-            l = login.Login().login()
-        else:
-            login.refresh_token()  
-        
-        analytics.emit(analytics.LOGIN_DONE)
-        user_id = login.user_id()
-        if user_id:
-            analytics.set_user_id(user_id)

build/lib/build/lib/build/lib/build/lib/build/lib/pyntcli/commands/sub_command.py

@@ -1,15 +0,0 @@
-import argparse
-
-class PyntSubCommand:
-    def __init__(self, name) -> None:
-        self.name = name
-        pass 
-
-    def get_name(self):
-        return self.name
-    
-    def add_cmd(self, parent: argparse._SubParsersAction) -> argparse.ArgumentParser: 
-        raise NotImplemented()
-    
-    def run_cmd(self, args: argparse.Namespace): 
-        raise NotImplemented()

build/lib/build/lib/build/lib/build/lib/build/lib/pyntcli/commands/util.py

@@ -1,71 +0,0 @@
-import time
-import socket
-import os
-import tempfile
-from contextlib import contextmanager
-from pathlib import Path
-import webbrowser
-
-from pyntcli.pynt_docker import pynt_container
-from pyntcli.ui import report
-from pyntcli.ui import pynt_errors
-from pyntcli.transport import pynt_requests
-
-
-def is_port_in_use(port: int) -> bool:
-    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
-        return s.connect_ex(('localhost', port)) == 0
-
-def find_open_port() -> int:
-    with socket.socket() as s:
-        s.bind(('', 0))            
-        return s.getsockname()[1] 
-
-HEALTHCHECK_TIMEOUT = 10
-HEALTHCHECK_INTERVAL = 0.1
-
-def wait_for_healthcheck(address): 
-    start = time.time()
-    while start + HEALTHCHECK_TIMEOUT > time.time(): 
-        try:
-            res = pynt_requests.get(address + "/healthcheck")
-            if res.status_code == 418:
-                return 
-        except: 
-            time.sleep(HEALTHCHECK_INTERVAL)
-    raise TimeoutError()
-
-def get_user_report_path(path,file_type):
-    path = Path(path)
-    if path.is_dir():
-        return os.path.join(path, "pynt_results_{}.{}".format(int(time.time()),file_type))
-    
-    return os.path.join(str(path.parent), path.stem + ".{}".format(file_type))
-    
-
-class HtmlReportNotCreatedException(Exception):
-    pass
-
-@contextmanager
-def create_default_file_mounts(args):
-    html_report_path = os.path.join(tempfile.gettempdir(), "results.html")
-    json_report_path = os.path.join(tempfile.gettempdir(), "results.json")
-
-    if "reporters" in args and args.reporters: 
-        html_report_path = os.path.join(os.getcwd(), "pynt_results.html")
-        json_report_path = os.path.join(os.getcwd(), "pynt_results.json")
-
-    mounts = []
-    with open(html_report_path, "w"), open(json_report_path, "w"):    
-        mounts.append(pynt_container.create_mount(json_report_path, "/etc/pynt/results/results.json"))
-        mounts.append(pynt_container.create_mount(html_report_path, "/etc/pynt/results/results.html"))
-    
-    yield mounts
-    
-    if os.stat(html_report_path).st_size == 0:
-        raise HtmlReportNotCreatedException()
-    
-    webbrowser.open("file://{}".format(html_report_path))
-
-    if os.stat(html_report_path).st_size > 0:
-        report.PyntReporter(json_report_path).print_summary()   

build/lib/build/lib/build/lib/build/lib/build/lib/pyntcli/pynt_docker/__init__.py

@@ -1 +0,0 @@
-from .pynt_container import *

build/lib/build/lib/build/lib/build/lib/build/lib/pyntcli/pynt_docker/pynt_container.py

@@ -1,218 +0,0 @@
-import docker
-from docker.errors import DockerException, APIError, ImageNotFound
-from docker.types import Mount
-import os
-import argparse
-from typing import List
-
-from pyntcli.ui import ui_thread
-from pyntcli.analytics import send as analytics
-from pyntcli.store import CredStore
-from pyntcli.auth.login import PYNT_CREDENTIALS 
-
-PYNT_DOCKER_IMAGE = "ghcr.io/pynt-io/pynt"
-
-def create_mount(src, destination, mount_type="bind"):
-    return Mount(target=destination, source=src, type=mount_type)
-
-class DockerNotAvailableException(Exception):
-    pass
-
-class ImageUnavailableException(Exception):
-    pass
-
-def get_docker_type():
-    try:
-        c = docker.from_env()
-        version_data = c.version()
-        platform = version_data.get("Platform")
-
-        if platform and platform.get("Name"):
-            return platform.get("Name")
-        
-        return ""
-
-    except DockerException:
-        raise DockerNotAvailableException()
-    except Exception: #TODO: This is since windows is not behaving nice
-        raise DockerNotAvailableException()
-
-
-class PyntBaseConatiner():
-    def __init__(self,docker_type,docker_arguments,mounts, environment={}) -> None:
-        self.docker_type = docker_type
-        self.docker_arguments = docker_arguments
-        self.mounts = mounts
-        self.environment = environment
-
-class PyntDockerPort:
-    def __init__(self, src, dest, name) -> None:
-        self.src = src
-        self.dest = dest
-        self.name = name
-
-def get_container_with_arguments(args: argparse.Namespace , *port_args: PyntDockerPort) -> PyntBaseConatiner:
-    docker_arguments = []
-    if "desktop" in get_docker_type().lower():
-        ports = {}
-        for p in port_args:
-            ports[str(p.src)] = int(p.dest)
-        docker_type = PyntDockerDesktopContainer(ports=ports)
-    else:
-        docker_type = PyntNativeContainer(network="host")
-        for p in port_args: 
-            docker_arguments.append(p.name)
-            docker_arguments.append(str(p.dest))
-   
-    if "insecure" in args and args.insecure:
-        docker_arguments.append("--insecure")
-    
-    if "dev_flags" in args: 
-        docker_arguments += args.dev_flags.split(" ")
-    
-    mounts = []
-    if "host_ca" in args and args.host_ca:
-        ca_name = os.path.basename(args.host_ca)
-        docker_arguments += ["--host-ca", ca_name]
-        mounts.append(create_mount(os.path.abspath(args.host_ca), "/etc/pynt/{}".format(ca_name)))
-
-    return PyntBaseConatiner(docker_type, docker_arguments, mounts, {PYNT_CREDENTIALS:CredStore().get_access_token()})
-    
-def _container_image_from_tag(tag: str) -> str:
-    if ":" in tag: 
-        return tag.split(":")[0]
-
-    return tag
-
-class PyntContainer():
-    def __init__(self, image_name, tag, detach, base_container: PyntBaseConatiner) -> None:
-        self.docker_client: docker.DockerClient = None
-        self.image = image_name if not os.environ.get("IMAGE") else os.environ.get("IMAGE")
-        self.tag = tag if not os.environ.get("TAG") else os.environ.get("TAG")
-        self.detach = detach
-        self.stdout = None 
-        self.running = False
-        self.container_name = ""
-        self.base_container = base_container
-
-    
-    def _create_docker_client(self):
-        self.docker_client = docker.from_env()
-        pat = os.environ.get("DOCKER_PASSWORD")
-        username = os.environ.get("DOCKER_USERNAME")
-        registry = os.environ.get("DOCKER_REGISTRY")
-        if pat and username and registry:
-            self.docker_client.login(username=username, password=pat, registry=registry)
-    
-    def _is_docker_image_up_to_date(self, image):
-        return True
-    
-    def _handle_outdated_docker_image(self, image):
-        return image
-    
-    def kill_other_instances(self):
-        for c in self.docker_client.containers.list():
-            if len(c.image.tags) and _container_image_from_tag(c.image.tags[0]) == self.image:
-                c.kill()
-    
-    def stop(self):
-        if not self.running:
-            return 
-
-        self.kill_other_instances()
-
-        self.docker_client.close()
-        self.docker_client = None
-        self.running = False
-    
-    def is_alive(self):
-        if not self.docker_client or not self.container_name:
-            return False
-
-        l = self.docker_client.containers.list(filters={"name": self.container_name})
-        if len(l) != 1:
-            return False
-        
-        return l[0].status == "running"
-    
-    def pull_image(self):
-        try:
-            return self.docker_client.images.pull(self.image, tag=self.tag)
-        except APIError as e:
-            analytics.emit(analytics.ERROR,{"error": "Unable to pull image from ghcr: {}".format(e)})
-            return None
-
-    def get_image(self):
-        try:
-            image = self.pull_image()
-            if not image:
-                image = self.docker_client.images.get('{}:{}'.format(self.image,self.tag))
-            return image
-        except ImageNotFound:
-            raise ImageUnavailableException()
-
-    def run(self): 
-        if not self.docker_client:
-            self._create_docker_client()
-        
-        self.running = True
-        self.kill_other_instances()
-
-        ui_thread.print(ui_thread.PrinterText("Pulling latest docker", ui_thread.PrinterText.INFO))
-        image = self.get_image()
-        ui_thread.print(ui_thread.PrinterText("Docker pull done", ui_thread.PrinterText.INFO))
-    
-        args = self.base_container.docker_arguments if self.base_container.docker_arguments else None
-
-        run_arguments = {
-                "image":image, 
-                "detach":self.detach,
-                "mounts":self.base_container.mounts,
-                "environment": self.base_container.environment,
-                "stream": True,
-                "remove": True,
-                "command": args
-        }
-
-        run_arguments.update(self.base_container.docker_type.get_argumets())
-
-        c = self.docker_client.containers.run(**run_arguments)
-        self.container_name = c.name
-        self.stdout = c.logs(stream=True)
-
-        PyntContainerRegistery.instance().register_container(self)
-
-class PyntDockerDesktopContainer():
-    def __init__(self, ports) -> None:
-        self.ports = ports
-    
-    def get_argumets(self):
-        return {"ports": self.ports} if self.ports else {}
-        
-class PyntNativeContainer():
-    def __init__(self, network) -> None:
-        self.network = network
-
-    def get_argumets(self):
-        return {"network": self.network} if self.network else {}
-
-
-class PyntContainerRegistery():
-    _instance = None
-
-    def __init__(self) -> None:
-        self.containers: List[PyntContainer] = []
-
-    @staticmethod
-    def instance():
-        if not PyntContainerRegistery._instance:
-            PyntContainerRegistery._instance = PyntContainerRegistery() 
-
-        return PyntContainerRegistery._instance
-
-    def register_container(self, c: PyntContainer):
-        self.containers.append(c) 
-    
-    def stop_all_containers(self):
-        for c in self.containers: 
-            c.stop()

build/lib/build/lib/build/lib/build/lib/build/lib/pyntcli/store/__init__.py

@@ -1 +0,0 @@
-from .store import CredStore

build/lib/build/lib/build/lib/build/lib/build/lib/pyntcli/store/json_connector.py

@@ -1,23 +0,0 @@
-import json
-
-from .store_connector import StoreConnector
-
-class JsonStoreConnector(StoreConnector):
-    def __init__(self, data) -> None:
-        if not data:
-            data = JsonStoreConnector.default_value()
-        super().__init__(data)
-        self.json = json.loads(self._data)
-
-    def get(self, key):
-        return self.json.get(key, None)
-
-    def put(self, key, value):
-       self.json[key] = value
-    
-    def dump(self):
-        return json.dumps(self.json, indent=2)
-
-    @staticmethod
-    def default_value():
-        return json.dumps({})

build/lib/build/lib/build/lib/build/lib/build/lib/pyntcli/store/store.py

@@ -1,58 +0,0 @@
-from typing import Type 
-import os 
-import platform
-
-from .json_connector import JsonStoreConnector
-from .store_connector import StoreConnector
-
-class Store():
-    def __init__(self, file_location: str, connector_type: Type[StoreConnector]) -> None :
-        self.file_location = file_location
-        self.connector:StoreConnector = None
-        self._file = None
-        self._connector_tpye = connector_type
-    
-    def _get_file_data(self):
-        if self.connector: 
-            return 
-
-        dirname = os.path.dirname(self.file_location)
-        if not os.path.exists(dirname):
-            os.makedirs(dirname)
-        
-        if not os.path.exists(self.file_location):
-            with open(self.file_location, "w") as f:
-                self.connector = self._connector_tpye(self._connector_tpye.default_value())
-                return
-
-        with open(self.file_location, "r+") as f:
-            self.connector = self._connector_tpye(f.read())
-
-    def get(self, key):
-        self._get_file_data()
-        return self.connector.get(key)
-
-    def put(self, key, value):
-        self._get_file_data()
-        self.connector.put(key, value)
-    
-    def get_path(self):
-        return self.file_location
-
-    def __enter__(self):
-        self._get_file_data()
-        return self 
-
-    def __exit__(self, type, value, traceback): 
-        with open(self.file_location, "w") as f:
-            f.write(self.connector.dump())
-
-class CredStore(Store):
-    def __init__(self) -> None:
-        dir = ".pynt"
-        super().__init__(file_location=os.path.join(os.path.expanduser("~"), dir, "creds.json"),
-                         connector_type=JsonStoreConnector)
-        
-    def get_access_token(self):
-        return self.get("token")["access_token"]
- 

build/lib/build/lib/build/lib/build/lib/build/lib/pyntcli/store/store_connector.py

@@ -1,17 +0,0 @@
-class StoreConnector():
-    def __init__(self, data) -> None:
-        self._data = data 
-
-    def get(self, key):
-        raise NotImplemented()
-
-    def put(self, key, value):
-        raise NotImplementedError()
-    
-    def dump(self):
-        raise NotImplementedError()
-
-    @classmethod
-    def default_value(cls):
-        raise NotImplementedError()
-

build/lib/build/lib/build/lib/build/lib/build/lib/pyntcli/ui/progress.py

@@ -1,39 +0,0 @@
-import websocket
-
-def connect_progress_ws(address):
-    ws = websocket.WebSocket()
-    ws.connect(address)
-    return ws
-
-def wrap_ws_progress(ws):
-    prev = 0
-    while ws.connected:
-        try:
-            current = int(ws.recv())
-            yield current - prev
-            prev = current
-        except websocket.WebSocketConnectionClosedException:
-            return
-
-class PyntProgress():
-    def __init__(self, address, healthcheck, total, description) -> None:
-        self.running = False
-        self.trackable = PyntProgress.create_trackable(address, healthcheck)
-        self.total = total
-        self.description = description
-
-    @staticmethod
-    def create_trackable(address, healthcheck):
-        try: 
-            healthcheck()
-            return wrap_ws_progress(connect_progress_ws(address))
-        except TimeoutError:
-            return None
-
-    def __enter__(self):
-        return self
-
-    def __exit__(self, exc_type, exc_value, exc_traceback):
-        self.running = False
-
-

build/lib/build/lib/build/lib/build/lib/build/lib/pyntcli/ui/pynt_errors.py

@@ -1,7 +0,0 @@
-from pyntcli.ui import ui_thread
-
-def unexpected_error():
-    ui_thread.print(ui_thread.PrinterText("An Unexpected Error Occurred",style=ui_thread.PrinterText.WARNING) \
-                        .with_line("") \
-                            .with_line("Please tell us about it in our community channel and we will help you figure it out:",style=ui_thread.PrinterText.HEADER) \
-                                .with_line("https://join.slack.com/t/pynt-community/shared_invite/zt-1mvacojz5-WNjbH4HN8iksmKpCLTxOiQ",style=ui_thread.PrinterText.HEADER))

build/lib/build/lib/build/lib/build/lib/build/lib/pyntcli/ui/report.py

@@ -1,62 +0,0 @@
-import json
-from rich.table import Table, Column
-from pyntcli.ui import ui_thread
-
-
-
-class PyntTable():
-    def __init__(self, title) -> None:
-        self.table = Table(title=title,title_style="bold")
-
-    def set_columns(self, columns):
-        for col in columns:
-            self.table.add_column(col.header, style=col.style)
-
-    def add_row(self, *values):
-        self.table.add_row(*values)
-
-    def print(self):
-        ui_thread.print(ui_thread.PrinterText("").with_line(""))
-        ui_thread.print(self.table)
-
-
-class PyntReporter():
-    def __init__(self,report_path) -> None:
-        try:
-            self.report = json.load(open(report_path))
-        except json.JSONDecodeError:
-            self.report = None
-        
-    def print_summary(self):
-        if not self.report:
-            return
-        
-        functional_tests = self.report["functionalTests"]
-        functional_table = PyntTable("Functional Tests")
-        
-        functional_table.set_columns([
-            Column(header="Endpoints"),
-            Column(header="Requests")])
-        
-        functional_table.add_row(str(functional_tests["Endpoints"]),
-                                 str(functional_tests["Requests"]))
-
-        security_tests = self.report["securityTests"]
-        security_table = PyntTable("Security Tests")
-        
-        security_table.set_columns([
-            Column(header="Errors",style="red"),
-            Column(header="Warnings",style="yellow"),
-            Column(header="Passed",style="green",),
-            Column(header="Did Not Run")])
-        
-        security_table.add_row(str(security_tests["Findings"]),
-                                 str(security_tests["Warnings"]),
-                                 str(security_tests["Passed"]),
-                                 str(security_tests["DidNotRun"]))
-        
-        functional_table.print()
-        security_table.print()
-        
-        
-