Merge pull request #97 from CiscoTestAutomation/release_24.3

Release 24.3

Author: omehrabi

docs/changelog/2024/march.rst

@@ -0,0 +1,76 @@
+March 2024
+==========
+
+ - Unicon v24.3 
+------------------------
+
+
+
+.. csv-table:: Module Versions
+    :header: "Modules", "Versions"
+
+        ``unicon.plugins``, v24.3 
+        ``unicon``, v24.3 
+
+Install Instructions
+^^^^^^^^^^^^^^^^^^^^
+
+.. code-block:: bash
+
+    bash$ pip install unicon.plugins
+    bash$ pip install unicon
+
+Upgrade Instructions
+^^^^^^^^^^^^^^^^^^^^
+
+.. code-block:: bash
+
+    bash$ pip install --upgrade unicon.plugins
+    bash$ pip install --upgrade unicon
+
+Features and Bug Fixes:
+^^^^^^^^^^^^^^^^^^^^^^^
+
+
+
+
+Changelogs
+^^^^^^^^^^
+--------------------------------------------------------------------------------
+                                      New                                       
+--------------------------------------------------------------------------------
+
+* backend
+    * Option to use `UNICON_BACKEND` environment variable to select backend
+
+
+--------------------------------------------------------------------------------
+                                      Fix                                       
+--------------------------------------------------------------------------------
+
+* bases
+    * connection
+        * add operating_mode to the connection object
+
+
+--------------------------------------------------------------------------------
+                                      New                                       
+--------------------------------------------------------------------------------
+
+* discovery_tokens
+    * Add prompt_recovery to dialog
+
+* iosxe
+    * Connection provider
+        * Add support for operating mode detection on connect()
+
+
+--------------------------------------------------------------------------------
+                                      Fix                                       
+--------------------------------------------------------------------------------
+
+* iosxr
+    * Modified connection provider
+        * Updated connection provider for handeling token discovery.
+
+

docs/changelog/index.rst

@@ -4,6 +4,7 @@ Changelog
 .. toctree::
    :maxdepth: 2
 
+   2024/march
    2024/february
    2024/january
    2023/november

docs/changelog_plugins/2024/march.rst

@@ -0,0 +1,54 @@
+March 2024
+==========
+
+ - Unicon.Plugins v24.3 
+------------------------
+
+
+
+.. csv-table:: Module Versions
+    :header: "Modules", "Versions"
+
+        ``unicon.plugins``, v24.3 
+        ``unicon``, v24.3 
+
+Install Instructions
+^^^^^^^^^^^^^^^^^^^^
+
+.. code-block:: bash
+
+    bash$ pip install unicon.plugins
+    bash$ pip install unicon
+
+Upgrade Instructions
+^^^^^^^^^^^^^^^^^^^^
+
+.. code-block:: bash
+
+    bash$ pip install --upgrade unicon.plugins
+    bash$ pip install --upgrade unicon
+
+Features and Bug Fixes:
+^^^^^^^^^^^^^^^^^^^^^^^
+
+
+
+
+Changelogs
+^^^^^^^^^^
+--------------------------------------------------------------------------------
+                                      Fix                                       
+--------------------------------------------------------------------------------
+
+* cheetah/ap
+    * Add more prompt handler to statemachine
+
+* token discovery
+    * Add more prompt to dialog
+    * Update dialog timeout
+
+* generic
+    * Added encryption selection pattern
+    * Removed duplicate enable_secret_handler and setup_enter_selection functions
+
+

docs/changelog_plugins/index.rst

@@ -4,6 +4,7 @@ Plugins Changelog
 .. toctree::
    :maxdepth: 2
 
+   2024/march
    2024/february
    2024/january
    2023/november

src/unicon/plugins/__init__.py

@@ -1,4 +1,4 @@
-__version__ = '24.2'
+__version__ = '24.3'
 
 supported_chassis = [
     'single_rp',

src/unicon/plugins/cheetah/ap/statemachine.py

@@ -1,5 +1,5 @@
 from unicon.plugins.generic.statemachine import GenericSingleRpStateMachine
-from unicon.plugins.generic.statements import GenericStatements
+from unicon.plugins.generic.statements import GenericStatements, default_statement_list
 from unicon.statemachine import State, Path, StateMachine
 from unicon.eal.dialogs import Dialog, Statement
 from unicon.plugins.generic.patterns import GenericPatterns
@@ -46,3 +46,5 @@ def create(self):
         self.add_path(enable_to_shell)
         self.add_path(shell_to_enable)
         self.add_path(enable_to_disable)
+
+        self.add_default_statements(default_statement_list)

src/unicon/plugins/generic/connection_provider.py

@@ -67,4 +67,4 @@ def get_connection_dialog(self):
                              self.connection.settings.PASSWORD_PROMPT)
         return con.connect_reply + \
                     Dialog(custom_auth_stmt + connection_statement_list
-                        if custom_auth_stmt else connection_statement_list)
+                        if custom_auth_stmt else connection_statement_list)

src/unicon/plugins/generic/patterns.py

@@ -85,3 +85,5 @@ def __init__(self):
         self.get_cursor_position = r'\x1b\[6n'
 
         self.new_password = r'^(Enter new password|Confirm password):\s*$'
+
+        self.enter_your_encryption_selection_2 = r'^.*?Enter your encryption selection( \[2])?:\s*$'

src/unicon/plugins/generic/service_statements.py

@@ -354,7 +354,8 @@ def config_session_locked_handler(context):
                          generic_statements.syslog_msg_stmt,
                          # Below statements have loop_continue=False
                          password_stmt, press_enter, press_return,
-                         connection_closed_stmt, eof_statement
+                         connection_closed_stmt, eof_statement,
+                         generic_statements.enter_your_encryption_selection_stmt
                          ]
 
 # ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++#

src/unicon/plugins/generic/statements.py

@@ -131,15 +131,16 @@ def syslog_wait_send_return(spawn, session):
     session['buffer_len'] = len(spawn.buffer)
 
 
-def chatty_term_wait(spawn, trim_buffer=False):
+def chatty_term_wait(spawn, trim_buffer=False, wait_time=None):
     """ Wait some time for any chatter to cease from the device.
     """
+    chatty_wait_time = wait_time or spawn.settings.ESCAPE_CHAR_CHATTY_TERM_WAIT
     for retry_number in range(spawn.settings.ESCAPE_CHAR_CHATTY_TERM_WAIT_RETRIES):
 
-        if buffer_settled(spawn, spawn.settings.ESCAPE_CHAR_CHATTY_TERM_WAIT):
+        if buffer_settled(spawn, chatty_wait_time):
             break
         else:
-            buffer_wait(spawn, spawn.settings.ESCAPE_CHAR_CHATTY_TERM_WAIT * (retry_number + 1))
+            buffer_wait(spawn, chatty_wait_time * (retry_number + 1))
 
     else:
         spawn.log.warning('The buffer has not settled because the device is chatty. '
@@ -334,35 +335,6 @@ def setup_enter_selection(spawn, context):
         spawn.sendline('2')
 
 
-def enable_secret_handler(spawn, context, session):
-    if 'password_attempts' not in session:
-        session['password_attempts'] = 1
-    else:
-        session['password_attempts'] += 1
-    if session.password_attempts > spawn.settings.PASSWORD_ATTEMPTS:
-        raise UniconAuthenticationError('Too many enable password retries')
-
-    enable_credential_password = get_enable_credential_password(context=context)
-    if enable_credential_password and len(enable_credential_password) >= \
-            spawn.settings.ENABLE_SECRET_MIN_LENGTH:
-        spawn.sendline(enable_credential_password)
-    else:
-        spawn.log.warning('Using enable secret from TEMP_ENABLE_SECRET setting')
-        enable_secret = spawn.settings.TEMP_ENABLE_SECRET
-        context['setup_selection'] = 0
-        spawn.sendline(enable_secret)
-
-
-def setup_enter_selection(spawn, context):
-    selection = context.get('setup_selection')
-    if selection is not None:
-        if str(selection) == '0':
-            spawn.log.warning('Not saving setup configuration')
-        spawn.sendline(f'{selection}')
-    else:
-        spawn.sendline('2')
-
-
 def ssh_tacacs_handler(spawn, context):
     result = False
     start_cmd = spawn.spawn_command
@@ -758,6 +730,12 @@ def __init__(self):
                                                 loop_continue=True,
                                                 continue_timer=False)
 
+        self.enter_your_encryption_selection_stmt = Statement(pattern=pat.enter_your_encryption_selection_2,
+                                                   action=setup_enter_selection,
+                                                   args=None,
+                                                   loop_continue=True,
+                                                   continue_timer=True)
+
 #############################################################
 #  Statement lists
 #############################################################
@@ -803,7 +781,8 @@ def __init__(self):
 
 initial_statement_list = [generic_statements.init_conf_stmt,
                           generic_statements.mgmt_setup_stmt,
-                          generic_statements.enter_your_selection_stmt
+                          generic_statements.enter_your_selection_stmt,
+                          generic_statements.enter_your_encryption_selection_stmt
                           ]
 
 
@@ -817,4 +796,5 @@ def __init__(self):
     default_statement_list + \
     authentication_statement_list + \
     initial_statement_list + \
-    pre_connection_statement_list
+    pre_connection_statement_list
+