pyntcli 0.1.71__py3-none-any.whl → 0.1.73__py3-none-any.whl

pyntcli/commands/root.py

@@ -7,33 +7,48 @@ 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)
+    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("\thar       -   run scan on static har file")
+        .with_line("\tcommand   -   run scan with a given command")
+        .with_line("\tlisten    -   run scan with a routed traffic")
+        .with_line("\tburp      -   run scan on a burp xml output file")
+        .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())
 
 
 def check_cicd_context():
-    if environ.get('GITHUB_ACTION') is not None:
+    if environ.get("GITHUB_ACTION") is not None:
         analytics.emit(analytics.CICD, {"env": "GitHub"})
         return
-    if environ.get('GITLAB_USER_ID') is not None:
+    if environ.get("GITLAB_USER_ID") is not None:
         analytics.emit(analytics.CICD, {"env": "GitLab"})
         return
-    if environ.get('JENKINS_HOME') is not None:
+    if environ.get("JENKINS_HOME") is not None:
         analytics.emit(analytics.CICD, {"env": "Jenkins"})
-    
 
-class BaseCommand():
+
+class BaseCommand:
     def __init__(self) -> None:
         self.base: argparse.ArgumentParser = None
         self.subparser: argparse._SubParsersAction = None
@@ -41,15 +56,21 @@ class BaseCommand():
     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 
-   
+        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(
+            "--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="")
         parser.add_argument("--transport-config", type=str, default="")
         parser.add_argument("--application-id", type=str, default="", required=False)
@@ -58,7 +79,7 @@ class BaseCommand():
     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):
@@ -67,11 +88,11 @@ class BaseCommand():
         if login.should_login():
             l = login.Login().login()
         else:
-            login.refresh_token()  
-        
+            login.refresh_token()
+
         analytics.emit(analytics.LOGIN_DONE)
         user_id = login.user_id()
         if user_id:
             analytics.set_user_id(user_id)
 
-        check_cicd_context()
+        check_cicd_context()

pyntcli/main.py

@@ -7,7 +7,7 @@ from pyntcli.pynt_docker import pynt_container
 from pyntcli.ui import ui_thread
 from pyntcli.ui import pynt_errors
 from pyntcli.ui import ui_thread
-from pyntcli.auth import login 
+from pyntcli.auth import login
 from pyntcli.analytics import send as analytics
 from requests.exceptions import SSLError
 from pyntcli.transport.pynt_requests import InvalidPathException, InvalidCertFormat
@@ -17,28 +17,36 @@ from pyntcli.commands.postman import PyntWebSocketException
 from pyntcli import __version__
 
 
+def shutdown_cli():
+    analytics.stop()
+    pynt_container.PyntContainerRegistery.instance().stop_all_containers()
+    ui_thread.stop()
+
 
 def signal_handler(signal_number, frame):
     ui_thread.print(ui_thread.PrinterText("Exiting..."))
-    
-    analytics.stop()
-    pynt_container.PyntContainerRegistery.instance().stop_all_containers() 
-    ui_thread.stop()
+
+    shutdown_cli()
 
     exit(0)
 
+
 def check_for_dependecies():
     pynt_container.get_docker_type()
 
+
 def print_header():
-    ui_thread.print(ui_thread.PrinterText(*ui_thread.pynt_header()) \
-                             .with_line(*ui_thread.pynt_version()) \
+    ui_thread.print(ui_thread.PrinterText(*ui_thread.pynt_header())
+                             .with_line(*ui_thread.pynt_version())
                              .with_line(""))
+
+
 def start_analytics():
     user_id = login.user_id()
     if user_id:
-       analytics.set_user_id(user_id)
-    
+        analytics.set_user_id(user_id)
+
+
 def main():
     print_header()
     try:
@@ -50,42 +58,42 @@ def main():
         cli.run_cmd(cli.parse_args(argv[1:]))
         analytics.stop()
     except pynt_cmd.PyntCommandException:
-       pynt_cmd.root.usage()
+        pynt_cmd.root.usage()
     except pynt_container.DockerNotAvailableException:
         ui_thread.print(ui_thread.PrinterText("Docker was unavailable, please make sure docker is installed and running.", ui_thread.PrinterText.WARNING))
         analytics.emit(analytics.ERROR, {"error": "docker unavailable"})
-    except SSLError: 
+    except SSLError:
         ui_thread.print(ui_thread.PrinterText("We encountered SSL issues and could not proceed, this may be the cause of a VPN or a Firewall in place.", ui_thread.PrinterText.WARNING))
-    except login.Timeout: 
+    except login.Timeout:
         ui_thread.print(ui_thread.PrinterText("Pynt CLI exited due to incomplete registration, please try again.", ui_thread.PrinterText.WARNING))
         analytics.emit(analytics.ERROR, {"error": "login timeout"})
     except login.InvalidTokenInEnvVarsException:
         ui_thread.print(ui_thread.PrinterText("Pynt CLI exited due to malformed credentials provided in env vars.", ui_thread.PrinterText.WARNING))
         analytics.emit(analytics.ERROR, {"error": "invalid pynt cli credentials in env vars"})
     except pynt_container.ImageUnavailableException:
-        analytics.emit(analytics.ERROR,{"error": "Couldnt pull pynt image and no local image found"})
+        analytics.emit(analytics.ERROR, {"error": "Couldnt pull pynt image and no local image found"})
         pynt_errors.unexpected_error()
     except HtmlReportNotCreatedException:
-        analytics.emit(analytics.ERROR,{"error": "Html report was not created"})
+        analytics.emit(analytics.ERROR, {"error": "Html report was not created"})
         pynt_errors.unexpected_error()
     except InvalidPathException as e:
         ui_thread.print(ui_thread.PrinterText("Pynt CLI exited due to invalid host-CA path: {}".format(e), ui_thread.PrinterText.WARNING))
-        analytics.emit(analytics.ERROR,{"error": "Host CA path provided was invalid"})
+        analytics.emit(analytics.ERROR, {"error": "Host CA path provided was invalid"})
     except InvalidCertFormat as e:
         ui_thread.print(ui_thread.PrinterText("Pynt CLI exited due to invalid host-CA. Please provide a file in PEM format: {}".format(e), ui_thread.PrinterText.WARNING))
-        analytics.emit(analytics.ERROR,{"error": "Host CA provided was not in valid pem format"})
+        analytics.emit(analytics.ERROR, {"error": "Host CA provided was not in valid pem format"})
     except PyntWebSocketException:
-        analytics.emit(analytics.ERROR,{"error": "postman websocket failed to connect"})
+        analytics.emit(analytics.ERROR, {"error": "postman websocket failed to connect"})
         pynt_errors.unexpected_error()
     except SomeFoundingsOrWargningsException as e:
         exit(1)
     except Exception as e:
-        analytics.emit(analytics.ERROR,{"error": "{}".format(e)})
+        analytics.emit(analytics.ERROR, {"error": "{}".format(e)})
         pynt_errors.unexpected_error()
     finally:
         log.flush_logger()
-        ui_thread.stop()
-        analytics.stop()
+        shutdown_cli()
+
 
 if __name__ == "__main__":
     main()

pyntcli/transport/pynt_requests.py

@@ -6,23 +6,29 @@ import certifi
 import tempfile
 
 
-
 class HostCaException(Exception):
     pass
 
+
 class InvalidPathException(HostCaException):
     pass
 
+
 class InvalidCertFormat(HostCaException):
     pass
 
+
 verify = True
 
+urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
+
+
 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):
@@ -31,24 +37,30 @@ def add_host_ca(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:
+    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)
+    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)
+    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)
+    return requests.put(url, data=data, verify=verify, **kwargs)
+
+
+def request_from_xml(method, url, proxies=None, data=None,  **kwargs):
+    return requests.request(method=method, url=url, data=data, verify=False, proxies=proxies, **kwargs)

pyntcli/ui/ui_thread.py

@@ -6,6 +6,7 @@ from rich.text import Text
 from rich.console import Console
 from rich.status import Status
 from rich.progress import Progress
+from rich.markup import escape
 from typing import Tuple
 
 from pyntcli import __version__ as cli_version
@@ -140,6 +141,9 @@ class Printer():
                 if isinstance(data, PyntProgress):
                     self._handle_progress(data)
                     continue
+                if isinstance(data, str):
+                    self.console.print(escape(str(data)))
+                    continue
                 else:
                     self.console.print(data)
             except queue.Empty:

{pyntcli-0.1.71.dist-info → pyntcli-0.1.73.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pyntcli
-Version: 0.1.71
+Version: 0.1.73
 Summary: Command line utility to handle all of Pynt's different integrations
 Author-email: Pynt-io <support@pynt.io>
 Project-URL: Homepage, https://pynt.io
@@ -12,4 +12,5 @@ Requires-Dist: pem
 Requires-Dist: certifi >=2017.4.17
 Requires-Dist: logzio-python-handler >=4.1.0
 Requires-Dist: websocket-client
+Requires-Dist: xmltodict
 

{pyntcli-0.1.71.dist-info → pyntcli-0.1.73.dist-info}/RECORD

@@ -1,18 +1,19 @@
-pyntcli/__init__.py,sha256=ebu6Nblu_UmpciwG6xnbUJm-16F-ZA6L-sagDt37smo,23
-pyntcli/main.py,sha256=cyDbu8Gbx-Uwri4z9KKL--BMMERunSzrZ0VTiX3JC48,4114
+pyntcli/__init__.py,sha256=B2xptQoAZtCL_O_fLMSWF71JDtlCZp8jLmdtHyMdosE,23
+pyntcli/main.py,sha256=cqratxd_ZJXLsP4xRbF2fmwj3Eo4kQ8WlbrwUtfZQ94,4125
 pyntcli/analytics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pyntcli/analytics/send.py,sha256=ewNAKnn3KbKwfDuOasgAsSGbUWILLD8rPaXIAlLucnM,2507
 pyntcli/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pyntcli/auth/login.py,sha256=WgF5r00bpM4c1__thzD6zCJ6207qxgtT0ixPvD1cEHA,5109
 pyntcli/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pyntcli/commands/command.py,sha256=lZBLLtkfbQ7fFAqWuMP00T_ZkUsnCpiAPd4p5sn4MzM,7870
-pyntcli/commands/har.py,sha256=s5TUw9JVhINRa1kaU6d784u6k_UAMcYu8ZzLk9W6JG4,2935
+pyntcli/commands/burp.py,sha256=wd-HuNyj6v6IHk6x9-DAaqXcm1cQa-RA5FlxveBd6wA,10427
+pyntcli/commands/command.py,sha256=tSjJj0fhQt2lCvtdu8H3IuiBlAllesx4f6ptNfuinTw,9435
+pyntcli/commands/har.py,sha256=Iq0455umKvlcMxcU4l70-NIZh7Wovry3f76u68drmQ4,3654
 pyntcli/commands/id_command.py,sha256=2J5oEa39uyBMMSCRHzuKh3_11CeUEYi7YLoBk1yuu_I,942
-pyntcli/commands/listen.py,sha256=2cVWc9athlLOmM1NLf-VVTqZENhlrbxW7yf7TRNVMro,7782
-pyntcli/commands/newman.py,sha256=GfLPgOx7Pmx1S6ZI8wJOiSzhxBFn4WuEjN-MWL3mW0Y,4233
-pyntcli/commands/postman.py,sha256=5eF14wjdDwJC8AvXjwyuZnlb6JZsPs_QWh5e77LauGA,4928
-pyntcli/commands/pynt_cmd.py,sha256=OBdgglzJJPFUOWtleORkLErpmd_0QkUF0hhK-_MRaak,2417
-pyntcli/commands/root.py,sha256=4TLH67_n1S4nxmlB2YzyDLZpx2GJLK2rXMEfD3Wze1M,2839
+pyntcli/commands/listen.py,sha256=aANo5DwZ6_TAuXdJX3L8eDk9I6ompPAqtaUgxnEbR54,8536
+pyntcli/commands/newman.py,sha256=_AspmTk98nY0Ct26WNK4fFgoYbqzmU90FfkygxJjgjY,4828
+pyntcli/commands/postman.py,sha256=x5sPR-KaawKzAkLRK9vJXDzZ3sx9-6TuH6SNd5tnM4Y,4946
+pyntcli/commands/pynt_cmd.py,sha256=NkzT8X4mI_fhzui_9KdAQXY5AnMGdoQkzDH6pcreLHU,2453
+pyntcli/commands/root.py,sha256=FEf2gXvFyO_RZi6XzFakSVnOQw_sGBopY5SL5gldudg,3248
 pyntcli/commands/sub_command.py,sha256=GF3-rE_qk2L4jGPFqHLm9SdGINmu3EakhjJTFyWjRms,374
 pyntcli/commands/util.py,sha256=IREZvHYJ7wEth8ujOkk5ZXZ4UrUA7jDv134_UueED4s,2995
 pyntcli/log/__init__.py,sha256=cOGwOYzMoshEbZiiasBGkj6wF0SBu3Jdpl-AuakDesw,19
@@ -24,17 +25,17 @@ pyntcli/store/json_connector.py,sha256=UGs3uORw3iyn0YJ8kzab-veEZToA6d-ByXYuqEleW
 pyntcli/store/store.py,sha256=Fbc7yxjht0EJ89eGF8eMgNQJ3sgYRTPmj2-sWQ5UUPs,1766
 pyntcli/store/store_connector.py,sha256=w4LzcpRZesUZL1f63RmLlWEFRtJ6Y6rcS6PkkGtO4MA,357
 pyntcli/transport/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pyntcli/transport/pynt_requests.py,sha256=a9siz45X1RESKCWoVDe1tnhzn0vjVdYd_25KIxQO4h0,1316
+pyntcli/transport/pynt_requests.py,sha256=KiEG3hNcwY7DLIJDCq-7LIPq54yYQcDBhHe3KhpqRTc,1563
 pyntcli/ui/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pyntcli/ui/progress.py,sha256=RrnO_jJNunoyupylakmWmHOEPw3lh99OHpKBzL6OBiE,1008
 pyntcli/ui/pynt_errors.py,sha256=UAr8OV5EM4zLhfWSmK_qss8nal2Ezo66Dk8ZMvR_RTs,546
 pyntcli/ui/report.py,sha256=W-icPSZrGLOubXgam0LpOvHLl_aZg9Zx9qIkL8Ym5PE,1930
-pyntcli/ui/ui_thread.py,sha256=JAonXc6aOFawaEixghclxHSCHZBieEWfZzsPef_ZNMI,4796
+pyntcli/ui/ui_thread.py,sha256=OVTbiIFMg2KgxAvHf7yy86xGm4RVS2vj_VYZkMi-SRY,4956
 tests/conftest.py,sha256=gToq5K74GtgeGQXjFvXSzMaE6axBYxAzcFG5XJPOXjI,427
 tests/auth/test_login.py,sha256=M6JRFTQRZrL6M2-iph_r-aBSQMMiFDncQbVYeObBFYU,3296
 tests/store/test_cred_store.py,sha256=_7-917EtNC9eKEumO2_lt-7KuDmCwOZFaowCm7DbA_A,254
-pyntcli-0.1.71.dist-info/METADATA,sha256=tG8dyjc1MzjQ8KtUmktrda1d4JW1m3Dq6EWGwMBakx4,438
-pyntcli-0.1.71.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-pyntcli-0.1.71.dist-info/entry_points.txt,sha256=kcGmqAxXDttNk2EPRcqunc_LTVp61gzakz0v-GEE2SY,43
-pyntcli-0.1.71.dist-info/top_level.txt,sha256=u9MDStwVHB7UG8PUcODeWCul_NvzL2EzoLvSlgwLHFs,30
-pyntcli-0.1.71.dist-info/RECORD,,
+pyntcli-0.1.73.dist-info/METADATA,sha256=YG6eADsoX_vyR3rTTkWvpkrKZeY8okAneTVRDvu3yZU,463
+pyntcli-0.1.73.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+pyntcli-0.1.73.dist-info/entry_points.txt,sha256=kcGmqAxXDttNk2EPRcqunc_LTVp61gzakz0v-GEE2SY,43
+pyntcli-0.1.73.dist-info/top_level.txt,sha256=u9MDStwVHB7UG8PUcODeWCul_NvzL2EzoLvSlgwLHFs,30
+pyntcli-0.1.73.dist-info/RECORD,,

{pyntcli-0.1.71.dist-info → pyntcli-0.1.73.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.42.0)
+Generator: bdist_wheel (0.43.0)
 Root-Is-Purelib: true
 Tag: py3-none-any