begin work on the Great Bomb Solver (tm)

Author: Stinggyray

bomb.py

@@ -0,0 +1,24 @@
+# Contains all the edgework stuff about the bomb.
+
+
+class Bomb:
+	def __init__(self, serial_number: str, lit_indicators: list = None, unlit_indicators: list = None,
+	             batteries: int = 0, holders: int = 0, ports: list = None, module_count: int = 0, time: float = 0):
+		if unlit_indicators is None:
+			unlit_indicators = []
+		if lit_indicators is None:
+			lit_indicators = []
+		if ports is None:
+			ports = []
+
+		self.serial_number = serial_number
+		self.lit_indicators = lit_indicators
+		self.unlit_indicators = unlit_indicators
+		self.holders = holders
+		self.batteries = batteries
+		self.ports = ports
+		self.module_count = module_count
+		self.time = time
+
+		self.aa_batteries = (batteries - holders) * 2
+		self.d_batteries = batteries - self.aa_batteries

main.py

@@ -0,0 +1,32 @@
+from bomb import Bomb
+import modules
+
+
+# The driver file for this project.
+# Here, we execute all code to solve our bombs.
+
+# -------------------#
+# EDGEWORK           #
+# -------------------#
+
+current_bomb = Bomb(
+	"ABCDEF", # serial number
+	lit_indicators=["NSA"],
+	unlit_indicators=[],
+	batteries=2,
+	holders=1,
+
+	ports=[
+		[],
+		["RJ", "RCA"]
+	],
+
+	module_count=15,
+	time=20
+)
+
+
+# -------------------#
+# SOLVING MODULES    #
+# -------------------#
+modules.button.solve_button(current_bomb, "why", "red")

modules/button/__init__.py

@@ -0,0 +1,34 @@
+from bomb import Bomb
+from termcolor import cprint
+
+
+def solve_button(bomb: Bomb, word: str, color: str):
+	# Preprocessing
+	word = word.upper()
+	color = color.upper()
+
+	# True is press and hold
+	# False is press and release
+	result = None
+	if word == "ABORT" and color == "BLUE":
+		result = True
+	elif bomb.batteries > 1 and word == "DETONATE":
+		result = False
+	elif color == "WHITE" and "CAR" in bomb.lit_indicators:
+		result = True
+	elif bomb.batteries > 2 and "FRK" in bomb.lit_indicators:
+		result = False
+	elif color == "YELLOW":
+		result = True
+	elif color == "RED" and word == "HOLD":
+		result = False
+	else:
+		result = True
+
+	if result:
+		print("Press and hold the button. Then observe the strip color.")
+		cprint("Blue: 4", 'blue')
+		cprint("Yellow: 5", 'yellow')
+		print("Anything else: 1")
+	else:
+		print("Press and IMMEDIATELY release.")

modules/complicatedwires/__init__.py

@@ -0,0 +1,2 @@
+from bomb import Bomb
+

standalone_modules/adjacentletters.py

@@ -0,0 +1,79 @@
+# All the adjacent letters in KTANE.
+arrays = {
+	"A": ["GJMOY", "HKPRW"],
+	"B": ["IKLRT", "CDFYZ"],
+	"C": ["BHIJW", "DEMTU"],
+	"D": ["IKOPQ", "CJTUW"],
+	"E": ["ACGIJ", "KSUWZ"],
+	"F": ["CERVY", "AGJPQ"],
+	"G": ["ACFNS", "HOQYZ"],
+	"H": ["LRTUX", "DKMPS"],
+	"I": ["DLOWZ", "EFNUV"],
+	"J": ["BQTUW", "EHIOS"],
+	"K": ["AFPXY", "DIORZ"],
+	"L": ["GKPTZ", "ABRVX"],
+	"M": ["EILQT", "BFPWX"],
+	"N": ["PQRSV", "AFGHL"],
+	"O": ["HJLUZ", "IQSTX"],
+	"P": ["DMNOX", "CFHKR"],
+	"Q": ["CEOPV", "BDIKN"],
+	"R": ["AEGSU", "BNOXY"],
+	"S": ["ABEKQ", "GMVYZ"],
+	"T": ["GVXYZ", "CJLSU"],
+	"U": ["FMVXZ", "BILNY"],
+	"V": ["DHMNW", "AEJQX"],
+	"W": ["DFHMN", "GLQRT"],
+	"X": ["BDFKW", "AJNOV"],
+	"Y": ["BCHSU", "EGMTW"],
+	"Z": ["JNRSY", "CLMPV"],
+}
+
+
+def calc_letters(letterlist):
+	if None in letterlist:
+		print("Invalid arguments. Please try again.")
+		return
+
+	for i in range(3):
+		for j in range(4):
+			cur_lists = arrays[letterlist[i][j]]
+			works = False
+
+			for letter in cur_lists[0]:
+				if (j > 0 and letterlist[i][j - 1] == letter) or (j < 3 and letterlist[i][j + 1] == letter):
+					works = True
+					break
+
+			for letter in cur_lists[1]:
+				if (i > 0 and letterlist[i - 1][j] == letter) or (i < 2 and letterlist[i + 1][j] == letter):
+					works = True
+					break
+
+			print(letterlist[i][j] if works else ".", end="")
+		print()
+
+
+def get_input_row():
+	row = input()
+
+	if len(row) != 4:
+		print("Invalid argument. Please enter that again.")
+		return get_input_row()
+
+	return list(row.upper())
+
+
+print("Input your buttons in this format:")
+print("""ABCD
+EFGH
+IJKL
+""")
+
+letters = [get_input_row(), get_input_row(), get_input_row()]
+
+print("Your input:")
+print(letters)
+print()
+print('Buttons to press:')
+
+calc_letters(letters)