some small fixes

CTFs_and_WarGames/WARGAMES/krypton/vige.py

@@ -0,0 +1,39 @@
+import sys
+from pygenere import Vigenere, VigCrack
+
+
+def get_key(msg):
+  # Vigenere Cypher
+  key =  VigCrack(msg).crack_codeword()
+  dec_msg = VigCrack(msg).crack_message()
+  dec_msg =  dec_msg.replace(" ", "")
+  return key, dec_msg
+
+
+def solve(msg, key):
+  dec_msg = Vigenere(msg).decipher(key)
+  dec_msg =  dec_msg.replace(" ", "")
+  return dec_msg
+
+
+
+if __name__ == '__main__':
+
+  # getting the key
+  with open('cipher', 'r') as f:
+        msg = f.readlines()
+        msg_in = msg[0].strip()
+        key, answer = get_key(msg_in)
+        print 'Message: ' + msg_in
+        print
+        print 'Answer: ' + answer
+        print '(key: ' + key + ')'
+  
+
+  # deciphering                                 
+  key = 'FREKEY'
+  with open('pass', 'r') as f:
+	msg = f.readlines()
+	answer = solve(msg[0].strip(), key)
+	print
+	print "The answer is: " + answer