147 lines
4.1 KiB
Python
147 lines
4.1 KiB
Python
from typing import List
|
|
from models import Customer, Location, Host
|
|
|
|
|
|
def load_customers() -> List[Customer]:
|
|
"""Load customer data. For now, returns mock data but can be replaced later."""
|
|
|
|
# Create hosts for each location
|
|
alpha_hq_hosts = [
|
|
Host(name="Web Server", address="10.10.1.5", type="SSH"),
|
|
Host(name="Admin Panel", address="https://10.10.1.10:8443", type="Web"),
|
|
Host(name="Database", address="10.10.1.20", type="SSH"),
|
|
]
|
|
|
|
alpha_branch_hosts = [
|
|
Host(name="File Server", address="10.10.2.5", type="SMB"),
|
|
Host(name="Backup Server", address="10.10.2.10", type="SSH"),
|
|
]
|
|
|
|
beta_main_hosts = [
|
|
Host(name="Main Portal", address="https://portal.beta.local", type="Web"),
|
|
Host(name="File Server", address="192.168.50.10", type="SMB"),
|
|
]
|
|
|
|
gamma_dc_hosts = [
|
|
Host(name="Application Server", address="172.16.0.100", type="SSH"),
|
|
Host(name="Monitoring Dashboard", address="https://monitor.gamma.net", type="Web"),
|
|
Host(name="Backup Server", address="172.16.0.150", type="SSH"),
|
|
]
|
|
|
|
gamma_dr_hosts = [
|
|
Host(name="DR Server", address="172.16.1.100", type="SSH"),
|
|
Host(name="Backup Storage", address="172.16.1.150", type="SMB"),
|
|
]
|
|
|
|
delta_dev_hosts = [
|
|
Host(name="Development Server", address="10.20.30.40", type="SSH"),
|
|
Host(name="CI/CD Pipeline", address="https://jenkins.delta.local:8080", type="Web"),
|
|
]
|
|
|
|
epsilon_prod_hosts = [
|
|
Host(name="Production API", address="https://api.epsilon.com", type="Web"),
|
|
Host(name="Database Cluster", address="10.5.0.50", type="PostgreSQL"),
|
|
Host(name="Redis Cache", address="10.5.0.60", type="Redis"),
|
|
]
|
|
|
|
epsilon_staging_hosts = [
|
|
Host(name="Staging API", address="https://staging-api.epsilon.com", type="Web"),
|
|
Host(name="Test Database", address="10.5.1.50", type="PostgreSQL"),
|
|
]
|
|
|
|
# Create locations
|
|
alpha_hq = Location(
|
|
name="Headquarters",
|
|
vpn_type="OpenVPN",
|
|
connected=False,
|
|
active=True,
|
|
hosts=alpha_hq_hosts
|
|
)
|
|
|
|
alpha_branch = Location(
|
|
name="Branch Office",
|
|
vpn_type="WireGuard",
|
|
connected=False,
|
|
active=False,
|
|
hosts=alpha_branch_hosts
|
|
)
|
|
|
|
beta_main = Location(
|
|
name="Main Office",
|
|
vpn_type="WireGuard",
|
|
connected=True,
|
|
active=True,
|
|
hosts=beta_main_hosts
|
|
)
|
|
|
|
gamma_dc = Location(
|
|
name="Data Center",
|
|
vpn_type="IPSec",
|
|
connected=False,
|
|
active=False,
|
|
hosts=gamma_dc_hosts
|
|
)
|
|
|
|
gamma_dr = Location(
|
|
name="DR Site",
|
|
vpn_type="OpenVPN",
|
|
connected=False,
|
|
active=False,
|
|
hosts=gamma_dr_hosts
|
|
)
|
|
|
|
delta_dev = Location(
|
|
name="Development Lab",
|
|
vpn_type="OpenVPN",
|
|
connected=False,
|
|
active=False,
|
|
hosts=delta_dev_hosts
|
|
)
|
|
|
|
epsilon_prod = Location(
|
|
name="Production",
|
|
vpn_type="WireGuard",
|
|
connected=False,
|
|
active=False,
|
|
hosts=epsilon_prod_hosts
|
|
)
|
|
|
|
epsilon_staging = Location(
|
|
name="Staging",
|
|
vpn_type="OpenVPN",
|
|
connected=False,
|
|
active=False,
|
|
hosts=epsilon_staging_hosts
|
|
)
|
|
|
|
# Create customers
|
|
customers = [
|
|
Customer(
|
|
name="Customer Alpha Corp",
|
|
locations=[alpha_hq, alpha_branch]
|
|
),
|
|
Customer(
|
|
name="Beta Industries",
|
|
locations=[beta_main]
|
|
),
|
|
Customer(
|
|
name="Gamma Solutions",
|
|
locations=[gamma_dc, gamma_dr]
|
|
),
|
|
Customer(
|
|
name="Delta Tech",
|
|
locations=[delta_dev]
|
|
),
|
|
Customer(
|
|
name="Epsilon Systems",
|
|
locations=[epsilon_prod, epsilon_staging]
|
|
),
|
|
]
|
|
|
|
return customers
|
|
|
|
|
|
def save_customers(customers: List[Customer]) -> None:
|
|
"""Save customer data. Placeholder for future implementation."""
|
|
# TODO: Implement saving to file/database
|
|
pass |