NTRU_IoT/infidel-code/py_wrappers/URG_PLAINGEN.py
2022-02-01 23:45:47 +07:00

32 lines
614 B
Python

import random
import string
plain_length = []
def randstring(n):
letters = string.digits
return ''.join(random.choice(letters) for i in range(n))
def rand_int(n):
range_start = 10**(n-1)
range_end = (10**n)-1
return random.randint(range_start, range_end)
i = 1
while i <=16 :
plain_length.append(i*1)
i = i + 1
base_name = "plain"
def file_handler(f_name, num, plain):
plain = str(plain)
num = str(num).zfill(2)
f = open("./plain/"+f_name+"_"+num+".txt", "w")
f.write(plain)
for i in plain_length:
plain = rand_int(i)
file_handler(base_name, i, plain)