NameGenerator.py 文件源码

def generatesurname(species):
    # check species
    if species == 1:
        sl=sum(1 for line in open('assets/Generators/NameGenerator/HumanSurnames.txt'))
        sr = random.randrange(sl)+1
        sn = linecache.getline('assets/Generators/NameGenerator/HumanSurnames.txt', sr)
        # Delete Newline at end
    elif species == 2:
        sl=sum(1 for line in open('assets/Generators/NameGenerator/RobotSurnames.txt'))
        sr = random.randrange(sl)+1
        sn = linecache.getline('assets/Generators/NameGenerator/RobotSurnames.txt', sr)
        # Delete Newline at end
    else:
        # error msg
        print('please specify species for name generator')
        return
    linecache.clearcache()
    sm = len(sn)-1
    lastname = sn[:sm]
    return lastname
# newname = Name('human',2)
# newsurname = Surname('human')
# print(newname,newsurname)