숫자야구

스터디원이 숫자야구 만들어보고 싶다하여 한 번 해봤습니다!

from random import *
import time

def insufficient(numlist): # 부족할 경우
    insufficient_cnt = 4 - len(numlist) # 부족한 개수를 계산
    print(f"중복된 수 혹은 적은 양의 수를 입력하였습니다. 추가로 {insufficient_cnt}개의 다른 수를 입력해주세요.")
    print("현재 입력된 수", end=' : ')
    numlist.sort()
    for i in range(len(numlist)):
        print(numlist[i], end=' ')
    print('')

    while True:
        num = input(": ") # 입력!

        if len(num) != insufficient_cnt: # 부족한 만큼보다 덜 입력했거나 더 입력한 경우(중복포함)
            print(f"다시 입력해주세요. ({insufficient_cnt}개의 숫자 필요)")
            print("현재 입력된 수", end=' : ')
            for i in range(len(numlist)):
                print(numlist[i], end=' ')
            print('')
            continue      
        else: # 부족한 만큼 잘 입력했다면?(중복 고려하지 않고 개수만)
            list_in_inspecting = numlist[:] # 조사리스트에 현재 리스트 넣기
            for i in range(insufficient_cnt): # 입력숫자 넣어주기
                list_in_inspecting.append(num[i])
            list_in_inspecting = list(set(list_in_inspecting)) # 중복제거
        

        if len(list_in_inspecting) == 4: # 부족한 수 입력 시 중복이 없었다면?
            numlist = list_in_inspecting[:]
            break
        else :
            print(f"다시 입력해주세요. (중복되지 않는 숫자 {insufficient_cnt}개를 입력해주세요)")
            continue
    return numlist

def exceed(numlist): # 넘칠 경우
    exceed_cnt = len(numlist) - 4 # 넘친 개수를 계산
    print(f"수를 초과하여 입력하였습니다. 제외할 수 {exceed_cnt}개를 입력해주세요.")
    while True:
        print("현재 입력된 수", end=' : ')
        numlist.sort()
        for i in range(len(numlist)):
            print(numlist[i], end=' ')
        print("")

        num = input(": ")
        if len(num) != exceed_cnt: # 넘친 만큼 보다 덜 혹은 더 입력했을 경우
            print(f"다시 입력해주세요. ({exceed_cnt}개의 숫자를 입력해주세요.)")
            continue # 다음 반복으로 넘기기
        
        # 넘친 만큼 잘 입력했다면?
        for i in range(exceed_cnt):
            if num[i] in numlist:
                numlist.remove(num[i])
        if len(numlist) == 4: # 4자리의 수가 완성이 되었다면
            result = numlist[:]
            break # 반복문 끝
        else :
            print("다시 입력해주세요. (제외할 수를 정확히 입력해주세요)") # 이미 있는 수를 입력하지 않았을 경우
            continue
    return result
        

def inspect(numlist):
    numlist = list(set(numlist))
    if len(numlist) < 4:
        result = insufficient(numlist)
        result.sort()
        print("입력된 수", end=' : ')
        for i in range(4):
            print(result[i], end=' ')
        print('')
        return result
    elif len(numlist) > 4:
        result = exceed(numlist)
        result.sort()
        print("입력된 수", end=' : ')
        for i in range(4):
            print(result[i], end=' ')
        print('')
        return result
    else:
        result = numlist
        return result

def insert():
    num = input("4자리 숫자를 중복없이 입력해주세요. [ex) 1234] : ")
    num = str(int(num))
    numlist = []
    for i in range(len(num)):
        numlist.append(num[i])
    numlist = list(set(numlist))
    numlist.sort()
    return inspect(numlist)

def answer():
    base = []
    for i in range(4):
        num = str(randint(1,9)) # 1
        while num in base:
            num = str(randint(1,9))
        base.append(num)
    base.sort()
    return base

def strike_ball(userchoice, base):
    strike = 0; ball = 0
    for i in range(4):
        if userchoice[i] == base[i]:
            strike += 1
        elif userchoice[i] in base:
            ball += 1
    for i in range(3):
        print('.', end=' ')
        time.sleep(0.7)
    print(f'\n{strike}스트라이크 {ball}볼')
    time.sleep(0.7)
    

#ㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡ
# main
start = 1
info = """
|게임 규칙|----------------------------------------------------
1.총 도전 횟수는 10회 입니다                               
2.숫자를 입력하면 '스트라이크 카운트'와 '볼 카운트'가 나옵니다.
(숫자를 정확하게 적어주세요!)
---------------------------------------------------------------
"""
while start == 1:
    print("---------------------------------------------------------------")
    print("<혼자 즐기는 숫자 야구 게임>")

    print(info, '테스트용 입니다. 지우면됨!') # 정답부분(테스트용)

    for i in range(3):
        print('.', end=' ')
        time.sleep(0.7)
    print("\n게임을 시작 하시겠습니까? Y/N ")
    while True:
        gamestart = input(": ").upper()
        if gamestart == "Y" or 'N':
            break
        else:
            print("다시 입력해주세요.")
            continue
    if gamestart == 'Y':
        base = answer()
        print(base)
        print("시작!")
        cnt = 0
        while cnt < 10:
            cnt += 1
            print("---------------------------------------------------------------")
            print(f"{cnt}번째 시도")
            userchoice = insert()
            userchoice.sort()
            if userchoice == base:
                print(f"성공!")
                break
            else:
                strike_ball(userchoice,base)
                continue
        if cnt == 10:
            time.sleep(0.7)
            print("---------------------------------------------------------------")
            print("저런... 10번의 기회 안에 숫자를 맞추지 못했습니다.")
            time.sleep(0.7)
            print("정답은 바로!")
            for i in range(4):
                print(base[i], end=' ')
                time.sleep(0.7)
            print("수고하셨습니다.")
            print("GAME OVER")
            time.sleep(2)
                
            
    elif gamestart == 'N':
        print("END")
        start = 0
    else:
        print("다시 입력해주세요.")

스터디에서 아직 예외처리나 클래스 부분을 다루지 않아서 일단 함수로만 구현을 해보았는데

나중에 이걸 다시 본다면 아마 지저분한 부분이 많지 않을까 라는 느낌??