Untitled

mail@pastecode.io avatar
unknown
plain_text
2 years ago
629 B
6
Indexable
def verifyExistence(query,bucket_name):    
    statement_type, table_name, columns, filename = parseSQL(query)
    file_name = filename if filename is not None \
    else table_name + ".parquet"
    # Verify if the file exists in the duckdb else create a table in duckdb
    con = getConnection(threading.get_ident())
    try:
         result = con.execute(query)
         con.close()
    except duckdb.CatalogException :
        con=getWriteConn(threading.get_ident())
        loadDataToDb(con, file_name=file_name,bucket_name=bucket_name)
        result = con.execute(query)
        con.close()
    return result