Carregar um CSV numa tabela

Tradução automática

Esta página foi traduzida automaticamente do inglês e ainda não foi revisada por um falante nativo; correções são bem-vindas no GitHub. Se algo não bater, vale o original em inglês. O código e os resultados são os mesmos do original, importados dos arquivos testados.

Tarefa: carregar uma lista de produtos no banco, tudo ou nada.

# Recipe: load the rows of a CSV file into a table, all or nothing.
paths:
  components: ./components
  migrations: ./migrations

datasources:
  db:
    driver: sqlite
    database: ./data/shop.db

CREATE TABLE products (
    sku TEXT PRIMARY KEY,
    name TEXT NOT NULL,
    price INTEGER NOT NULL CHECK (price > 0)
);

sku,name,price
MUG-1,Mug,30
TEE-1,T-shirt,80
STK-1,Sticker,5

A ação lê o arquivo com q:data e depois insere cada linha dentro de uma q:transaction; as consultas dentro dela, também as do loop, usam a fonte de dados dela.

<q:component name="Import">
  <q:action name="load" method="POST">
    <q:data name="rows" source="import/products.csv" type="csv">
      <q:column name="price" type="integer" />
    </q:data>

    <!-- One transaction for the whole file: if one row is refused (a sku
         already there, a price that is not positive), none stays. -->
    <q:transaction datasource="db">
      <q:loop items="{rows}" var="row">
        <q:query name="inserted">
          INSERT INTO products (sku, name, price) VALUES (:sku, :name, :price)
          <q:param name="sku" value="{row.sku}" type="string" />
          <q:param name="name" value="{row.name}" type="string" />
          <q:param name="price" value="{row.price}" type="integer" />
        </q:query>
      </q:loop>
    </q:transaction>

    <q:redirect url="/" flash="Loaded {len(rows)} products." />
  </q:action>

  <q:query name="products" datasource="db">SELECT sku, name, price FROM products ORDER BY sku</q:query>

  <ui:window title="Products">
    <q:if condition="flash">
      <ui:alert variant="success">{flash}</ui:alert>
    </q:if>
    <ui:text>{products_result.recordCount} products</ui:text>
    <ui:table source="{products}" />
    <ui:form on-submit="load" submit="Load import/products.csv" />
  </ui:window>
</q:component>

O segundo teste carrega o arquivo quando um dos produtos já está lá: a segunda linha falha, e a primeira, já inserida, é desfeita.

<q:test name="the file's rows land in the table" page="/">
  <test:submit action="load" />
  <test:expect redirect="/" flash="Loaded 3 products." />
  <test:expect table="products" count="3" />
  <test:expect table="products" count="1" where="sku = 'TEE-1' AND price = 80" />
  <test:expect text="3 products" />
</q:test>

<q:test name="a file that fails halfway leaves nothing behind" page="/">
  <!-- TEE-1 is already there: the second row fails, so the first is undone too. -->
  <test:given table="products" sku="TEE-1" name="T-shirt" price="80" />
  <test:submit action="load" />
  <test:expect status="500" />
  <test:expect table="products" count="1" />
  <test:expect table="products" count="0" where="sku = 'MUG-1'" />
</q:test>

tests/load.test.q
  PASS  the file's rows land in the table
  PASS  a file that fails halfway leaves nothing behind
2 passed, 0 failed

Veja DATA-1 e DB-4.