PostgreSQL
 sql >> Base de Dados >  >> RDS >> PostgreSQL

Postgresql EM CONFLITO em sqlalchemy

Usando Postgres 9.6.1, sqlachemy 1.1.4 e psycopg2 2.6.2:


  1. Converta sua estrutura de dados em um dicionário. De Pandas é
    import pandas
    from sqlalchemy import MetaData
    from sqlalchemy.dialects.postgresql import insert
    import psycopg2
    
    # The dictionary should include all the values including index values
    insrt_vals = df.to_dict(orient='records')
    

  2. Conecte-se ao banco de dados por meio de sqlalchemy . Em vez disso, tente o driver psycog2 abaixo e a função COPY nativa, que ignora toda a indexação do postgres.
    csv_data = os.path.realpath('test.csv')
    con = psycopg2.connect(database = 'db01', user = 'postgres')
    cur = con.cursor()
    cur.execute("\copy stamm_data from '%s' DELIMITER ';' csv header" % csv_data)
    con.commit()
    

  3. Executar
    results = engine.execute(do_nothing_stmt)
    # Get number of rows inserted
    rowcount = results.rowcount
    

Aviso:


Este método não funciona com NaT está fora da caixa.

Tudo junto

tst_df = pd.DataFrame({'colA':['a','b','c','a','z', 'q'],
              'colB': pd.date_range(end=datetime.datetime.now() , periods=6),
              'colC' : ['a1','b2','c3','a4','z5', 'q6']})


insrt_vals = tst_df.to_dict(orient='records')
engine =      sqlalchemy.create_engine("postgresql://user:[email protected]/postgres")
connect = engine.connect()
meta = MetaData(bind=engine)
meta.reflect(bind=engine)
table = meta.tables['tstbl']
insrt_stmnt = insert(table).values(insrt_vals)

do_nothing_stmt  = insrt_stmnt.on_conflict_do_nothing(index_elements=['colA','colB'])
results = engine.execute(do_nothing_stmt)

Em vez das etapas 2 e 3 , usando psycog2 driver com o comando copy no postgres é mais rápido para arquivos maiores (aproximando-se de um gig) porque desativa toda a indexação da tabela.
csv_data = os.path.realpath('test.csv')