Calculando números primos - Python

Um clássico problema de programação é o calculo para descobrir números primos de forma rápida e precisa. Por isso, decidi fazer no meu tempo ocioso uma função em python que retorne se um número é ou não primo. Usei o teorema que diz que todo o número* não* primo é divisivel por outro menor ou igual a sua raiz quadrada e o resultado foi o seguinte:

     import math     def primo(a):         for i in xrange(2,int(math.sqrt(a)+1)):             if(a%i==0):                 print("%d nao e primo" % a)                 return 0        print("%d e primo" % a)         return 1