标签：opt encypt encrypt guest dataset host intersect

import getopt
import sys
import uuid
import hashlib
import csv
def getIds(totals,encrypt):
    seed=set()
    for i in range(totals):
        uid=uuid.uuid1()
        if encrypt:
            encrypt_id=hashlib.sha256((str(uid)+"saltValue").encode("utf-8")).hexdigest()
            seed.add(encrypt_id)
        else:

            seed.add(str(uid).replace('-',''))
    return seed
def gen_Datasets():
    guest,host,intersect,encrypt_mode=getDataNum()
    ids=list(getIds(intersect+host+guest,encrypt=encrypt_mode))
    dataset_intersect=ids[:intersect]
    dataset_host=ids[intersect:host]
    dataset_host.extend(dataset_intersect)
    dataset_guest=ids[intersect:guest]
    dataset_guest.extend(dataset_intersect)
    print(len(dataset_intersect),len(dataset_host),len(dataset_guest))

    write_csvtb("./intersect.csv",dataset_intersect)
    write_csvtb("./host_a.csv", dataset_intersect)
    write_csvtb("./guest_b.csv", dataset_intersect)
    print("finish all write ")
    # with open("./intersect.txt","w")as i:
    #     for line in dataset_intersect:
    #        print(line)
    #        i.writelines(line)
    # print("finish intersect========")
    # with open("./host_a.txt","w") as h:
    #     for line in dataset_host:
    #        print(line)
    #        h.write(line)
    # print("finsh host===============")
    # with open("./guest_b.txt","w") as g:
    #     for line in dataset_guest:
    #         # print(line)
    #         g.write(line)
    # print("finish guest=======")
def getDataNum():
    def usage():
        print("examples:-m: --help \n-e or --encrypt=: sha256\n-i or --intersect=: intersect datasets number\n-g or --guest : guest number\n -h or --host : host dataset number ")
    opts, args = getopt.getopt(sys.argv[1:], '-m-i:g:h:e:', ['help', 'host=', 'guest=','intersect=','encrypt='])
    guest,host,intersect=None,None,None
    encrypt = False
    for opt_name, opt_value in opts:

        if opt_name in ('-m', '--help'):
            usage()
            sys.exit()
        if opt_name in ('-g', '--guest'):
           guest = int(opt_value)
        if opt_name in ('-h', '--host'):
            host = int(opt_value)
        if opt_name in ('-i','--intersect'):
            intersect=int(opt_value)
        if opt_name in('-e','--encrypt'):
            if int(opt_value)==1:
                encrypt=True
            elif int(opt_value)==0:
                encrypt=False
            else:
                raise ValueError("encrypt must between 0 and 1")
    return guest,host,intersect,encrypt


def write_csvtb(path,dt):
    with open(path,"w",newline='')as f:
        writer=csv.writer(f)
        for i in dt:
            writer.writerow([i])

if __name__ == '__main__':
    gen_Datasets()

标签：opt,encypt,encrypt,guest,dataset,host,intersect
来源： https://www.cnblogs.com/SunshineKimi/p/11204454.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。