from utils import (
    get_db_connection,
)

class CompanyService:
    @staticmethod
    def get_company_id_by_cnpj(cnpj: str):
        conn = get_db_connection()
        try:
            with conn.cursor() as cur:
                query = "SELECT id FROM companies WHERE cnpj = %s"
                cur.execute(query, (cnpj,))
                result = cur.fetchone()
                return result[0] if result else None
        finally:
            conn.close()