Adding in block volume script

Author: greyhoundforty

.pre-commit-config.yaml

@@ -1,12 +1,16 @@
 # .pre-commit-config.yaml
 repos:
--   repo: https://github.com/Yelp/detect-secrets
+  - repo: https://github.com/Yelp/detect-secrets
     rev: v1.4.0
     hooks:
-    -   id: detect-secrets
-        args: ['--baseline', '.secrets.baseline']
+      - id: detect-secrets
+        args: ["--baseline", ".secrets.baseline"]
         exclude: package.lock.json
--   repo: https://github.com/asottile/reorder-python-imports
+  - repo: https://github.com/asottile/reorder-python-imports
     rev: v3.12.0
     hooks:
-    -   id: reorder-python-imports
+      - id: reorder-python-imports
+  - repo: https://github.com/pre-commit/mirrors-prettier
+    rev: v3.0.3 # Use the sha / tag you want to point at
+    hooks:
+      - id: prettier

classic-virtual/details.py

@@ -1,19 +1,94 @@
 import json
+import socket
+import threading
+from concurrent.futures import ThreadPoolExecutor
 
 import SoftLayer
+from rich import box
+from rich import print as rprint
+from rich.console import Console
+from rich.table import Table
 
 client = SoftLayer.create_client_from_env()
 
-vm_id = 137682113
+def get_vms(client):
+  vms = client['Account'].getVirtualGuests()
+  return [vm for vm in vms if vm.get('primaryIpAddress')]
 
-def print_keys(d, parent_key=''):
-  for k, v in d.items():
-    key = parent_key + '.' + k if parent_key else k
-    if isinstance(v, dict):
-      print_keys(v, key)
-    else:
-      print(key)
+def get_vm_details(vm):
+  # Get VM details
+  vm_id = vm['id']
+  mask = "mask[id,globalIdentifier,fullyQualifiedDomainName,hostname,domain,createDate,modifyDate,provisionDate,notes,dedicatedAccountHostOnlyFlag,privateNetworkOnlyFlag,primaryBackendIpAddress,primaryIpAddress,networkComponents[id,status,speed,maxSpeed,name,macAddress,primaryIpAddress,port,primarySubnet,securityGroupBindings[securityGroup[id, name]]],lastKnownPowerState.name,powerState,status,maxCpu,maxMemory,datacenter,activeTransaction[id, transactionStatus[friendlyName,name]],lastOperatingSystemReload.id,blockDevices,blockDeviceTemplateGroup[id, name, globalIdentifier],postInstallScriptUri,operatingSystem[passwords[username,password],softwareLicense.softwareDescription[manufacturer,name,version,referenceCode]],softwareComponents[passwords[username,password,notes],softwareLicense[softwareDescription[manufacturer,name,version,referenceCode]]],hourlyBillingFlag,userData,billingItem[id,package,nextInvoiceTotalRecurringAmount,nextInvoiceChildren[description,categoryCode,recurringFee,nextInvoiceTotalRecurringAmount],children[description,categoryCode,nextInvoiceTotalRecurringAmount],orderItem[id,order.userRecord[username],preset.keyName]],tagReferences[id,tag[name,id]],networkVlans[id,vlanNumber,networkSpace],dedicatedHost.id,transientGuestFlag,lastTransaction[transactionGroup]]" 
+  return client['Virtual_Guest'].getObject(id=vm_id, mask=mask)
 
-vm_details = client['Virtual_Guest'].getObject(id=vm_id)
+def check_port(ip, port):
+  sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+  result = sock.connect_ex((ip, port))
+  sock.close()
+  return port if result == 0 else None
 
-print_keys(vm_details)
+def scan_hosts(hosts, port):
+  results = []
+  with ThreadPoolExecutor() as executor:
+    futures = [executor.submit(check_port, vm['primaryIpAddress'], port) 
+               for vm in hosts]
+    for future in futures:
+      port, vm = future.result() 
+      if port:
+        results.append((port, vm))
+  return results
+      
+def main():
+  
+  windows_table = Table(show_header=True, header_style="white", box=box.ROUNDED)
+  windows_table.add_column("Server ID")
+  windows_table.add_column("Public IP")
+  windows_table.add_column("Open Port")
+  windows_table.add_column("Provision Date")
+  windows_table.add_column("Provisioned By")
+
+  linux_table = Table(show_header=True, header_style="blue", box=box.ROUNDED)
+  linux_table.add_column("Server ID")
+  linux_table.add_column("Public IP")
+  linux_table.add_column("Open Port")
+  linux_table.add_column("Provision Date")
+  linux_table.add_column("Provisioned By")
+
+  try:
+    client = SoftLayer.create_client_from_env()
+    
+    vms = get_vms(client)
+    
+    win_vms = []
+    nix_vms = []
+    for vm in vms:
+      details = get_vm_details(vm)
+      if details['operatingSystem']['softwareLicense']['softwareDescription']['referenceCode'].startswith('WIN_'):
+        win_vms.append(details)  
+      else:
+        nix_vms.append(details)
+
+    rprint("Checking ports on Windows hosts")  
+    win_ports = scan_hosts(win_vms, 3389)
+    for port, vm in win_ports:
+      windows_table.add_row(str(vm['id']), str(vm['primaryIpAddress']), 
+                            str(port), str(vm['provisionDate']), 
+                            str(vm['billingItem']['orderItem']['order']['userRecord']['username']))
+
+    rprint("Checking ports on Linux hosts")
+    nix_ports = scan_hosts(nix_vms, 22)
+    for port in nix_ports:
+      vm = port['vm']
+      linux_table.add_row(str(vm['id']), str(vm['primaryIpAddress']), 
+                          str(port), str(vm['provisionDate']),
+                          str(vm['billingItem']['orderItem']['order']['userRecord']['username']))
+    
+    console = Console()
+    console.print(windows_table)
+    console.print(linux_table)
+
+  except Exception as e:
+    rprint(f"Error: {e}")
+  
+if __name__ == "__main__":
+  main()

classic-virtual/scanGuests.py

@@ -8,7 +8,7 @@
 from rich.console import Console
 from rich.table import Table
 
-windows_table = Table(show_header=True, header_style="green", box=box.ROUNDED)  
+windows_table = Table(show_header=True, header_style="white", box=box.ROUNDED)  
 linux_table = Table(show_header=True, header_style="blue", box=box.ROUNDED) 
 
 linux_table.add_column("Server ID")
@@ -17,6 +17,12 @@
 linux_table.add_column("Provision Date")
 linux_table.add_column("Provisioned By")
 
+windows_table.add_column("Server ID")
+windows_table.add_column("Public IP")
+windows_table.add_column("Open Port")
+windows_table.add_column("Provision Date")
+windows_table.add_column("Provisioned By")
+
 def check_port(ip, port):
 
   sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
@@ -69,45 +75,59 @@ def check_port(ip, port):
       'provision_user': provision_user
     })
 
-batch_size = 10
-print("Checking ports on Nix hosts")
-# Split nix_hosts into batches 
-nix_batches = [nix_hosts[i:i+batch_size] for i in range(0, len(nix_hosts), batch_size)]
-
-for batch in nix_batches:
+def nix_host_scan():
+  batch_size = 10
+  print("Checking ports on Nix hosts")
+  # Split nix_hosts into batches 
+  nix_batches = [nix_hosts[i:i+batch_size] for i in range(0, len(nix_hosts), batch_size)]
+
+  for batch in nix_batches:
+
+    threads = []
+
+    for vm in batch:
+        ip = vm['public_ip']
+        open_port = check_port(ip, 22)
+        id = vm['server_id']
+        user = vm['provision_user']
+        date = vm['provision_date']
+    
+        if open_port:
+          linux_table.add_row(str(id), str(ip), str(open_port), str(date), str(user))
+  
 
-  threads = []
+    console = Console()
 
-  for vm in batch:
-      ip = vm['public_ip']
-      open_port = check_port(ip, 22)
-      id = vm['server_id']
-      user = vm['provision_user']
-      date = vm['provision_date']
-  
-      if open_port:
-        linux_table.add_row(str(id), str(ip), str(open_port), str(date), str(user))
+    console.print(linux_table)
 
-print("Checking ports on Windows hosts")
-# Split windows_hosts into batches 
-win_batches = [win_hosts[i:i+batch_size] for i in range(0, len(win_hosts), batch_size)]
+def win_host_scan():
+  batch_size = 10
+  print("Checking ports on Windows hosts")
+  # Split windows_hosts into batches 
+  win_batches = [win_hosts[i:i+batch_size] for i in range(0, len(win_hosts), batch_size)]
 
-for batch in win_batches:
+  for batch in win_batches:
 
-  threads = []
+    threads = []
 
-  for vm in batch:
-      ip = vm['public_ip']
-      open_port = check_port(ip, 3389)
-      id = vm['server_id']
-      user = vm['provision_user']
-      date = vm['provision_date']
+    for vm in batch:
+        ip = vm['public_ip']
+        open_port = check_port(ip, 3389)
+        id = vm['server_id']
+        user = vm['provision_user']
+        date = vm['provision_date']
+    
+        if open_port:
+          windows_table.add_row(str(id), str(ip), str(open_port), str(date), str(user))
 
-      if open_port:
-        windows_table.add_row(str(id), str(ip), str(open_port), str(date), str(user))
+  console = Console()
+  console.print(windows_table)
 
+try:
+  nix_host_scan()
+  win_host_scan()
 
-console = Console()
+except:
+  print("Error: unable to start thread")
 
-console.print(linux_table)
-console.print(windows_table)
+#