pyntcli 0.1.81__py3-none-any.whl → 0.1.83__py3-none-any.whl

pyntcli/__init__.py

@@ -1 +1 @@
-__version__ = "0.1.81"
+__version__ = "0.1.83"

pyntcli/commands/burp.py

@@ -10,6 +10,7 @@ from subprocess import Popen, PIPE
 from functools import partial
 import xmltodict
 import base64
+import pyntcli.log.log as log
 from xml.parsers.expat import ExpatError
 
 from pyntcli.pynt_docker import pynt_container
@@ -30,6 +31,9 @@ methods = [
     "connect",
 ]
 
+logger = log.get_logger()
+supported_encoding = ["utf-8", "latin1", "utf-16", "cp1252"]
+
 
 def is_valid_method(method: str):
     if method.lower() in methods:
@@ -37,9 +41,31 @@ def is_valid_method(method: str):
     return False
 
 
+def decode_request(item) -> str:
+    error = None
+    for encoding in supported_encoding:
+        try:
+            return base64.b64decode(item["request"]["#text"]).decode(encoding)
+        except UnicodeDecodeError as e:
+            error = e
+            continue
+        except Exception as e:
+            error = e
+            break
+
+    ui_thread.print(
+        ui_thread.PrinterText(
+            f"Error decoding request: {error}"),
+        ui_thread.PrinterText.WARNING)
+    logger.error(f"Error decoding request: {error}")
+
+    raise e
+
+
 def replay_req(item, proxy_port):
     url = item["url"]
-    decoded_req = base64.b64decode(item["request"]["#text"]).decode("utf-8")
+    decoded_req = decode_request(item)
+
     method = decoded_req.split("\r\n")[0].split(" ")[0]
 
     if not is_valid_method(method):
@@ -289,12 +315,12 @@ class BurpCommand(sub_command.PyntSubCommand):
                 json_report_path = util.get_user_report_path(full_path, "json")
 
             if html_report:
-                with open(html_report_path, "w",encoding="utf-8") as html_file:
+                with open(html_report_path, "w", encoding="utf-8") as html_file:
                     html_file.write(html_report)
                 webbrowser.open("file://{}".format(html_report_path))
 
             if json_report:
-                with open(json_report_path, "w",encoding="utf-8") as json_file:
+                with open(json_report_path, "w", encoding="utf-8") as json_file:
                     json_file.write(json_report)
                 reporter = cli_reporter.PyntReporter(json_report_path)
                 reporter.print_summary()

pyntcli/commands/root.py

@@ -1,6 +1,4 @@
 import argparse
-import sys
-
 from os import environ
 
 from pyntcli.auth import login
@@ -15,22 +13,16 @@ def root_usage():
         .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("\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("\tpynt-id   -   view your pynt-id to use when running pynt in CI pipeline")
+        .with_line("\tlogout    -   log out from your user")
         .with_line("")
-        .with_line(
-            "Run pynt [COMMAND] -h to get help on a specific command",
-            style=ui_thread.PrinterText.INFO,
-        )
+        .with_line("Run pynt [COMMAND] -h to get help on a specific command", style=ui_thread.PrinterText.INFO,)
     )
 
 

pyntcli/main.py

@@ -1,6 +1,6 @@
 from sys import argv, exit
 import signal
-
+import os
 import pyntcli.log.log as log
 from pyntcli.commands import pynt_cmd
 from pyntcli.pynt_docker import pynt_container
@@ -16,6 +16,8 @@ from pyntcli.commands.util import SomeFindingsOrWarningsException
 from pyntcli.commands.postman import PyntWebSocketException
 from pyntcli import __version__
 
+from pyntcli.store import CredStore
+
 
 def shutdown_cli():
     analytics.stop()
@@ -48,9 +50,25 @@ def start_analytics():
         log.add_user_details(user_id)
 
 
+def logout():
+    creds_path = CredStore().get_path()
+    try:
+        if os.path.isfile(creds_path):
+            os.remove(creds_path)
+            ui_thread.print("you have successfully logged out")
+            return
+    except Exception:
+        ui_thread.print(f"not able to log out - try to manually delete the {creds_path} file")
+
+    ui_thread.print("you are not logged in. run pynt --help and choose the required command")
+
+
 def main():
     print_header()
     try:
+        if argv[1] == "logout":
+            logout()
+            return
         log.set_source(__version__)
         start_analytics()
         check_for_dependecies()

pyntcli/store/store.py

@@ -10,8 +10,8 @@ class Store():
         self.file_location = file_location
         self.connector: StoreConnector = None
         self._file = None
-        self._connector_tpye = connector_type
-
+        self._connector_type = connector_type
+            
     def _get_file_data(self):
         if self.connector:
             return
@@ -22,11 +22,11 @@ class Store():
 
         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())
+                self.connector = self._connector_type(self._connector_type.default_value())
                 return
 
         with open(self.file_location, "r+") as f:
-            self.connector = self._connector_tpye(f.read())
+            self.connector = self._connector_type(f.read())
 
     def get(self, key):
         self._get_file_data()

{pyntcli-0.1.81.dist-info → pyntcli-0.1.83.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pyntcli
-Version: 0.1.81
+Version: 0.1.83
 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

{pyntcli-0.1.81.dist-info → pyntcli-0.1.83.dist-info}/RECORD

@@ -1,11 +1,11 @@
-pyntcli/__init__.py,sha256=4reVRut6pFQ0MxDHMjUh2C4vPGa3csSjDj46nUuBiXs,23
-pyntcli/main.py,sha256=qTGMG2JSpieg-1RRgBz9dPL4vHo1WJJIxTPQYAuzxuY,5138
+pyntcli/__init__.py,sha256=YNVbuQj-JrlVU3yGuccLkZ5P-L-37tQUcshGno0Ijlo,23
+pyntcli/main.py,sha256=wg2Is1ckRCb3EwsLFshDsbPKvntPOR7sDB2Nq-DW4wk,5689
 pyntcli/analytics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pyntcli/analytics/send.py,sha256=pJOyOWl3g_Vm9apKK3LzNVqsnC6zsWA1bCK3ZegbLpc,3637
 pyntcli/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pyntcli/auth/login.py,sha256=TljsRXbEkNI1YUrKm5mlTw4YiecYScYUsit8Z8vstss,5228
 pyntcli/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pyntcli/commands/burp.py,sha256=KuxeX9I0ZXSVAMHdt9OYFok79X9IK6z71-ZN-_fnrvU,10565
+pyntcli/commands/burp.py,sha256=2FqLc65SjuXdenUS1ggC6dGRU_iSA52Ft_8OnfGsm_8,11193
 pyntcli/commands/command.py,sha256=0O7Za_cjT6vDkDfM0OTMPB6DLI3U1r1R7lXQydz7458,9495
 pyntcli/commands/har.py,sha256=mSCbTUnxQrKzJd-dAWoc6Tkw6tU1LDH7Ha1w2ylrrrg,3654
 pyntcli/commands/id_command.py,sha256=UBEgMIpm4vauTCsKyixltiGUolNg_OfHEJvJ_i5BpJY,943
@@ -13,7 +13,7 @@ pyntcli/commands/listen.py,sha256=JoDo3BirYf9f3Fr21qAQ7kIDmLbM5cuFzQtf8RQ_nhs,84
 pyntcli/commands/newman.py,sha256=y0KolwMgsvoqPz2mp0QRug_qNr-ftOZbu_tN7h4bH7I,4826
 pyntcli/commands/postman.py,sha256=GWq4NJJ_9WdFiXk5rv2nTyMM27w50XLh4LKkuuWpw4I,4721
 pyntcli/commands/pynt_cmd.py,sha256=KOl9guUtesO2JcMM5nPKKkjnK6F9HV4jHHcoUk4KVhw,2825
-pyntcli/commands/root.py,sha256=GijCi8hqe8sXEo6faWimlCmT8d782yjrw1IWJT5RAMk,3320
+pyntcli/commands/root.py,sha256=dmgdzoFuf5LkwrkwvWf1MtlwTBgsVpS85Yr_cQCVuGA,3291
 pyntcli/commands/sub_command.py,sha256=GF3-rE_qk2L4jGPFqHLm9SdGINmu3EakhjJTFyWjRms,374
 pyntcli/commands/util.py,sha256=NJBJk4tKzrWmilYdo-7CROH5IXKl0UQAPKaMzRZtxm8,2827
 pyntcli/log/__init__.py,sha256=cOGwOYzMoshEbZiiasBGkj6wF0SBu3Jdpl-AuakDesw,19
@@ -23,7 +23,7 @@ pyntcli/pynt_docker/container_utils.py,sha256=_Onn7loInzyJAG2-Uk6CGpsuRyelmUFHOv
 pyntcli/pynt_docker/pynt_container.py,sha256=y4m8wZU16gqFPaMzEJKKHut4DbjPxr_du8g7I4-AcHk,8761
 pyntcli/store/__init__.py,sha256=xuS9OB21F6B1sUx5XPGxz_6WpG6-KTMbuq50RrZS5OY,29
 pyntcli/store/json_connector.py,sha256=UGs3uORw3iyn0YJ8kzab-veEZToA6d-ByXYuqEleWsA,560
-pyntcli/store/store.py,sha256=Kf4IFCAu0i0DPVGgntrSYUiwz6kmW9HvItKnCT-MosE,1905
+pyntcli/store/store.py,sha256=9KwalOd1EA1VtYwr9oJgBsPgUYakX5uyif_sNXGQ614,1917
 pyntcli/store/store_connector.py,sha256=w4LzcpRZesUZL1f63RmLlWEFRtJ6Y6rcS6PkkGtO4MA,357
 pyntcli/transport/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pyntcli/transport/pynt_requests.py,sha256=KiEG3hNcwY7DLIJDCq-7LIPq54yYQcDBhHe3KhpqRTc,1563
@@ -35,8 +35,8 @@ pyntcli/ui/ui_thread.py,sha256=OVTbiIFMg2KgxAvHf7yy86xGm4RVS2vj_VYZkMi-SRY,4956
 tests/conftest.py,sha256=gToq5K74GtgeGQXjFvXSzMaE6axBYxAzcFG5XJPOXjI,427
 tests/auth/test_login.py,sha256=KFlzWhXBAuwdi7GXf16gCB3ya94LQG2wjcSChE149rQ,3798
 tests/store/test_cred_store.py,sha256=_7-917EtNC9eKEumO2_lt-7KuDmCwOZFaowCm7DbA_A,254
-pyntcli-0.1.81.dist-info/METADATA,sha256=D5hWFtXYjvdJp7uIjMBlj8Z_yYP0pobzSV1Ce6T1onM,463
-pyntcli-0.1.81.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-pyntcli-0.1.81.dist-info/entry_points.txt,sha256=kcGmqAxXDttNk2EPRcqunc_LTVp61gzakz0v-GEE2SY,43
-pyntcli-0.1.81.dist-info/top_level.txt,sha256=u9MDStwVHB7UG8PUcODeWCul_NvzL2EzoLvSlgwLHFs,30
-pyntcli-0.1.81.dist-info/RECORD,,
+pyntcli-0.1.83.dist-info/METADATA,sha256=XcLyBCUa_7GOUyGDqDBpGeWC5M-i1C4DM1CoWtCL7ok,463
+pyntcli-0.1.83.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+pyntcli-0.1.83.dist-info/entry_points.txt,sha256=kcGmqAxXDttNk2EPRcqunc_LTVp61gzakz0v-GEE2SY,43
+pyntcli-0.1.83.dist-info/top_level.txt,sha256=u9MDStwVHB7UG8PUcODeWCul_NvzL2EzoLvSlgwLHFs,30
+pyntcli-0.1.83.dist-info/RECORD,,