mirror of
https://github.com/viq/NewsBlur.git
synced 2025-04-13 09:38:09 +00:00
64 lines
1.6 KiB
Python
Executable file
64 lines
1.6 KiB
Python
Executable file
#!/usr/bin/env python
|
|
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import time
|
|
|
|
import digitalocean
|
|
|
|
OLD = False
|
|
# Set env var OLD=1 to use existing servers
|
|
if os.environ.get("OLD", False):
|
|
OLD = True
|
|
|
|
if OLD:
|
|
TOKEN_FILE = "/srv/secrets-newsblur/keys/digital_ocean.readprod.token"
|
|
else:
|
|
TOKEN_FILE = "/srv/secrets-newsblur/keys/digital_ocean.token"
|
|
|
|
try:
|
|
api_token = open(TOKEN_FILE, "r").read().strip()
|
|
except IOError:
|
|
print(f" ---> Missing Digital Ocean API token: {TOKEN_FILE}")
|
|
exit()
|
|
|
|
outfile = f"/srv/newsblur/ansible/inventories/digital_ocean{'.old' if OLD else ''}.ini"
|
|
|
|
# Install from https://github.com/do-community/do-ansible-inventory/releases
|
|
ansible_inventory_cmd = f"do-ansible-inventory -t {api_token} --out {outfile}"
|
|
subprocess.call(ansible_inventory_cmd, shell=True)
|
|
|
|
with open(outfile, "r") as original:
|
|
data = original.read()
|
|
with open(outfile, "w") as modified:
|
|
modified.write("127.0.0.1 ansible_connection=local\n" + data)
|
|
|
|
exit() # Too many requests if we run the below code
|
|
|
|
do = digitalocean.Manager(token=api_token)
|
|
droplets = do.get_all_droplets()
|
|
|
|
print("\n ---> Checking droplets: %s\n" % (" ".join([d.name for d in droplets])))
|
|
|
|
|
|
def check_droplets_created():
|
|
i = 0
|
|
droplets = do.get_all_droplets()
|
|
|
|
for instance in droplets:
|
|
if instance.status == "new":
|
|
print(".", end=" ")
|
|
sys.stdout.flush()
|
|
i += 1
|
|
time.sleep(i)
|
|
break
|
|
else:
|
|
print(" ---> All booted!")
|
|
return True
|
|
|
|
|
|
i = 0
|
|
while True:
|
|
if check_droplets_created():
|
|
break
|