nebuleair_pro_4g/SARA/R5/setPDP.py

'''
  ____    _    ____      _    
 / ___|  / \  |  _ \    / \   
 \___ \ / _ \ | |_) |  / _ \  
  ___) / ___ \|  _ <  / ___ \ 
 |____/_/   \_\_| \_\/_/   \_\
                              
Script to set the PDP context for the SARA R5


/usr/bin/python3 /var/www/nebuleair_pro_4g/SARA/R5/setPDP.py

'''
import serial
import time
import sys
import json
import re

#get data from config
def load_config(config_file):
    try:
        with open(config_file, 'r') as file:
            config_data = json.load(file)
        return config_data
    except Exception as e:
        print(f"Error loading config file: {e}")
        return {}

#Fonction pour mettre à jour le JSON de configuration
def update_json_key(file_path, key, value):
    """
    Updates a specific key in a JSON file with a new value.

    :param file_path: Path to the JSON file.
    :param key: The key to update in the JSON file.
    :param value: The new value to assign to the key.
    """
    try:
        # Load the existing data
        with open(file_path, "r") as file:
            data = json.load(file)
        
        # Check if the key exists in the JSON file
        if key in data:
            data[key] = value  # Update the key with the new value
        else:
            print(f"Key '{key}' not found in the JSON file.")
            return
        
        # Write the updated data back to the file
        with open(file_path, "w") as file:
            json.dump(data, file, indent=2)  # Use indent for pretty printing
        
        print(f"💾 updating '{key}' to '{value}'.")
    except Exception as e:
        print(f"Error updating the JSON file: {e}")

# Define the config file path
config_file = '/var/www/nebuleair_pro_4g/config.json'
# Load the configuration data
config = load_config(config_file)
baudrate = config.get('SaraR4_baudrate', 115200)       #baudrate du sara R4
device_id = config.get('deviceID', '').upper()         #device ID en maj

ser_sara = serial.Serial(
    port='/dev/ttyAMA2',
    baudrate=baudrate, #115200 ou 9600
    parity=serial.PARITY_NONE, #PARITY_NONE, PARITY_EVEN or PARITY_ODD
    stopbits=serial.STOPBITS_ONE,
    bytesize=serial.EIGHTBITS,
    timeout = 2
)

def read_complete_response(serial_connection, timeout=2, end_of_response_timeout=2, wait_for_lines=None, debug=True):
    '''
    Fonction très importante !!!
    Reads the complete response from a serial connection and waits for specific lines.
    '''
    if wait_for_lines is None:
        wait_for_lines = []  # Default to an empty list if not provided

    response = bytearray()
    serial_connection.timeout = timeout
    end_time = time.time() + end_of_response_timeout
    start_time = time.time()

    while True:
        elapsed_time = time.time() - start_time  # Time since function start
        if serial_connection.in_waiting > 0:
            data = serial_connection.read(serial_connection.in_waiting)
            response.extend(data)
            end_time = time.time() + end_of_response_timeout  # Reset timeout on new data

            # Decode and check for any target line
            decoded_response = response.decode('utf-8', errors='replace')
            for target_line in wait_for_lines:
                if target_line in decoded_response:
                    if debug:
                        print(f"[DEBUG] 🔎 Found target line: {target_line} (in {elapsed_time:.2f}s)")
                    return decoded_response  # Return response immediately if a target line is found
        elif time.time() > end_time:
            if debug:
                print(f"[DEBUG] Timeout reached. No more data received.")
            break
        time.sleep(0.1)  # Short sleep to prevent busy waiting

    # Final response and debug output
    total_elapsed_time = time.time() - start_time
    if debug:
        print(f"[DEBUG] ⏱️ elapsed time: {total_elapsed_time:.2f}s. ⏱️")
    # Check if the elapsed time exceeded 10 seconds
    if total_elapsed_time > 10 and debug:
        print(f"[ALERT] 🚨 The operation took too long 🚨")
        print(f'<span style="color: red;font-weight: bold;">[ALERT] ⚠️{total_elapsed_time:.2f}s⚠️</span>')

    return response.decode('utf-8', errors='replace')  # Return the full response if no target line is found

try:
    print('Start script')

    # 1. Check connection
    print('➡️Check SARA R5 connexion')
    command = f'ATI0\r'
    ser_sara.write(command.encode('utf-8'))
    response_SARA_1 = read_complete_response(ser_sara, wait_for_lines=["OK"])
    print(response_SARA_1, end="")
    time.sleep(1)

    # 2. Activate PDP context 1
    print('➡️Activate PDP context 1')
    command = f'AT+CGACT=1,1\r'
    ser_sara.write(command.encode('utf-8'))
    response_SARA_2 = read_complete_response(ser_sara, wait_for_lines=["OK"])
    print(response_SARA_2, end="")
    time.sleep(1)

    # 2. Set the PDP type
    print('➡️Set the PDP type to IPv4 referring to the outputof the +CGDCONT read command')
    command = f'AT+UPSD=0,0,0\r'
    ser_sara.write(command.encode('utf-8'))
    response_SARA_3 = read_complete_response(ser_sara, wait_for_lines=["OK"])
    print(response_SARA_3, end="")
    time.sleep(1)

    # 2. Profile #0 is mapped on CID=1.
    print('➡️Profile #0 is mapped on CID=1.')
    command = f'AT+UPSD=0,100,1\r'
    ser_sara.write(command.encode('utf-8'))
    response_SARA_3 = read_complete_response(ser_sara, wait_for_lines=["OK"])
    print(response_SARA_3, end="")
    time.sleep(1)

    # 2. Set the PDP type
    print('➡️Activate the PSD profile #0: the IPv4 address is already assigned by the network.')
    command = f'AT+UPSDA=0,3\r'
    ser_sara.write(command.encode('utf-8'))
    response_SARA_3 = read_complete_response(ser_sara, wait_for_lines=["OK","+UUPSDA"])
    print(response_SARA_3, end="")
    time.sleep(1)

   

except Exception as e:
    print("An error occurred:", e)
    traceback.print_exc()  # This prints the full traceback