python每周一练：人生模拟器（三）

import random
import time
from random import randint as ran

def removeN():
    b1 = []
    b2 = []
    b3 = []
    b4 = []
    b5 = []
    with open('whiteTalent.txt','r') as usrFile:
        asd = usrFile.readlines()
        for j in range(len(asd)):
            c = str(asd[j])
            asd[j] = c[0:len(c)-1]
        b1 = asd
    with open('greenTalent.txt','r') as usrFile:
        asd = usrFile.readlines()
        for j in range(len(asd)):
            c = str(asd[j])
            asd[j] = c[0:len(c)-1]
        b2 = asd
    with open('blueTalent.txt','r') as usrFile:
        asd = usrFile.readlines()
        for j in range(len(asd)):
            c = str(asd[j])
            asd[j] = c[0:len(c)-1]
        b3 = asd
    with open('purpleTalent.txt','r') as usrFile:
        asd = usrFile.readlines()
        for j in range(len(asd)):
            c = str(asd[j])
            asd[j] = c[0:len(c)-1]
        b4 = asd
    with open('goldenTalent.txt','r') as usrFile:
        asd = usrFile.readlines()
        for j in range(len(asd)):
            c = str(asd[j])
            asd[j] = c[0:len(c)-1]
        b5 = asd
    return b1,b2,b3,b4,b5

point_list = [0,0,0,0]
spe_point = [0.0]
hidden_point_list = [0,0]
total_point = 30
while True:
    print("----------beauty:{}----------".format(point_list[0]))
    print("-------intelligence:{}-------".format(point_list[1]))
    print("--------background:{}--------".format(point_list[2]))
    print("---------physique:{}---------".format(point_list[3]))
    print("Remaining: {} points".format(total_point))
    process = input("Action(add point/remove point/confirm/random distribution): ")
    if process == "add point":
        print("You have {} points to spend, choose which you want to add on.".format(total_point))
        choose = input("beauty/intelligence/background/physique: ")
        choP = int(input("points you want to add on: "))
        if choP <= total_point:
            if choose == "beauty":
                point_list[0] += choP
                total_point -= choP
            elif choose == "intelligence":
                point_list[1] += choP
                total_point -= choP
            elif choose == "background":
                point_list[2] += choP
                total_point -= choP
            elif choose == "physique":
                point_list[3] += choP
                total_point -= choP
            else:
                raise RuntimeError('what do you think you are typing in?')
        else:
            raise RuntimeError('Do some grade one math before you type in!')
    elif process == "remove point":
        print("Choose which you want to remove from.")
        choose = input("beauty/intelligence/background/physique: ")
        choP = int(input("points you want to remove from: "))
        if choose == "beauty":
            if choP<= point_list[0]:
                point_list[0] -= choP
                total_point += choP
            else:
                raise RuntimeError('Do some grade one math before you type in!')
        elif choose == "intelligence":
            if choP<=point_list[1]:
                point_list[1] -= choP
                total_point += choP
            else:
                raise RuntimeError('Do some grade one math before you type in!')
        elif choose == "background":
            if choP<=point_list[2]:
                point_list[2] -= choP
                total_point += choP
            else:
                raise RuntimeError('Do some grade one math before you type in!')
        elif choose == "physique":
            if choP<=point_list[3]:
                point_list[3] -= choP
                total_point += choP
            else:
                raise RuntimeError('Do some grade one math before you type in!')
        else:
            raise RuntimeError('what do you think you are typing in?')
    elif process == "confirm":
        if total_point == 0:
            print("All points used. \nbeauty:{}, intelligence:{}, background:{}, physique:{}. \nLife simulator initialized.".format(point_list[0],point_list[1],point_list[2],point_list[3]))
            time.sleep(3)
            break
        else:
            print("Still have points that are not used.")
    elif process == "random distribution":
        point_list[0] = 0
        point_list[1] = 0
        point_list[2] = 0
        point_list[3] = 0
        total_point = 30
        ran1 = ran(1,7)
        total_point -= ran1 * 4
        point_list[0] += ran1
        point_list[1] += ran1
        point_list[2] += ran1
        point_list[3] += ran1
        while(total_point>0):
            for i in range(4):
                ran2 = ran(0,total_point//2 if total_point != 1 else total_point)
                total_point -= ran2
                point_list[i] = point_list[i] + ran2
                if total_point <= 0:
                    break

talent1,talent2,talent3,talent4,talent5 = removeN()

given_talent = []
plan = [[4,3,2,1,0],[5,3,2,0,0],[7,3,0,0,0],[3,3,2,2,0],[9,1,0,0,0]]#4,3,2,1,0 ; 5,3,2,0,0 ; 7,3,0,0,0 ; 4,2,2,1,1 ; 3,3,2,2,0 ; 9,1,0,0,0
planed = random.choice(plan)#just being nice
for index, item in enumerate(planed):
    if item != 0:
        if index == 0:
            for i in range(item):
                a = random.choice(talent1)
                if a not in given_talent:
                    given_talent.append(a)
        if index == 1:
            for i in range(item):
                a = random.choice(talent2)
                if a not in given_talent:
                    given_talent.append(a)
        if index == 2:
            for i in range(item):
                a = random.choice(talent3)
                if a not in given_talent:
                    given_talent.append(a)
        if index == 3:
            for i in range(item):
                a = random.choice(talent4)
                if a not in given_talent:
                    given_talent.append(a)
        if index == 4:
            for i in range(item):
                a = random.choice(talent5)
                if a not in given_talent:
                    given_talent.append(a)
print(given_talent)
for i in range(3):
    choice = input("Choose your three talents: ")
    talent.append(choice)

新增：

1. 清除bug

2. enumerate读取list

3. 配置可能分配结果

4. 功能性优化