66 lines
2.0 KiB
Python
66 lines
2.0 KiB
Python
|
|
import base64, hashlib
|
|
import subprocess
|
|
import ipaddress
|
|
from typing import Dict, Any, List
|
|
|
|
# pickle subroutines
|
|
import pickle
|
|
from pathlib import Path
|
|
|
|
print("Importing Helpers")
|
|
# subnet helper app
|
|
def is_ip_in_subnets(ip, subnet):
|
|
try:
|
|
ip_obj = ipaddress.IPv4Address(ip)
|
|
subnet_obj = ipaddress.IPv4Network(subnet)
|
|
if ip_obj in subnet_obj:
|
|
return True
|
|
return False
|
|
except ValueError as e:
|
|
# If the IP address is not valid, raise an error
|
|
return False
|
|
|
|
# subroutine to run a command, return stdout as array unless zero_only then return [0]
|
|
def run_command(cmd, zero_only=False, use_shell=True, req_check = True):
|
|
# Run the command and capture the output
|
|
result = subprocess.run(cmd, shell=use_shell, check=req_check, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
# Decode the byte output to a string
|
|
output = result.stdout.decode('utf-8')
|
|
# Split the output into lines and store it in an array
|
|
output_lines = [line for line in output.split('\n') if line]
|
|
# Return result
|
|
try:
|
|
return output_lines[0] if zero_only else output_lines
|
|
except:
|
|
return output_lines
|
|
|
|
def short_uuid(value: str, length=8):
|
|
hasher = hashlib.md5()
|
|
hasher.update(value.encode('utf-8'))
|
|
full_hex = hasher.hexdigest()
|
|
return full_hex[:length]
|
|
|
|
# test subroutine
|
|
def get_hostname():
|
|
hostname_command = "hostname"
|
|
return run_command(hostname_command, zero_only = True)
|
|
|
|
|
|
# pickle helpers
|
|
# Where the pickled state will live
|
|
STATE_FILE = Path(__file__).parent / "storage_api_state.pkl"
|
|
|
|
def save_state(obj: object, path: Path | str = STATE_FILE) -> None:
|
|
path = Path(path)
|
|
path.parent.mkdir(parents=True, exist_ok=True)
|
|
with path.open("wb") as f:
|
|
pickle.dump(obj, f, protocol=pickle.HIGHEST_PROTOCOL)
|
|
print("Pickle saved")
|
|
|
|
def load_state(path: Path | str = STATE_FILE) -> object | None:
|
|
path = Path(path)
|
|
if path.is_file():
|
|
with path.open("rb") as f:
|
|
return pickle.load(f)
|
|
return None |