python使用自己封装的库

我想做一个随机生成用户信息的工具库：人名随机，性别随机，还有粉丝随机。中国人名分姓和名，我用first_name.txt存姓，用last_name.txt存名

分别解析2个文件，得到我们需要的数据，代码实现：

ln_path = '/Users/cykj/DeskTop/last_name.txt'
fn_path = '/Users/cykj/DeskTop/first_name.txt'

fn = []
ln1 = []
ln2 = []

#获取姓
with open(fn_path,'r') as f:
    for line in f.readlines():
        fn.append(line.split('\n')[0])

print(fn)

#获取名
with open(ln_path,'r') as f:
    for line in f.readlines():
        if len(line.split('\n')[0]) == 1:
            #print(len(line.split('\n')[0]))
            ln1.append(line.split('\n')[0])
        else:
            ln2.append(line.split('\n')[0])

print(ln1)
print('='*70)
print(ln2)

#用户信息基类
import  random
class FakeUser:
    def fake_name(self,amount = 1,one_word = False,two_words = False):
        n = 0
        while n <= amount:
            if one_word:
                full_name = random.choice(fn) +random.choice(ln1)
            elif two_words:
                full_name = random.choice(fn) +random.choice(ln2)
            else:
                full_name = random.choice(fn) +random.choice(ln1 + ln2)
            yield  full_name
            n +=1

    def fake_gender(self,amount =1):
        n =0
        while n <= amount:
            gender = random.choice(['男','女','未知'])
            yield  gender
            n +=1

#派生类
class SonUser(FakeUser):
    def get_followers(self,amount,few = True,a_lot = False):
        n = 0
        while n <= amount:
            if few:
                followers = random.randrange(1,50)
            elif a_lot:
                followers = random.randrange(200,10000)
            yield  followers
            n +=1

#调用
user_a = FakeUser()
user_b = SonUser()

print('宋末年间取名法