Dual-CAPsMAN support, routerOS 7.13

mktxp/datasource/capsman_ds.py

@@ -14,24 +14,29 @@
 
 from mktxp.datasource.base_ds import BaseDSProcessor
 from mktxp.datasource.wireless_ds import WirelessMetricsDataSource
-
+from mktxp.flow.router_entry import RouterEntryWirelessType
 
 class CapsmanInfo:
     @staticmethod
-    def capsman_path(router_entry):
-        if WirelessMetricsDataSource.is_legacy(router_entry):
-            return '/caps-man'
-        else:
+    def capsman_paths(router_entry):
+        if router_entry.wireless_type == RouterEntryWirelessType.DUAL:
+            return ['/caps-man', f'/interface/wifi/capsman']
+        elif router_entry.wireless_type == RouterEntryWirelessType.WIRELESS:
+            return ['/caps-man']
+        else:    
             wireless_package = WirelessMetricsDataSource.wireless_package(router_entry)
-            return f'/interface/{wireless_package}/capsman'
+            return [f'/interface/{wireless_package}/capsman']
 
     @staticmethod
-    def registration_table_path(router_entry):
-        if WirelessMetricsDataSource.is_legacy(router_entry):
-            return '/caps-man/registration-table'
-        else:
+    def registration_table_paths(router_entry):
+        if router_entry.wireless_type == RouterEntryWirelessType.DUAL:
+            return ['/caps-man/registration-table', f'/interface/wifi/registration-table']
+        elif router_entry.wireless_type == RouterEntryWirelessType.WIRELESS:
+            return ['/caps-man/registration-table']
+        else:    
             wireless_package = WirelessMetricsDataSource.wireless_package(router_entry)
-            return f'/interface/{wireless_package}/registration-table'
+            return [f'/interface/{wireless_package}/registration-table']
+
 
 class CapsmanCapsMetricsDataSource:
     ''' Caps Metrics data provider
@@ -41,8 +46,9 @@ class CapsmanCapsMetricsDataSource:
         if metric_labels is None:
             metric_labels = []                
         try:
-            capsman_path = CapsmanInfo.capsman_path(router_entry)
-            remote_caps_records = router_entry.api_connection.router_api().get_resource(f'{capsman_path}/remote-cap').get()
+            remote_caps_records = []
+            for capsman_path in CapsmanInfo.capsman_paths(router_entry):
+                remote_caps_records.extend(router_entry.api_connection.router_api().get_resource(f'{capsman_path}/remote-cap').get())
             return BaseDSProcessor.trimmed_records(router_entry, router_records = remote_caps_records, metric_labels = metric_labels)
         except Exception as exc:
             print(f'Error getting CAPsMAN remote caps info from router{router_entry.router_name}@{router_entry.config_entry.hostname}: {exc}')
@@ -57,8 +63,9 @@ class CapsmanRegistrationsMetricsDataSource:
         if metric_labels is None:
             metric_labels = []                
         try:
-            registration_table_path = CapsmanInfo.registration_table_path(router_entry)
-            registration_table_records = router_entry.api_connection.router_api().get_resource(f'{registration_table_path}').get()
+            registration_table_records = []
+            for registration_table_path in CapsmanInfo.registration_table_paths(router_entry):
+                registration_table_records.extend(router_entry.api_connection.router_api().get_resource(f'{registration_table_path}').get())
             
             # With wifiwave2, Mikrotik renamed the field 'rx-signal' to 'signal' 
             # For backward compatibility, including both variants
@@ -75,10 +82,9 @@ class CapsmanRegistrationsMetricsDataSource:
 class CapsmanInterfacesDatasource:
     ''' Data provider for CAPsMaN interfaces
     '''
-
     @staticmethod
     def metric_records(router_entry, *, metric_labels = None):
-        if not WirelessMetricsDataSource.is_legacy(router_entry):
+        if not router_entry.wireless_type in (RouterEntryWirelessType.DUAL, RouterEntryWirelessType.WIRELESS):
             return None            
         if metric_labels is None:
             metric_labels = []                

mktxp/datasource/routerboard_ds.py

@@ -29,3 +29,25 @@ class RouterboardMetricsDataSource:
             print(f'Error getting system routerboard info from router{router_entry.router_name}@{router_entry.config_entry.hostname}: {exc}')
             return None
 
+    @staticmethod
+    def firmware_version(router_entry):
+        try:
+            version_st = router_entry.api_connection.router_api().get_resource('/system/routerboard').call('print', {'proplist':'current-firmware'})[0]
+            if version_st.get('current-firmware'):
+                return version_st['current-firmware']
+            return None
+        except Exception as exc:
+            print(f'Error getting routerboard current-firmware from router{router_entry.router_name}@{router_entry.config_entry.hostname}: {exc}')
+            return None
+
+    @staticmethod
+    def firmware_version(router_entry):
+        try:
+            version_st = router_entry.api_connection.router_api().get_resource('/system/routerboard').call('print', {'proplist':'upgrade-firmware'})[0]
+            if version_st.get('upgrade-firmware'):
+                return version_st['upgrade-firmware']
+            return None
+        except Exception as exc:
+            print(f'Error getting routerboard upgrade-firmware from router{router_entry.router_name}@{router_entry.config_entry.hostname}: {exc}')
+            return None
+

mktxp/datasource/wireless_ds.py

@@ -11,21 +11,17 @@
 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 ## GNU General Public License for more details.
 
-
 from mktxp.datasource.base_ds import BaseDSProcessor
 from mktxp.datasource.package_ds import PackageMetricsDataSource
-
+from mktxp.flow.router_entry import RouterEntryWirelessType
 
 class WirelessMetricsDataSource:
     ''' Wireless Metrics data provider
     '''             
-    WIFIWAVE2 = 'wifiwave2'
     WIRELESS = 'wireless'
+    WIFIWAVE2 = 'wifiwave2'
     WIFI = 'wifi'
 
-    WIFI_PACKAGE = 'wifi-qcom'
-    WIFI_AC_PACKAGE = 'wifi-qcom-ac'
-
     @staticmethod
     def metric_records(router_entry, *, metric_labels = None, add_router_id = True):
         if metric_labels is None:
@@ -45,20 +41,13 @@ class WirelessMetricsDataSource:
             print(f'Error getting wireless registration table info from router{router_entry.router_name}@{router_entry.config_entry.hostname}: {exc}')
             return None
 
-
     @staticmethod
     def wireless_package(router_entry):
-        if not router_entry.wifi_package:
-            if PackageMetricsDataSource.is_package_installed(router_entry, package_name = WirelessMetricsDataSource.WIFI_PACKAGE):
-              router_entry.wifi_package = WirelessMetricsDataSource.WIFI
-            elif PackageMetricsDataSource.is_package_installed(router_entry, package_name = WirelessMetricsDataSource.WIFI_AC_PACKAGE):
-              router_entry.wifi_package = WirelessMetricsDataSource.WIFI
-            elif PackageMetricsDataSource.is_package_installed(router_entry, package_name = WirelessMetricsDataSource.WIFIWAVE2):
-              router_entry.wifi_package = WirelessMetricsDataSource.WIFIWAVE2
-            else:
-              router_entry.wifi_package = WirelessMetricsDataSource.WIRELESS
-        return router_entry.wifi_package
+        if router_entry.wireless_type in (RouterEntryWirelessType.DUAL, RouterEntryWirelessType.WIRELESS):
+            return WirelessMetricsDataSource.WIRELESS
+        elif router_entry.wireless_type == RouterEntryWirelessType.WIFIWAVE2:
+            return WirelessMetricsDataSource.WIFIWAVE2
+        else:
+            return WirelessMetricsDataSource.WIFI
+
 
-    @staticmethod
-    def is_legacy(router_entry):
-        return WirelessMetricsDataSource.wireless_package(router_entry) == WirelessMetricsDataSource.WIRELESS