githuba
diff --git a/‎ciphers/XOR_cipher.py‎
Lines changed: 33 additions & 63 deletions b/‎ciphers/XOR_cipher.py‎
Lines changed: 33 additions & 63 deletions
@@ -38,20 +38,14 @@ def encrypt(self, content, key):
 		# precondition
 		assert (isinstance(key,int) and isinstance(content,str))
 
-		# testing for default arguments
-		if (key == 0):
-			if (self.__key == 0):
-				key = 1
-			else:
-				key = self.__key
+		key = key or self.__key or 1
 
 		# make sure key can be any size
 		while (key > 255):
 			key -= 255
 
 		# This will be returned
 		ans = []
-		resultNumber = 0
 
 		for ch in content:
 			ans.append(chr(ord(ch) ^ key))
@@ -69,20 +63,14 @@ def decrypt(self,content,key):
 		# precondition
 		assert (isinstance(key,int) and isinstance(content,list))
 
-		# testing for default arguments
-		if (key == 0):
-			if (self.__key == 0):
-				key = 1
-			else:
-				key = self.__key
+		key = key or self.__key or 1
 
 		# make sure key can be any size
 		while (key > 255):
 			key -= 255
 
 		# This will be returned
 		ans = []
-		resultNumber = 0
 
 		for ch in content:
 			ans.append(chr(ord(ch) ^ key))
@@ -101,20 +89,14 @@ def encrypt_string(self,content, key = 0):
 		# precondition
 		assert (isinstance(key,int) and isinstance(content,str))
 
-		# testing for default arguments
-		if (key == 0):
-			if (self.__key == 0):
-				key = 1
-			else:
-				key = self.__key
+		key = key or self.__key or 1
 
 		# make sure key can be any size
 		while (key > 255):
 			key -= 255
 
 		# This will be returned
 		ans = ""
-		resultNumber = 0
 
 		for ch in content:
 			ans += chr(ord(ch) ^ key)
@@ -132,21 +114,15 @@ def decrypt_string(self,content,key = 0):
 		# precondition
 		assert (isinstance(key,int) and isinstance(content,str))
 
-		# testing for default arguments
-		if (key == 0):
-			if (self.__key == 0):
-				key = 1
-			else:
-				key = self.__key
+		key = key or self.__key or 1
 
 		# make sure key can be any size
 		while (key > 255):
 			key -= 255
 
 		# This will be returned
 		ans = ""
-		resultNumber = 0
-
+		
 		for ch in content:
 			ans += chr(ord(ch) ^ key)
 
@@ -166,15 +142,12 @@ def encrypt_file(self, file, key = 0):
 		assert (isinstance(file,str) and isinstance(key,int))
 
 		try:
-			fin = open(file,"r")
-			fout = open("encrypt.out","w+")
-
-			# actual encrypt-process
-			for line in fin:
-				fout.write(self.encrypt_string(line,key))
+			with open(file,"r") as fin:
+				with open("encrypt.out","w+") as fout:
 
-			fin.close()
-			fout.close()
+					# actual encrypt-process
+					for line in fin:
+						fout.write(self.encrypt_string(line,key))
 
 		except:
 			return False
@@ -195,15 +168,12 @@ def decrypt_file(self,file, key):
 		assert (isinstance(file,str) and isinstance(key,int))
 
 		try:
-			fin = open(file,"r")
-			fout = open("decrypt.out","w+")
-
-			# actual encrypt-process
-			for line in fin:
-				fout.write(self.decrypt_string(line,key))
+			with open(file,"r") as fin:
+				with open("decrypt.out","w+") as fout:
 
-			fin.close()
-			fout.close()
+					# actual encrypt-process
+					for line in fin:
+						fout.write(self.decrypt_string(line,key))
 
 		except:
 			return False
@@ -214,26 +184,26 @@ def decrypt_file(self,file, key):
 
 
 # Tests
-# crypt = XORCipher()
-# key = 67
+crypt = XORCipher()
+key = 67
 
-# # test enrcypt
-# print crypt.encrypt("hallo welt",key)
-# # test decrypt
-# print crypt.decrypt(crypt.encrypt("hallo welt",key), key)
+# test enrcypt
+print crypt.encrypt("hallo welt",key)
+# test decrypt
+print crypt.decrypt(crypt.encrypt("hallo welt",key), key)
 
-# # test encrypt_string
-# print crypt.encrypt_string("hallo welt",key)
+# test encrypt_string
+print crypt.encrypt_string("hallo welt",key)
 
-# # test decrypt_string
-# print crypt.decrypt_string(crypt.encrypt_string("hallo welt",key),key)
+# test decrypt_string
+print crypt.decrypt_string(crypt.encrypt_string("hallo welt",key),key)
 
-# if (crypt.encrypt_file("test.txt",key)):
-# 	print "encrypt successful"
-# else:
-# 	print "encrypt unsuccessful"
+if (crypt.encrypt_file("test.txt",key)):
+	print "encrypt successful"
+else:
+	print "encrypt unsuccessful"
 
-# if (crypt.decrypt_file("a.out",key)):
-# 	print "decrypt successful"
-# else:
-# 	print "decrypt unsuccessful"
+if (crypt.decrypt_file("encrypt.out",key)):
+	print "decrypt successful"
+else:
+	print "decrypt unsuccessful"