Adiciona mais remoções

sapl/legacy/scripts/street_sweeper.py

@@ -32,9 +32,15 @@ mapa['comissao'] = ['composicao_comissao']
 mapa['sessao_legislativa'] = ['composicao_mesa']
 mapa['tipo_expediente'] = ['expediente_sessao_plenaria']
 
-tab_pk = {'materia_legislativa': 'cod_materia'}
-
-
+"""
+mapa['autor'] = ['tipo_autor', 'partido', 'comissao', 'parlamentar']
+mapa['parlamentar'] = ['autor', 'autoria', 'composicao_comissao',
+                       'composicao_mesa', 'dependente', 'filiacao',
+                       'mandato', 'mesa_sessao_plenaria', 'oradores',
+                       'oradores_expediente', 'ordem_dia_presenca',
+                       'registro_votacao_parlamentar', 'relatoria',
+                       'sessao_plenaria_presenca', 'unidade_tramitacao']
+"""
 
 def get_ids_excluidos(cursor, query):
     """
@@ -88,7 +94,13 @@ def remove_excluidas(cursor):
 
 
 def remove_proposicao_invalida(cursor):
-    return cursor.execute("DELETE FROM proposicao WHERE cod_mat_ou_doc is null")
+    return cursor.execute(
+        "DELETE FROM proposicao WHERE cod_mat_ou_doc is null")
+
+
+def remove_materia_assunto_invalida(cursor):
+    return cursor.execute(
+        "DELETE FROM materia_assunto WHERE cod_assunto = 0")
 
 
 def shotgun_remove(cursor):
@@ -100,7 +112,10 @@ def shotgun_remove(cursor):
 
 
 if __name__ == '__main__':
-    connection = pymysql.connect(host=HOST, user=USER, password=PASSWORD, db=DB)
+    connection = pymysql.connect(host=HOST,
+                                 user=USER,
+                                 password=PASSWORD,
+                                 db=DB)
     cursor = connection.cursor()
     # MATERIA LEGISLATIVA
     remove_tabelas(cursor, 'materia_legislativa', 'cod_materia')
@@ -112,9 +127,17 @@ if __name__ == '__main__':
     remove_tabelas(cursor, 'sessao_legislativa', 'cod_sessao_leg')
     # EXPEDIENTE SESSAO
     remove_tabelas(cursor, 'tipo_expediente', 'cod_expediente')
+    # AUTOR
+    remove_tabelas(cursor, 'autor', 'cod_autor')
+    # PARLAMENTAR
+    remove_tabelas(cursor, 'parlamentar', 'cod_parlamentar')
+
     # PROPOSICAO
     remove_proposicao_invalida(cursor)
 
+    # MATERIA_ASSUNTO
+    remove_materia_assunto_invalida(cursor)
+
     # shotgun_remove(cursor)
 
     cursor.close()