numeros primos

[sabeque]

hola a todos. Soy principiante en python y quiero hacer un programita que busque en una lista numeros primos y no primos y los guarde en archivos de texto diferentes.
Hasta ahora escribi esto:

Código Python:

# -*- coding: utf-8 -*-
 
class buscadordeconvinaciones:
    import itertools
    # ahora la idea sera generar convinaciones de una lista
    noprimos = 0
    primos = 0
    # parametro general en el que se debera buscar
    lgral = [0,1,2,3,4]
    archivo=open('/home/datosprimos2.txt','w')
    archivo2=open('/home/datosnoprimos2.txt','w')
    archivo.close()
    archivo2.close()
    for g in itertools.combinations(lgral, 4):
        for i in g:
            if i == 0:
                noprimos = noprimos + 1
            elif i == 1:
                noprimos += 1
            elif i == 2:
                primos += 1
            elif i == 3:
                primos += 1
            elif i == 4:
                noprimos += 1
        if primos < 2:
            archivo=open('/home/datosprimos2.txt','a')
            archivo.write ("\n")
            archivo.write (str(g))
            archivo.close()
        
        else:
            archivo2=open('/home/datosnoprimos2.txt','a')
            archivo2.write ("\n")
            archivo2.write (str(g))
            archivo2.close()
        print g
 

pero no esta funcionando bien. Alguien puede ayudarme? es mas, la lista es 0 a 10. puse corta para dar la idea nomas. Ante todo gracias