adding get_models() back in, since we will need it in home assistant

solax/discovery.py

@@ -1,6 +1,7 @@
 import asyncio
 import logging
 import typing
+from typing import List
 
 from solax.inverter import Inverter, InverterError
 from solax.inverters import (
@@ -102,3 +103,9 @@ async def discover(host, port, pwd="", model=None) -> Inverter:
     discover_state = DiscoveryState()
     await discover_state.discover(host, port, pwd, model)
     return discover_state.get_discovered_inverter()
+
+
+def get_models() -> List[str]:
+    models = list(map(lambda inverter: inverter.__name__, REGISTRY))
+    models.sort()
+    return models

tests/test_discovery.py

@@ -16,6 +16,7 @@ async def test_discovery_with_model(inverters_fixture):
     conn, inverter_class, _ = inverters_fixture
     rt_api = await solax.real_time_api(*conn, "", inverter_class.__name__)
     assert rt_api.inverter.__class__ == inverter_class
+    assert inverter_class.__name__ in solax.discovery.get_models()
 
 
 @pytest.mark.asyncio