multi_segment_controller/db_handler.py

import sqlite3
from pathlib import Path


class CellLimits():

    def __init__(self, min_voltage, max_voltage, min_current, max_current, min_temperature, max_temperature):
        if min_voltage > max_voltage:
            raise ValueError("min_voltage cannot be greater than max_voltage")
        if min_current > max_current:
            raise ValueError("min_current cannot be greater than max_current")
        if min_temperature > max_temperature:
            raise ValueError("min_temperature cannot be greater than max_temperature")
        self.min_voltage = min_voltage
        self.max_voltage = max_voltage
        self.min_current = min_current
        self.max_current = max_current
        self.min_temperature = min_temperature
        self.max_temperature = max_temperature

class DatabaseHandler:
    def __init__(self, db_path="cell_data.db"):
        # add code for initializing database connection
        pass


    def get_cell_capacity_nom(self, cell_id):
        # add code for getting nominal cell capacity from database
        pass

    def get_cell_capacity(self, cell_id):
        # add code for getting cell capacity from database
        pass


    def update_cell_capacity(self, cell_id, estimated_capacity_mah):
        # add code for updating cell capacity in database
        pass

    def get_cell_limits(self, cell_id):
        # add code for getting cell limits from database
        return None