Scrambler __str__

Author: denismaggior8

examples/render_enigma.py

@@ -100,7 +100,8 @@ def render_enigma_diagram(enigma):
 etw = EnigmaZEtw()
 enigma = EnigmaZ(rotor3, rotor2, rotor1, reflector, etw, True)
 enigma.input_string('0'*41)
-Utils.render_enigma_diagram(enigma)
+print(enigma.reflector)
+#Utils.render_enigma_diagram(enigma)
 #render_enigma_diagram(enigma)
 
 plugboard = PlugboardPassthrough()
@@ -113,8 +114,9 @@ def render_enigma_diagram(enigma):
 reflector = ReflectorUKWBThin()
 etw = EtwPassthrough()
 enigma = EnigmaM4(plugboard, rotor1, rotor2, rotor3, rotor4, reflector, etw, True)
-print(enigma.input_string('d'))
-Utils.render_enigma_diagram(enigma)
+#print(enigma.input_string('d'))
+print(enigma.reflector)
+#Utils.render_enigma_diagram(enigma)
 #render_enigma_diagram(enigma)
 
 # Setup logging
@@ -130,8 +132,8 @@ def render_enigma_diagram(enigma):
 enigma = EnigmaM3(plugboard,rotor3, rotor2, rotor1,reflector,etw,True)
 other_enigma = copy.deepcopy(enigma)
 enigma.input_string('d')
-print(enigma.plugboard.journal)
-Utils.render_enigma_diagram(enigma)
+print(enigma.reflector)
+#Utils.render_enigma_diagram(enigma)
 #render_enigma_diagram(enigma)
 #render_enigma_diagram(other_enigma)
 #Utils.render_enigma_diagram(enigma)

src/enigma/enigmapython/Scrambler.py

@@ -57,4 +57,7 @@ def set_scrambler_ring(self, ring):
  def __shift(letter, shift, alphabet_list):
  for i in range(0, len(alphabet_list)):
  if alphabet_list[i] == letter:
- return alphabet_list[(i + shift) % len(alphabet_list)]
+ return alphabet_list[(i + shift) % len(alphabet_list)]
+ 
+ def __str__(self):
+ return self.wiring