massive product load implementation, not tested
This commit is contained in:
@@ -1,8 +1,15 @@
|
||||
import logging
|
||||
import csv
|
||||
|
||||
from django.shortcuts import render
|
||||
from django.conf import settings
|
||||
|
||||
# Create your views here.
|
||||
from rest_framework import viewsets
|
||||
from rest_framework.permissions import IsAuthenticatedOrReadOnly
|
||||
from rest_framework.permissions import IsAuthenticatedOrReadOnly, IsAdminUser
|
||||
from rest_framework.decorators import api_view, permission_classes
|
||||
|
||||
import requests
|
||||
|
||||
from products.models import Product
|
||||
from products.serializers import ProductSerializer
|
||||
@@ -10,7 +17,78 @@ from products.serializers import ProductSerializer
|
||||
from back_latienda.permissions import IsCreator
|
||||
|
||||
|
||||
logging.basicConfig(
|
||||
filename='logs/product-load.log',
|
||||
filemode='w',
|
||||
format='%(levelname)s:%(message)s',
|
||||
level=logging.INFO,
|
||||
)
|
||||
|
||||
|
||||
class ProductViewSet(viewsets.ModelViewSet):
|
||||
queryset = Product.objects.all()
|
||||
serializer_class = ProductSerializer
|
||||
permission_classes = [IsAuthenticatedOrReadOnly, IsCreator]
|
||||
|
||||
|
||||
@api_view(['POST',])
|
||||
@permission_classes([IsAdminUser,])
|
||||
def load_coop_products(request):
|
||||
"""Read CSV file being received
|
||||
Parse it to create products for related Company
|
||||
"""
|
||||
try:
|
||||
csv_file = request.FILES['csv_file']
|
||||
if csv_file.name.endswith('.csv') is not True:
|
||||
logging.error(f"File {csv_file.name} is not a CSV file")
|
||||
return Response({"errors":{"details": "File is not CSV type"}})
|
||||
|
||||
logging.info(f"Reading contents of {csv_file.name}")
|
||||
decoded_file = csv_file.read().decode('utf-8').splitlines()
|
||||
csv_reader = csv.DictReader(decoded_file, delimiter=',')
|
||||
counter = 0
|
||||
for row in csv_reader:
|
||||
if '' in (row['nombre-producto'], row['descripcion'], row['precio']):
|
||||
logging.error(f"Required data missing: {row}")
|
||||
continue
|
||||
try:
|
||||
# download image references in csv
|
||||
image_url = row['image'].strip()
|
||||
response = requests.get(image_url, stream=True)
|
||||
if response.status_code == 200:
|
||||
path = f"{setting.BASE_DIR}media/{row['nombre-producto'].strip()}.{image.url.split('/')[-1]}"
|
||||
logging.info(f"Saving product image to: {path}")
|
||||
new_image = open(path, 'wb')
|
||||
for chunk in response:
|
||||
new_image.write(chunk)
|
||||
new_image.close()
|
||||
else:
|
||||
logging.warninig(f"Image URL did not work: {image_url}")
|
||||
new_image = None
|
||||
# assemble instance data
|
||||
product_data = {
|
||||
'id': None if row['id'].strip()=='' else row['id'].strip(),
|
||||
'name': row['nombre-coop'].strip(),
|
||||
'short_name': row['nombre-producto'].strip(),
|
||||
'description': row['descripcion'].strip(),
|
||||
'image': new_image,
|
||||
'url': row['url'].strip(),
|
||||
'precio': row['precio'].strip(),
|
||||
'shipping_cost': row['gastos-envio'].strip(),
|
||||
'shipping_terms': row['cond-envio'].strip(),
|
||||
'discount': row['descuento'].strip(),
|
||||
'stock': row['stock'].strip(),
|
||||
'tags': row['tags'].strip(),
|
||||
'category': row['categoria'].strip(),
|
||||
'identifiers': row['identificadores'].strip(),
|
||||
}
|
||||
Product.objects.create(**product_data)
|
||||
logging.info(f"Created Product: {product_data}")
|
||||
counter += 1
|
||||
except Exception as e:
|
||||
logging.error(f"Could not parse {row}")
|
||||
return Response()
|
||||
except Exception as e:
|
||||
return Response({"errors": {"details": str(type(e))}})
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user