konsthol/SecDep
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Derp. Fix missing constant post rename

Browse Source
This commit is contained in:
konsthol
2023-09-11 00:54:39 +03:00
parent 8824794fc4
commit 5948ad8368
2 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
secdep.py
View File

@@ -1316,14 +1316,16 @@ def create_node(provider, name=None, location=None, size=None, image=None, confi
console.print("Node is initializing, please wait...", style="bold white")
console.print("ip to connect to", style="bold white")
console.print("[bold white]\nIP: %s[/bold white]" % (node.public_ips[0]))
# Here is where we write the node's ip to the file
with open(SECDEP_HOSTS_FILE, "a") as nodesFile:
nodesFile.write(node.public_ips[0]+"\n")
# Here is where we write the node's ip to the file with either :22100 or nothing depending on the deploy parameter
console.print("[u]ssh command:[/u]", style="bold white")
if args.deploy:
console.print("[bold white]\nssh -p 22100 -i %s secdep@%s\n[/bold white]" % (SECDEP_SSH_PRIVATE_KEY, node.public_ips[0]))
with open(SECDEP_HOSTS_FILE, "a") as nodesFile:
nodesFile.write(node.public_ips[0]+":22100\n")
else:
console.print("[bold white]\nssh -i %s secdep@%s\n[/bold white]" % (SECDEP_SSH_PRIVATE_KEY, node.public_ips[0]))
with open(SECDEP_HOSTS_FILE, "a") as nodesFile:
nodesFile.write(node.public_ips[0]+"\n")
status.stop()
return node
@@ -1487,7 +1489,7 @@ def node_action(action, provider, awsRegion=None):
lines = nodesFile.readlines()
with open(SECDEP_HOSTS_FILE, "w") as nodesFile:
for line in lines:
if line.strip("\n") != node.public_ips[0]:
if line.strip("\n") != node.public_ips[0] and line.strip("\n") != node.public_ips[0]+":22100":
nodesFile.write(line)
case _:
console.print("[u]Invalid[/u] action command", style="bold red")
@@ -1535,7 +1537,7 @@ def node_action_all(action, provider, awsRegion=None):
lines = nodesFile.readlines()
with open(SECDEP_HOSTS_FILE, "w") as nodesFile:
for line in lines:
if line.strip("\n") != node.public_ips[0]:
if line.strip("\n") != node.public_ips[0] and line.strip("\n") != node.public_ips[0]+":22100":
nodesFile.write(line)
case _:
console.print("[u]Invalid[/u] action command", style="bold red")