ejung/sec_ex01
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix file path bug and add name==main

Browse Source
This commit is contained in:
Eggert Jung
2020-11-25 08:10:49 +01:00
parent ecb3157a91
commit 30768b4592
1 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/vigenere/break_vig.py
View File

@@ -28,13 +28,14 @@ def crack(substrings, words):
tree.update(probe(curr, substrings, words))
parser = argparse.ArgumentParser()
parser.add_argument('FILE')
parser.add_argument('--keylen', type=int, metavar="INT", required=True)
args = parser.parse_args()
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument('FILE')
parser.add_argument('--keylen', type=int, metavar="INT", required=True)
args = parser.parse_args()
t = ''.join([x for x in open(args.FILE, "r").read().lower() if x.isalpha()])
substrings = [(t[i:i+args.keylen]) for i in range(0, len(t), args.keylen)]
words = open(os.path.dirname(__file__)+"/words.txt", "r").read()
t = ''.join([x for x in open(args.FILE, "r").read().lower() if x.isalpha()])
substrings = [(t[i:i+args.keylen]) for i in range(0, len(t), args.keylen)]
words = open(os.path.abspath(os.path.dirname(__file__))+"/words.txt", "r").read()
print(crack(substrings, words))
print(crack(substrings, words))