import random
tanks=['Tank','Flank Guard','Machine Gun','Sniper','Twin','Assassin',
       'Auto 3','Destroyer','Gunner','Hunter','Overseer','Quad Tank',
       'Smasher','Trapper','Tri-Angle','Triple Shot','Twin Flank',
       'Annihilator','Auto 5','Auto Gunner','Auto Smasher','Auto Trapper',
       'Battleship','Booster','Factory','Fighter','Gunner Trapper',
       'Hybrid','Landmine','Manager','Mega Trapper','Necromancer',
       'Octo Tank','Overlord','Overtrapper','Penta Shot','Predator',
       'Ranger','Rocketeer','Skimmer','Spike','Sprayer','Spread Shot',
       'Stalker','Streamliner','Tri-Trapper','Triple Twin','Triplet']
sandbox=input('Include Sandbox-exclusive tanks (Y/N): ').upper()
if sandbox=='Y':
    tanks.extend(['Arena Closer','Gunner Dominator','Destroyer Dominator',
                  'Trapper Dominator','Mothership'])
while True:
    g=input('>>> ').strip().title()
    if g=='End': break
    elif g=='Save':
        try:
            with open('diep.io_builds.txt','a') as file:
                file.write(s[49:]+'\n\n')
            continue
        except NameError: continue
    if g in tanks: tank=g
    elif any(x.isalpha() for x in g): continue
    else: tank=random.choice(tanks)
    if '/' not in g:
        if tanks.index(tank) not in [12,20,28,40]: stats=list(range(8))*7
        elif tank!=tanks[20]: stats=list(range(4))*10
        else: stats=list(range(8))*10
        random.shuffle(stats)
        stats,build=stats[:33],[]
        for i in range(4): build.append(stats.count(i))
        if tanks.index(tank) not in [12,28,40]:
            for i in range(4,8): build.append(stats.count(i))
    else:
        try: build=list(map(int,g.split('/')))
        except ValueError: continue
        if len(build) not in [4,8] or max(build)>10 or min(build)<0 or sum(build)>33: continue
        if len(build)==4: tank=random.choice([tanks[12],tanks[28],tanks[40]])
        elif max(build)>7: tank=tanks[20]
    s='_'*48+'\n*{}*'.format(tank)
    s+='\n[1] Health Regen:\t {} {}'.format(build[0],'- '*build[0])
    s+='\n[2] Max Health:\t\t {} {}'.format(build[1],'- '*build[1])
    s+='\n[3] Body Damage:\t {} {}'.format(build[2],'- '*build[2])
    if tanks.index(tank) not in [12,28,40]:
        s+='\n[4] Bullet Speed:\t {} {}'.format(build[3],'- '*build[3])
        s+='\n[5] Bullet Penetration:\t {} {}'.format(build[4],'- '*build[4])
        s+='\n[6] Bullet Damage:\t {} {}'.format(build[5],'- '*build[5])
        s+='\n[7] Reload:\t\t {} {}'.format(build[6],'- '*build[6])
    s+='\n[8] Movement Speed:\t {} {}'.format(build[-1],'- '*build[-1])
    s+='\n'+'_'*48
    print(s)