from pprint import pprint

from utils.functions.models import rows_to_dict_list


def nf_inform(cursor, nf, especiais=False):
    filtra_especial = "" if especiais else "AND f.NUMERO_CAIXA_ECF = 0"
    sql = f"""
        SELECT
          f.BASE_ICMS VALOR
        , f.QTDE_EMBALAGENS VOLUMES
        , f.DATA_AUTORIZACAO_NFE DATA
        , CAST( COALESCE( '0' || f.COD_STATUS, '0' ) AS INT )
          COD_STATUS
        , COALESCE( f.MSG_STATUS, ' ' ) MSG_STATUS
        , f.SITUACAO_NFISC SITUACAO
        , f.NATOP_NF_NAT_OPER NAT
        , f.NATOP_NF_EST_OPER UF
        , fe.DOCUMENTO NF_DEVOLUCAO
        , c.NOME_CLIENTE
          || ' (' || lpad(c.CGC_9, 8, '0')
          || '/' || lpad(c.CGC_4, 4, '0')
          || '-' || lpad(c.CGC_2, 2, '0')
          || ')' CLIENTE
        , f.NUMERO_CAIXA_ECF especial
        FROM FATU_050 f -- fatura
        JOIN PEDI_010 c -- cliente
          ON c.CGC_9 = f.CGC_9
         AND c.CGC_4 = f.CGC_4
        LEFT JOIN OBRF_010 fe -- nota fiscal de entrada/devolução
          ON fe.NOTA_DEV = f.NUM_NOTA_FISCAL
         AND fe.SITUACAO_ENTRADA <> 2 -- não cancelada
        WHERE f.NUM_NOTA_FISCAL = %s
          {filtra_especial} -- filtra_especial
    """
    cursor.execute(sql, [nf])
    return rows_to_dict_list(cursor)

@@ -55,33 +55,81 @@
  import 
-rows_to_dict_list
+dictlist%0Afrom utils.functions.queries import debug_cursor_execute
 %0A%0A%0Adef n
@@ -164,16 +164,27 @@
 is=False
+, empresa=1
 ):%0A    f
@@ -250,16 +250,73 @@
 CF = 0%22%0A
+    filtra_empresa = f%22AND f.CODIGO_EMPRESA = %7Bempresa%7D%22%0A
     sql 
@@ -365,16 +365,45 @@
 S VALOR%0A
+        , f.CODIGO_EMPRESA  %0A
         
@@ -1291,10 +1291,57 @@
 L = 
-%25s
+%7Bnf%7D%0A          %7Bfiltra_empresa%7D -- filtra_empresa
 %0A   
@@ -1400,23 +1400,29 @@
     
+debug_
 cursor
-.
+_
 execute(
 sql,
@@ -1421,17 +1421,19 @@
 ute(
-sql, %5Bnf%5D
+cursor, sql
 )%0A  
@@ -1445,21 +1445,12 @@
 urn 
-rows_to_
 dict
-_
 list
@@ -1456,10 +1456,31 @@
 t(cursor
+, name_case=str.upper
 )%0A

query nf_inform filtra de empresa