"""Handle db interactions for platforms."""
from typing import Optional

import MySQLdb as mdb
from MySQLdb.cursors import DictCursor

def platforms_by_species(
        conn: mdb.Connection,
        speciesid: int,
        offset: int = 0,
        limit: Optional[int] = None
) -> tuple[dict, ...]:
    """Retrieve platforms by the species"""
    _query = ("SELECT * FROM GeneChip WHERE SpeciesId=%s "
              "ORDER BY GeneChipName ASC")
    if bool(limit) and limit > 0:
        _query = f"{_query} LIMIT {limit} OFFSET {offset}"
    with conn.cursor(cursorclass=DictCursor) as cursor:
        cursor.execute(_query, (speciesid,))
        return tuple(dict(row) for row in cursor.fetchall())


def species_platforms_count(conn: mdb.Connection, species_id: int) -> int:
    """Get the number of platforms in the database for a particular species."""
    with conn.cursor(cursorclass=DictCursor) as cursor:
        cursor.execute(
            "SELECT COUNT(GeneChipName) AS count FROM GeneChip "
            "WHERE SpeciesId=%s",
            (species_id,))
        return int(cursor.fetchone()["count"])

def platform_by_id(conn: mdb.Connection, platformid: int) -> Optional[dict]:
    """Retrieve a platform by its ID"""
    with conn.cursor(cursorclass=DictCursor) as cursor:
        cursor.execute("SELECT * FROM GeneChip WHERE Id=%s",
                       (platformid,))
        result = cursor.fetchone()
        if bool(result):
            return dict(result)

    return None