Python-сообщество

A=[[0 for i in range(8)] for i in range(8)]  "#рисует шахматную доску
def print_A(A):
    for i in A:
        print(i)
#вводит координаты короля
jk=int(input('координаты короля по вертикали: '))
print('= ',jk)
ik=int(input('координаты короля по горизонтали: '))
print('= ',ik)
A[jk][ik]='K'

#вводит координаты ладьи(их три)
i=0
while i<3:
    y=int(input('координаты ладьи по вертикали: '))
    print('= ',y)
    x=int(input('координаты ладьи по горизонтали: '))
    print('= ',x)
    i=i+1  
    A[y][x]='L'
    L=A[y][x]

import sys, pprint

def input_data():
    data = [[0 for i in xrange(8)] for j in xrange(8)]
    cl = [int(i) for i in sys.argv[1:]]
    king, data[cl[0]][cl[1]] = (cl[0], cl[1]), 1
    for i in xrange(2,7,2): data[cl[i]][cl[i+1]] = 2
    pprint.pprint(data)
    return (king, data)

def msg(i, j):
    print "rook with coords: (%i, %i)" % (i, j)

def way_check(way, data):
    for (i, j) in way:
        if data[i][j] == 2:
            msg(i, j)
            return True
    return False

def strike(king, data):
    check = False
    # left to the king
    way = [(i, king[1]) for i in range(0,king[0])]
    way.reverse()
    check = way_check(way, data)
    # right to the king
    way = [(i, king[1]) for i in range(king[0] + 1, 8)]
    check = way_check(way, data)
    # up to the king
    way = [(king[0], i) for i in range(0, king[1])]
    way.reverse()
    check = way_check(way, data)
    # down to the king
    way = [(king[0], i) for i in range(king[1] + 1, 8)]
    check = way_check(way, data)
    if check:
        print 'king is safe'

def main():
    (king, data) = input_data()
    strike(king, data)

if __name__ == '__main__':
    main()