import random

tablica = [0,1,2,
           3,4,5,
           6,7,8]

def plansza ():
  print (tablica[0], "|", tablica[1], "|", tablica[2])
  print ("----------")
  print (tablica[3], "|", tablica[4], "|", tablica[5])
  print ("----------")
  print (tablica[6], "|", tablica[7], "|", tablica[8])
  print("")


def spr_linijke(znak, pole1, pole2, pole3):
  if tablica[pole1] == znak and tablica[pole2] == znak and tablica[pole3] == znak:
    return True

def spr_wszystkie(znak):
  if spr_linijke(znak, 0, 3, 6):
    return True
  if spr_linijke(znak, 1, 4, 7):
    return True
  if spr_linijke(znak, 2, 5, 8):
    return True

  if spr_linijke(znak, 6, 7, 8):
    return True
  if spr_linijke(znak, 3, 4, 5):
    return True
  if spr_linijke(znak, 0, 1, 2):
    return True

  if spr_linijke(znak, 2, 4, 6):
    return True
  if spr_linijke(znak, 0, 4, 8):
    return True

def wolne_pole(nr_pola):
  if tablica[nr_pola] != 'o' and tablica[nr_pola] != 'x':
    return True
  return False

def zwroc_wolny_rog():
  if wolne_pole(0):
    return 0
  elif wolne_pole(2):
    return 2
  elif wolne_pole(6):
    return 6
  elif wolne_pole(8):
    return 8
  else:
    return None

def oba_kolka(pole1,pole2):
  if tablica[pole1] == tablica[pole2] and tablica[pole1] == 'o':
    return True
  return False

def zwroc_wygrana_ukos():
  for i in [0,2,6,8]:
    if oba_kolka(i,4) and wolne_pole(8-i):
      return 8-i

  if (oba_kolka(0,8) or oba_kolka(6,2)) and wolne_pole(4):
    return 4
  return None



wygrana = False

while wygrana != True:

  użytkownik = input("Wybierz pole: ")
  użytkownik = int(użytkownik)

  if tablica[użytkownik] != 'x' and tablica[użytkownik] != 'o':
    tablica[użytkownik] = 'x'

    if spr_wszystkie('x') == True:
      print ("~~X wygrywa!~~")
      wygrana = True
      break

    while True:
      random.seed() #Play a random generator

      ukos = zwroc_wygrana_ukos()

      if ukos != None:
        przeciwnik = ukos
      else:
        rog = zwroc_wolny_rog()
        if wolne_pole(4):
          przeciwnik = 4
        elif rog != None:
          przeciwnik = rog
        else:
          przeciwnik = random.randint (0,8)

      if tablica[przeciwnik] != 'o' and tablica[przeciwnik] != 'x':
        tablica[przeciwnik] = 'o'

        if spr_wszystkie('o') == True:
          print ("~~O wygrywa!~~")
          wygrana = True
          break

        break

  else:
    print ("To pole jest już zajęte!")


  plansza()