palabra especial

"""Escribir una función que tome como argumento un fichero y una palabra especial y cree dos nuevos ficheros, uno con todas las líneas
del fichero original que contenga la palabra especial y el otro con las que no la contengan"""

def palabra_especial(fichero,palabra):
    fp1=open(fichero)
    fp2=open("palabra especial(ESPECIAL).txt","w")
    fp3=open("palabra especial(RESTO).txt","w")
    linea=fp1.readline()
    while linea:
        if palabra in linea:
            fp2.write(linea)
        else:
            fp3.write(linea)
        linea=fp1.readline()
    fp1.close()
    fp2.close()
    fp3.close()
palabra_especial("palabra especial.txt", "policia")