python_example/enc_dec_example.py at main · nicecoding1/python_example · 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
110
111
112
113
114
115
116
117
importos.path
importstring
importrandom
importpickle

defprint_menu():
print()
print("암호화/복호화 프로그램입니다 ^o^")
print("1. 암호화")
print("2. 복호화")
print("3. 종료")
print()

defrecord_file_in_check(f):
record= []
ifnotos.path.isfile("record.txt"):
withopen("record.txt", "w", encoding="utf8"):
pass
returnFalse
withopen("record.txt", "r", encoding="utf8") asfile:
record=file.read()
ifrecord.find(f) ==-1:
returnFalse
else:
returnTrue

whileTrue:
print_menu()
m=input("원하시는 기능을 선택하세요: ")

ifm=="1":
f=input("암호화를 할 파일 이름을 입력하세요: ")
ifnotos.path.isfile(f):
print("파일이 존재하지 않습니다.")
continue

test=record_file_in_check(f)
iftest:
print("이미 암호화가 완료된 파일입니다.")
continue

enc_key=string.ascii_letters+string.digits
t=list(enc_key)
random.shuffle(t)
enc_value="".join(t)

keylist= {}
foriinrange(len(enc_key)):
keylist[enc_key[i]] =enc_value[i]

f2=f[:f.index('.')]+".pwd"
withopen(f2, "wb") as_f:
pickle.dump(keylist, _f, pickle.HIGHEST_PROTOCOL)

src_data=""
withopen(f, "r", encoding="utf8") as_f:
src_data=_f.read()

enc_data=""
foriinsrc_data:
ifiinkeylist:
enc_data+=keylist[i]
else:
enc_data+=i

withopen(f, "w", encoding="utf8") as_f:
_f.write(enc_data)

withopen("record.txt", "a", encoding="utf8") as_f:
_f.write(f+"\n")

print("암호화를 완료했습니다.")

elifm=="2":
f=input("복호화를 할 파일 이름을 입력하세요: ")
ifnotos.path.isfile(f):
print("파일이 존재하지 않습니다.")
continue

test=record_file_in_check(f)
ifnottest:
print("암호화 되지 않은 파일입니다.")
continue

f2=f[:f.index('.')]+".pwd"

keylist= {}
keyreverse= {}
withopen(f2, "rb") as_f:
keylist=pickle.load(_f)

foriinkeylist:
_v=keylist[i]
keyreverse[_v] =i

enc_data=""
withopen(f, "r", encoding="utf8") as_f:
enc_data=_f.read()

src_data=""
foriinenc_data:
ifiinkeyreverse:
src_data+=keyreverse[i]
else:
src_data+=i

withopen(f, "w", encoding="utf8") as_f:
_f.write(src_data)

print("복호화를 완료했습니다.")

elifm=="3":
break

else:
print("잘못된 입력입니다.")
continue