Kumi
776e3e00ea
Removed the --provider argument from the command-line parser and set the provider variable directly to "hetzner". Simplifies the script by removing an unnecessary user input field. Relevant for deployments exclusively using Hetzner.
67 lines
No EOL
2.3 KiB
Python
67 lines
No EOL
2.3 KiB
Python
import subprocess
|
|
import configparser
|
|
from hcloud import Client
|
|
import argparse
|
|
import pathlib
|
|
|
|
UPDATE_SCRIPT_PATH = pathlib.Path(__file__).parent / "update_local_config.py"
|
|
CONFIG_PATH = pathlib.Path(__file__).parent / "config.ini"
|
|
|
|
def get_hetzner_client():
|
|
config = configparser.ConfigParser()
|
|
config.read(CONFIG_PATH)
|
|
api_token = config["hetzner"]["api_token"]
|
|
return Client(token=api_token)
|
|
|
|
def list_hetzner_servers(client):
|
|
return client.servers.get_all()
|
|
|
|
def delete_hetzner_server(client, server):
|
|
server.delete()
|
|
|
|
def run_update_script(provider, location, server_type, interface, config_path):
|
|
command = [
|
|
"python", UPDATE_SCRIPT_PATH,
|
|
"--provider", provider,
|
|
"--location", location,
|
|
"--server_type", server_type,
|
|
"--interface", interface
|
|
]
|
|
|
|
if config_path:
|
|
command.extend(["--config_path", config_path])
|
|
|
|
result = subprocess.run(command, capture_output=True, text=True)
|
|
if result.returncode != 0:
|
|
raise Exception(f"Error executing update script: {result.stderr}")
|
|
print(result.stdout)
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="Manage Hetzner servers and update local Wireguard VPN.")
|
|
parser.add_argument("--location", type=str, required=True, help="Server location")
|
|
parser.add_argument("--server_type", type=str, required=True, help="Server type")
|
|
parser.add_argument("--interface", type=str, required=True, help="Wireguard interface (e.g., wg0)")
|
|
parser.add_argument("--config_path", type=str, help="Path to the Wireguard configuration file")
|
|
|
|
args = parser.parse_args()
|
|
|
|
provider = "hetzner"
|
|
location = args.location
|
|
server_type = args.server_type
|
|
interface = args.interface
|
|
config_path = args.config_path
|
|
|
|
# Step 1: Get Hetzner client and list existing servers
|
|
client = get_hetzner_client()
|
|
existing_servers = list_hetzner_servers(client)
|
|
|
|
# Step 2: Run the update_local_config.py script
|
|
run_update_script(provider, location, server_type, interface, config_path)
|
|
|
|
# Step 3: Delete old servers that existed before running the update script
|
|
for server in existing_servers:
|
|
delete_hetzner_server(client, server)
|
|
print(f"Deleted server: {server.name}")
|
|
|
|
if __name__ == "__main__":
|
|
main() |