import sqlite3

​

def get_next_id_starting_from(reference_id, prefix):

    # Connect to the SQLite database

    conn = sqlite3.connect('contra.db')

    cursor = conn.cursor()

​

    # Select the minimum reference ID from the 'contra' table where reference_id starts with the specified prefix and is greater than the specified ID

    cursor.execute("SELECT MIN(reference_id) FROM contra WHERE reference_id LIKE ? AND reference_id > ?", (f'{prefix}%', reference_id,))

    next_id = cursor.fetchone()[0]

​

    # Close the connection

    conn.close()

​

    return next_id if next_id is not None else None

​

def get_next_c_id(reference_id):

    return get_next_id_starting_from(reference_id, 'c')

​

def get_next_p_id(reference_id):

    return get_next_id_starting_from(reference_id, 'p')