paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-10-28 18:43:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

droplets: Make assert_droplet_does_not_exist take droplet_name as argument.

Browse Source
This commit is contained in:
Vishnu KS
2021-03-18 00:27:37 +05:30
committed by Tim Abbott
parent 13b9cf200b
commit fa623dbd62
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tools/droplets/create.py
View File

@@ -86,20 +86,20 @@ def fork_exists(username: str) -> bool:
sys.exit(1) sys.exit(1)
def assert_droplet_does_not_exist(my_token: str, username: str, recreate: bool) -> None: def assert_droplet_does_not_exist(my_token: str, droplet_name: str, recreate: bool) -> None:
print(f"Checking to see if droplet for {username} already exists...") print(f"Checking to see if droplet {droplet_name} already exists...")
manager = digitalocean.Manager(token=my_token) manager = digitalocean.Manager(token=my_token)
my_droplets = manager.get_all_droplets() my_droplets = manager.get_all_droplets()
for droplet in my_droplets: for droplet in my_droplets:
if droplet.name.lower() == f"{username}.zulipdev.org": if droplet.name.lower() == droplet_name:
if not recreate: if not recreate:
print( print(
"Droplet for user {} already exists. Pass --recreate if you " "Droplet {} already exists. Pass --recreate if you "
"need to recreate the droplet.".format(username) "need to recreate the droplet.".format(droplet_name)
) )
sys.exit(1) sys.exit(1)
else: else:
print(f"Deleting existing droplet for {username}.") print(f"Deleting existing droplet {droplet_name}.")
droplet.destroy() droplet.destroy()
return return
print("...No droplet found...proceeding.") print("...No droplet found...proceeding.")
@@ -270,7 +270,9 @@ if __name__ == "__main__":
fork_exists(username=username) fork_exists(username=username)
api_token = config["digitalocean"]["api_token"] api_token = config["digitalocean"]["api_token"]
assert_droplet_does_not_exist(my_token=api_token, username=username, recreate=args.recreate) assert_droplet_does_not_exist(
my_token=api_token, droplet_name=droplet_domain_name, recreate=args.recreate
)
# set user_data # set user_data
user_data = set_user_data(username=username, userkey_dicts=public_keys) user_data = set_user_data(username=username, userkey_dicts=public_keys)