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

build/lib/pyntcli/transport/pynt_requests.py

@@ -1,54 +0,0 @@
-import requests
-import urllib3
-import os
-import pem
-import certifi
-import tempfile
-
-
-
-class HostCaException(Exception):
-    pass
-
-class InvalidPathException(HostCaException):
-    pass
-
-class InvalidCertFormat(HostCaException):
-    pass
-
-verify = True
-
-def disable_tls_termination():
-    global verify
-    verify = False
-    urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
-
-def add_host_ca(ca_path):
-    global verify
-    if not os.path.isfile(ca_path):
-        raise InvalidPathException("{} - No such file".format(ca_path))
-    if not pem.parse_file(ca_path):
-        raise InvalidCertFormat("{} - Invalid Format".format(ca_path))
-
-    cert_data = open(ca_path, "rb").read()
-    
-    cafile = certifi.where()
-    ca_chain = open(cafile, 'rb').read()
-    
-    bundle_path = os.path.join(tempfile.gettempdir(), "bundle.pem")
-    with open(bundle_path,"wb") as f:
-        f.write(ca_chain)
-        f.write(b'\n')
-        f.write(cert_data)
-    
-    verify = bundle_path
-
-    
-def get(url, params=None, **kwargs):
-    return requests.get(url,params=params,verify=verify,**kwargs)
-
-def post(url, data=None, json=None, **kwargs):
-    return requests.post(url,data=data,json=json, verify=verify,**kwargs)
-
-def put(url, data=None, **kwargs):
-    return requests.put(url,data=data,verify=verify,**kwargs)

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/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/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()
-        
-        
-        

build/lib/pyntcli/ui/ui_thread.py

@@ -1,172 +0,0 @@
-from threading import Thread
-import queue
-import os
-import time
-from rich.text import Text
-from rich.console import Console
-from rich.status import Status
-from rich.progress import Progress
-from typing import Tuple
-
-from pyntcli import __version__ as cli_version
-from pyntcli.ui.progress import PyntProgress
-
-class PrinterText():
-    DEFAULT = 0
-    HEADER = 1
-    INFO = 2
-    WARNING = 3
-
-    def __init__(self,text,style=DEFAULT):
-        self.text = Text(text, PrinterText.get_style(style))
-
-    @staticmethod
-    def get_style(style):
-        if style == PrinterText.INFO:
-            return "bold blue"
-        if style == PrinterText.WARNING:
-            return "bold red"
-        if style == PrinterText.HEADER:
-            return "bold"
-        if style == PrinterText.DEFAULT:
-            return None
-
-    def with_line(self, line, style=DEFAULT):
-        self.text.append(os.linesep)
-        self.text.append(Text(line, PrinterText.get_style(style)))
-        return self
-
-class AnsiText():
-    def __init__(self, data) -> None:
-        self.data = data
-    
-    @staticmethod
-    def wrap_gen(gen):
-        for v in gen:
-            yield AnsiText(v)
-
-class Spinner():
-    def __init__(self, prompt, style) -> None:
-        self.prompt = prompt
-        self.style = style 
-        self.runnning = False
-    
-    def __enter__(self):
-        return self 
-
-    def __exit__(self, exc_type, exc_value, exc_traceback):
-        self.running = False
-
-def pynt_version()-> Tuple[str,int]: 
-    return "Pynt CLI version " + cli_version,PrinterText.DEFAULT
-
-def pynt_header()-> Tuple[str,int]:
-    return "API Security testing autopilot",PrinterText.DEFAULT
-
-def gen_func_loop(gen):
-    for l in gen:
-        data = l
-        if type(data) == bytes: 
-            data = l.decode("utf-8")
-        if not isinstance(data, AnsiText) and data[-1] == "\n":
-            data = data[:-1]
-        _print(data)
-        
-def print_generator(gen):
-    t = Thread(target=gen_func_loop, args=(gen,), daemon=True) 
-    t.start()
-
-def _print(s): 
-    Printer.instance().print(s)
-
-def print(s): 
-    if type(s) == bytes: 
-        s = s.decode("utf-8")
-    _print(s)
-
-def stop():
-    Printer.instance().stop()
-
-class Printer():
-    _instace = None
-
-    def __init__(self) -> None:
-        self.running = False
-        self.run_thread = Thread(target=self._print_in_loop, daemon=True)
-        self.print_queue = queue.Queue()
-        self.console = Console(tab_size=4)
-   
-    @staticmethod 
-    def instance():
-        if not Printer._instace:
-            Printer._instace = Printer() 
-            Printer._instace.start() 
-
-        return Printer._instace
-
-    def start(self):
-        self.running = True
-        self.run_thread.start()
-    
-    def _handle_spinner(self, spinner):
-        spinner.running = True 
-        s = Status(spinner.prompt, spinner=spinner.style, console=self.console)
-        s.start()
-        while spinner.running and self.running:
-            time.sleep(0.5)
-        s.stop()
-    
-    def _handle_progress(self, progress):
-        if not progress.trackable:
-            return
-        progress.running = True 
-        with Progress(console=self.console, transient=True) as p:
-            t = p.add_task(description=progress.description, total=progress.total)
-            for update in progress.trackable:
-                if not (progress.running and self.running):
-                    return
-                p.update(t, advance=update)
-                time.sleep(0.5)
-
-    def _print_in_loop(self):
-        while self.running:
-            try: 
-                data = self.print_queue.get(timeout=1)
-                if isinstance(data, list): 
-                    data = data[0]
-                if isinstance(data, Spinner):
-                    self._handle_spinner(data)
-                    continue
-                if isinstance(data, PyntProgress):
-                    self._handle_progress(data)
-                    continue
-                else:
-                    self.console.print(data)
-            except queue.Empty:
-                pass
-        
-        while not self.print_queue.empty():
-            self.console.print(self.print_queue.get())
-
-
-    def print(self, data):
-        if isinstance(data,PrinterText):
-            data = data.text
-        if isinstance(data, AnsiText):
-            data = Text.from_ansi(data.data.decode())
-        self.print_queue.put(data)
-    
-    def stop(self):
-        self.running = False
-        self.run_thread.join()
-
-def spinner(prompt, style): 
-    s = Spinner(prompt, style)
-    _print([s])
-    return s
-
-def progress(what_to_track,healthcheck, description, total=100):
-    pointer_to_progress = [PyntProgress(what_to_track,healthcheck, total, description)]
-    _print(pointer_to_progress)
-    return pointer_to_progress[0]
-