password/veiji1.1.py at master · qjm100/password · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
#_*_coding:utf-8_*_
import getopt
import re
from string import ascii_lowercase as lowercase
import sys
try:
    import pyfiglet
except ImportError:
    print ("请安装pyfiglet 请运行 pip install pyfiglet")

ke = None
def En(p, key):
    p = get_trim_text(p)
    ptLen = len(p)
    keyLen = len(key)

    quotient = ptLen // keyLen  # 商
    remainder = ptLen % keyLen  # 余

    out = ""
    for i in range(0, quotient):
        for j in range(0, keyLen):
            c = int((ord(p[i * keyLen + j]) - ord('a') + ord(key[j]) - ord('a')) % 26 + ord('a'))

            out += chr(c)

    for i in range(0, remainder):
        c = int((ord(p[quotient * keyLen + i]) - ord('a') + ord(key[i]) - ord('a')) % 26 + ord('a'))
        out += chr(c)

    return out

def De(output, key):
    ptLen = len(output)
    keyLen = len(key)

    quotient = ptLen // keyLen
    remainder = ptLen % keyLen

    inp = ""

    for i in range(0, quotient):
        for j in range(0, keyLen):
            c = int((ord(output[i * keyLen + j]) - ord('a') - (ord(key[j]) - ord('a'))) % 26 + ord('a'))
            # global input
            inp += chr(c)

    for i in range(0, remainder):
        c = int((ord(output[quotient * keyLen + i]) - ord('a') - (ord(key[i]) - ord('a'))) % 26 + ord('a'))
        # global input
        inp += chr(c)
        return inp
def helpyou ():
    print ("欢迎来到维吉尼亚\n-h --help 显示帮助\n-k --key 必要参数，秘钥\n-e --jiami输入加密信息\n-d --jiemi 输入要解密信息")

def get_trim_text(text):
    text = text.lower()
    trim_text = ''
    for l in text:
        if lowercase.find(l) >= 0:
            trim_text += l
    return trim_text
def reg (r):
    v = re.split(' ',r)
    for i in v:
        i += i
    return i

if __name__ == '__main__':


    a = pyfiglet.Figlet (font='slant')
    print (a.renderText ('virginia'))

    try:

        longargs = ['help=','jiami=','jiemi=','key=']
        opts,args= getopt.getopt( sys.argv[1:], 'he:d:k:', longargs)

        b = True
        for o, a in opts:
            if o in ("--key","-k"):
                ke = a

            if o in ("--help" ,"-h"):
                helpyou()
                sys.exit()
            if o in ("--jiami" ,"-e"):
                if ke == None:
                    print ("无秘钥，请确保输入秘钥！")
                    break
                jiami = reg (a)
                print (En (jiami,ke))
                b = False
            if o in ("--jiemi" ,"-d"):
                if ke == None:
                    print ("无秘钥，请确保输入秘钥！")
                    break

                jiemi = reg(a)
                print (De (jiemi,ke))
                b = False


    except getopt.GetoptError:
        helpyou ()

    if b == True:
        helpyou ()