-
Notifications
You must be signed in to change notification settings - Fork 78
/
Copy pathimportando_produtos.py
43 lines (36 loc) · 1.05 KB
/
importando_produtos.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import csv
from projeto.produto.models import Produto
def csv_to_list(filename: str) -> list:
'''
Lê um csv e retorna um OrderedDict.
Créditos para Rafael Henrique
https://bit.ly/2FLDHsH
'''
with open(filename) as csv_file:
reader = csv.DictReader(csv_file, delimiter=',')
csv_data = [line for line in reader]
return csv_data
def save_data(data):
'''
Salva os dados no banco.
'''
aux = []
for item in data:
produto = item.get('produto')
ncm = str(item.get('ncm'))
importado = True if item.get('importado') == 'True' else False
preco = item.get('preco')
estoque = item.get('estoque')
estoque_minimo = item.get('estoque_minimo')
obj = Produto(
produto=produto,
ncm=ncm,
importado=importado,
preco=preco,
estoque=estoque,
estoque_minimo=estoque_minimo,
)
aux.append(obj)
Produto.objects.bulk_create(aux)
data = csv_to_list('fix/produtos.csv')
save_data(data)