ts_shape.loader.timeseries.timescale_loader

Classes:

• TimescaleDBDataAccess –

TimescaleDBDataAccess

TimescaleDBDataAccess(start_timestamp: str, end_timestamp: str, uuids: List[str], db_config: Dict[str, str])

Methods:

• fetch_data_as_dataframe –

  Retrieves timeseries data from TimescaleDB and returns it as a single DataFrame.

Source code in src/ts_shape/loader/timeseries/timescale_loader.py

def __init__(self, start_timestamp: str, end_timestamp: str, uuids: List[str], db_config: Dict[str, str]):
    self.start_timestamp = start_timestamp
    self.end_timestamp = end_timestamp
    self.uuids = uuids
    self.db_config = db_config
    self.engine = create_engine(
        f'postgresql+psycopg2://{db_config["db_user"]}:{db_config["db_pass"]}@{db_config["db_host"]}/{db_config["db_name"]}'
    )

fetch_data_as_dataframe

fetch_data_as_dataframe() -> DataFrame

Retrieves timeseries data from TimescaleDB and returns it as a single DataFrame. :return: A combined DataFrame with data for all specified UUIDs within the time range.

Source code in src/ts_shape/loader/timeseries/timescale_loader.py

def fetch_data_as_dataframe(self) -> pd.DataFrame:
    """
    Retrieves timeseries data from TimescaleDB and returns it as a single DataFrame.
    :return: A combined DataFrame with data for all specified UUIDs within the time range.
    """
    df_list = [chunk for uuid in self.uuids for chunk in self._fetch_data(uuid)]
    return pd.concat(df_list, ignore_index=True) if df_list else pd.DataFrame()