consulta no muestra lo esperado

[pithon]

necesito recuperar el total de los campos cantidad de los distintos codigo where el precio no sea null, esta seria la tabla:

Código MySQL:

+---------------------+--------+--------+--------+-----------------+
| fecha               | codigo | cantidad| stock| precio       |
+---------------------+--------+--------+--------+-----------------+
| 2016-02-11 11:00:00 |    a222| 4| 3   | 20       |
| 2016-02-11 11:00:00 |    d234 | 2   | 1 | 10 |
| 2016-03-01 10:00:00 |    d234| 2| 12   | 5       |
+---------------------+--------+--------+--------+-----------------+
 
 
 
SELECT 
    DISTINCT codigo AS codigos, 
    SUM(IF(precio is not null,cantidad,0))AS total_cantidad
FROM tabla;
 

me tendria que mostrar algo asi:

Código MySQL:

+---------------------+-----------------+
| codigo               |total    |
+---------------------+-----------------+
| a222                 | 4       |
| d234                 | 4|
+---------------------+-----------------+
 

pero solo me marca un codigo, quizas tube algun error de sintaxis ya que esta hecho a mano.
saludos.