Suoran piirto

from turtle import *

def akseli(koko,akselinNimi):
a.penup()
a.goto(0,0)
a.pendown()
a.bk(koko)
a.fd(2*koko)
a.stamp()
a.penup()
a.lt(90)
a.fd(10)
a.write(akselinNimi)
a.rt(90)

def koordinaatisto():
akseli(koko,'x')
a.lt(90)
akseli(koko,'y')

def pilko(suora):
if suora[suora.index('=')+1:suora.index('x')] == '':
k = 1
elif suora[suora.index('=')+1:suora.index('x')] == '-':
k = -1
else:
k = float(suora[suora.index('=')+1:suora.index('x')])
b = float(suora[suora.index('x')+1:])
return [k,b]

def piirra(suora):
osat = pilko(suora)
a.penup()
a.goto(-koko,osat[0]*-koko + osat[1])
a.pendown()

for i in range(-koko,koko):
a.goto(i, osat[0]*i + osat[1])

def yhtalo(suora):
a.penup()
a.goto(koko/2,-koko/2)
a.write(suora,20)
a.hideturtle()

a = Turtle()
a.speed(0)

koko = 250

suora = input(u'Anna piirrettävän suoran yhtälö')

koordinaatisto()
piirra(suora)
yhtalo(suora)