Add speed column to ports table

resource_management/hardware/router.py

@@ -3,13 +3,12 @@ import json
 from jnpr.junos import Device
 from jnpr.junos.exception import ConnectError, ConnectTimeoutError
 from jnpr.junos.op.fpc import FpcHwTable
-from jnpr.junos.op.fpc import FpcInfoTable
-from jnpr.junos.op.ethport import EthPortTable
-from jnpr.junos.op.inventory import ModuleTable
 from jnpr.junos.op.phyport import PhyPortTable
 
 logger = logging.getLogger(__name__)
 
+PORT_SPEEDS = { "ge": 1, "xe": 10, "et": 40 }
+
 nodes = [
     { "_id": 1, "fqdn": "core1.spa.private" },
     { "_id": 2, "fqdn": "core2.spa.private" },
@@ -48,7 +47,8 @@ def load_router():
             _ports.get()
             for key, value in _ports.items():
                 line_card_position = int(key[3])
-                ports.append( { "name": key[-3:], "speed": key[:2], "<line_card_position>": line_card_position } )
+                speed = PORT_SPEEDS[key[:2]]
+                ports.append( { "name": key[-3:], "speed": speed, "<line_card_position>": line_card_position } )
 
         except Exception as err:
             print("Error reading hardware information for {1}: {0}".format(err, devname))

resource_management/migrations/versions/cdbd931a47d4_add_speed_column_to_ports_table.py

+"""add speed column to ports table
+
+Revision ID: cdbd931a47d4
+Revises: 8901d3bca270
+Create Date: 2022-10-24 08:02:14.845868
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = 'cdbd931a47d4'
+down_revision = '8901d3bca270'
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    with op.batch_alter_table('ports') as batch_op:
+        batch_op.add_column(
+            Column('speed', sa.Integer, nullable=False)),
+            insert_after='name'
+        )
+
+
+def downgrade() -> None:
+    op.drop_column('ports','speed')