diff options
| -rw-r--r-- | src/backend/setbranch.py | 30 |
1 files changed, 20 insertions, 10 deletions
diff --git a/src/backend/setbranch.py b/src/backend/setbranch.py index 249f3f8..ffec3b8 100644 --- a/src/backend/setbranch.py +++ b/src/backend/setbranch.py @@ -3,11 +3,13 @@ import animation import git import os +import random import signal import sys import sisyphus.checkenv import sisyphus.getcolor import sisyphus.getfs +import sisyphus.mirrors import sisyphus.purgeenv import sisyphus.setjobs import sisyphus.setprofile @@ -79,25 +81,33 @@ def ins_p_cfg_repo(branch, remote): sisyphus.getfs.p_cfg_dir, depth=1, branch=branch) -def brch_s_warn(branch, remote): +def set_brch_master_index(): + mirrorList = sisyphus.mirrors.getList() + odd_indices = [i + 1 for i in range(len(mirrorList)) if (i + 1) % 2 == 1] + chosen_index = random.choice(odd_indices) + sisyphus.mirrors.setActive(chosen_index) + + +def set_brch_next_index(): + mirrorList = sisyphus.mirrors.getList() + even_indices = [i + 1 for i in range(len(mirrorList)) if (i + 1) % 2 == 0] + chosen_index = random.choice(even_indices) + sisyphus.mirrors.setActive(chosen_index) + + +def set_bhst_index(branch, remote): if "master" in branch: print(sisyphus.getcolor.green + "\nActive branch switched:" + " " + sisyphus.getcolor.reset + "'" + branch + "'") print(sisyphus.getcolor.green + "Active remote switched:" + " " + sisyphus.getcolor.reset + "'" + remote + "'") - print(sisyphus.getcolor.bright_yellow + "\nUse" + sisyphus.getcolor.reset + " " + "'" + "sisyphus mirror set 3" + "'" + " " + sisyphus.getcolor.bright_yellow + - "or" + sisyphus.getcolor.reset + " " + "'" + "sisyphus mirror set 7" + "'" + " " + sisyphus.getcolor.bright_yellow + "to pair the binhost" + sisyphus.getcolor.reset) - print(sisyphus.getcolor.bright_yellow + "Use" + sisyphus.getcolor.reset + " " + "'" + - "sisyphus branch --help" + "'" + " " + sisyphus.getcolor.bright_yellow + "for help" + sisyphus.getcolor.reset) + set_brch_master_index() # Set binhost index for branch 'master' (random odd index) elif "next" in branch: print(sisyphus.getcolor.green + "\nActive branch switched:" + " " + sisyphus.getcolor.reset + "'" + branch + "'") print(sisyphus.getcolor.green + "Active remote switched:" + " " + sisyphus.getcolor.reset + "'" + remote + "'") - print(sisyphus.getcolor.bright_yellow + "\nUse" + sisyphus.getcolor.reset + " " + "'" + "sisyphus mirror set 4" + "'" + " " + sisyphus.getcolor.bright_yellow + - "or" + sisyphus.getcolor.reset + " " + "'" + "sisyphus mirror set 8" + "'" + " " + sisyphus.getcolor.bright_yellow + "to pair the binhost" + sisyphus.getcolor.reset) - print(sisyphus.getcolor.bright_yellow + "Use" + sisyphus.getcolor.reset + " " + "'" + - "sisyphus branch --help" + "'" + " " + sisyphus.getcolor.bright_yellow + "for help" + sisyphus.getcolor.reset) + set_brch_next_index() # Set binhost index for branch 'next' (random even index) def start(branch, remote): @@ -112,7 +122,7 @@ def start(branch, remote): ins_p_cfg_repo(branch, remote) sisyphus.setjobs.start() sisyphus.setprofile.start() - brch_s_warn(branch, remote) + set_bhst_index(branch, remote) else: print(sisyphus.getcolor.bright_red + "\nNo internet connection; Aborting!\n" + sisyphus.getcolor.reset) |