# -*- coding: utf-8 -*-

# @Time : 2017/5/12 17:40

# @Author : set3rnal

# @Site :

# @File : Hill Cipher.py

# @Software: PyCharm

from numpy import *

Dic = {chr(i + 97): i for i in range(26)}

def decode(pwd, org):

temp = []

result = []

while True:

if len(pwd) % 3 != 0:

pwd.append(pwd[-1])

else:

break

for i in pwd:

temp.append(Dic.get(i))

temp = array(temp)

temp = temp.reshape(len(pwd) / 3, 3)

print "temp:"

print temp

xx = matrix(temp) * org

for j in range(len(pwd) / 3):

for i in range(3):

if (int(xx[j, i]) >= 26):

result.append(chr(xx[j, i] % 26 + 97))

# print xx[j, i] % 26

else:

# print xx[j, i]

result.append(chr(xx[j, i] + 97))

return result

def get_vmatrix(org):

org_adjoin = org.I * linalg.det(org) # .I 返回矩阵的逆矩阵 linalg.det 返回矩阵的行列式

print "org_adjoin"

print org_adjoin

org_det = int(str(abs(linalg.det(org))).split('.')[0])

print "org_det"

print org_det

for i in range(1, 26):

if i * org_det % 26 == 1:

break

org_mod = -org_adjoin * i % 26

org_mod = matrix(org_mod)

print "org_mod"

print org_mod

temp = []

for i in range(org_mod.shape[0]):

for j in range(org_mod.shape[1]):

temp.append(int(str(org_mod[i, j]).split('.')[0]))

org_final = matrix(temp).reshape(org_mod.shape[0], org_mod.shape[1])

print "temp"

print temp

print "org_final"

print org_final

return org_final

if __name__ == '__main__':

pwd = list("fin")

org = matrix(array([[6, 24, 1], [13, 16, 10], [20, 17, 15]]))

org_vm = get_vmatrix(org)

result = decode(pwd, org_vm)

print "".join(result)

# deorg = matrix(array([[8, 5, 10], [21 , 8, 21], [21, 12, 8]]))

# result = decode(result, deorg)

# print "".join(result)

# pwd = "wjamdbkdeibr"

# pwd = list(pwd)

# org = matrix(array([[1, 2, 3], [4, 5, 6], [7, 8, 10]]))

# org_vm = get_vmatrix(org)

# print org_vm

# print "Your flag is :" + "".join(decode(pwd, org_vm))