Merge branch 'master' of bitbucket.org:enreda/back-latienda

This commit is contained in:
pablogg
2021-02-26 14:28:41 +01:00
12 changed files with 188 additions and 62 deletions

View File

@@ -51,3 +51,7 @@ SIMPLE_JWT = {
} }
EMAIL_BACKEND = 'django.core.mail.backends.console.EmailBackend' EMAIL_BACKEND = 'django.core.mail.backends.console.EmailBackend'
# disable web interface for REST backend
# REST_FRAMEWORK['DEFAULT_RENDERER_CLASSES'] = ('rest_framework.renderers.JSONRenderer',)

View File

@@ -80,3 +80,8 @@ SIMPLE_JWT = {
'SLIDING_TOKEN_REFRESH_LIFETIME': timedelta(days=1), 'SLIDING_TOKEN_REFRESH_LIFETIME': timedelta(days=1),
} }
# disable web interface for REST backend
REST_FRAMEWORK['DEFAULT_RENDERER_CLASSES'] = ('rest_framework.renderers.JSONRenderer',)

View File

@@ -46,6 +46,7 @@ class Company(models.Model):
is_validated = models.BooleanField('Validado', default=False, null=True, blank=True) is_validated = models.BooleanField('Validado', default=False, null=True, blank=True)
is_active = models.BooleanField('Activado', default=False, null=True, blank=True) is_active = models.BooleanField('Activado', default=False, null=True, blank=True)
credentials = JSONField(null=True) credentials = JSONField(null=True)
shipping_terms = models.TextField('Condiciones de envío', null=True, blank=True)
# internal # internal
created = models.DateTimeField('date of creation', auto_now_add=True) created = models.DateTimeField('date of creation', auto_now_add=True)

View File

@@ -452,8 +452,7 @@ class LoadCoopManagerTestCase(APITestCase):
# send in request # send in request
response = self.client.post(self.endpoint, files) response = self.client.post(self.endpoint, files)
# check response
# check re sponse
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
# check for object creation # check for object creation
self.assertEquals(company_count + 5, self.company_model.objects.count()) self.assertEquals(company_count + 5, self.company_model.objects.count())

View File

@@ -1,4 +1,5 @@
import logging import logging
from io import BytesIO
from django.contrib.auth import get_user_model from django.contrib.auth import get_user_model
from django.contrib.sites.shortcuts import get_current_site from django.contrib.sites.shortcuts import get_current_site
@@ -8,11 +9,17 @@ from django.template.loader import render_to_string
from django.core.mail import EmailMessage from django.core.mail import EmailMessage
from django.contrib.auth.tokens import PasswordResetTokenGenerator from django.contrib.auth.tokens import PasswordResetTokenGenerator
from django.conf import settings from django.conf import settings
from django.core.validators import validate_email, EmailValidator, URLValidator, ValidationError
from rest_framework_simplejwt.tokens import RefreshToken from rest_framework_simplejwt.tokens import RefreshToken
import requests
from PIL import Image
from django.core.files import File
from tagulous.models import TagModel from tagulous.models import TagModel
from companies.models import Company
User = get_user_model() User = get_user_model()
@@ -98,3 +105,97 @@ def reformat_google_taxonomy(file_name):
line = line.replace(' > ', '/') line = line.replace(' > ', '/')
destination_file.write(line) destination_file.write(line)
def coop_loader(csv_reader, request=None):
"""
Parse csv data and extract:
- coop data
- manager user data
Return counts
"""
coop_counter = 0
user_counter = 0
for row in csv_reader:
# trim strings
for key in row:
if row[key]: row[key] = row[key].strip().lower()
# import ipdb; ipdb.set_trace()
if '' in (row['cif'], row['nombre-coop'], row['email']):
logging.error(f"Required data missing: {row}")
continue
# validate email
try:
validate_email(row['email'])
except ValidationError:
logging.warning(f"Invalid email value '{row['email']}', skipped")
continue
# validate URLs
if row['url'].startswith('http') is not True:
row['url'] = 'http://' + row['url']
if row['logo-url'].startswith('http') is not True:
row['logo-url'] = 'http://' + row['logo-url']
validator = URLValidator()
try:
validator(row['url'])
except ValidationError:
logging.warning(f"Invalid url value '{row['url']}'")
row['url'] = None
try:
validator(row['logo-url'])
except ValidationError:
logging.warning(f"Invalid logo URL value '{row['logo-url']}'")
row['logo-url'] = None
# validate boolean
try:
shop = bool(row['es-tienda'])
except:
logging.warning(f"Invalid valur for es-tiends: {row['es-tienda']}")
shop = None
# create instances
try:
coop_data = {
'cif': row['cif'],
'company_name': row['nombre-coop'],
'short_name': row['nombre-corto'],
'shop': shop,
'shop_link': row['url'],
'phone': row['telefono'],
'address': row['direccion'],
}
coop = Company.objects.create(**coop_data)
# image logo data
if row['logo-url'] is not None:
try:
# get image
headers={"User-Agent" : "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/51.0.2704.103 Safari/537.36"}
response = requests.get(row['logo-url'], stream=True, headers=headers)
assert(response.status_code==200)
response.raw.decode_content = True
image = Image.open(response.raw)
# save using File object
img_io = BytesIO()
image.save(img_io, format=image.format)
coop.logo.save(f"{coop.company_name}.{image.format.lower()}", File(img_io), save=False)
coop.save()
except AssertionError as e:
logging.error(f"Source image [{row['logo-url']}] not reachable: {response.status_code}")
except Exception as e:
logging.error(f"Could not add image to COOP {coop.company_name} from [{row['logo-url']}]: {str(e)}")
#
logging.info(f"Created Coop: {coop_data}")
coop_counter += 1
coop_user = User.objects.create_user(email=row['email'], company=coop, role='COOP_MANAGER', is_active=False)
# send confirmation email
if request is not None:
send_verification_email(request, coop_user)
logging.info(f"Created User: {coop_user}")
user_counter += 1
except Exception as e:
import ipdb; ipdb.set_trace()
logging.error(f"Could not parse {row}")
return coop_counter, user_counter

View File

@@ -183,35 +183,11 @@ def load_coop_managers(request):
logging.info(f"Reading contents of {csv_file.name}") logging.info(f"Reading contents of {csv_file.name}")
decoded_file = csv_file.read().decode('utf-8').splitlines() decoded_file = csv_file.read().decode('utf-8').splitlines()
csv_reader = csv.DictReader(decoded_file, delimiter=',') csv_reader = csv.DictReader(decoded_file, delimiter=',')
coop_counter = 0 coop_count, user_count = utils.coop_loader(csv_reader, request)
user_counter = 0
for row in csv_reader:
if '' in (row['cif'], row['nombre-coop'], row['email']):
logging.error(f"Required data missing: {row}")
continue
try:
coop_data = {
'cif': row['cif'].strip(),
'company_name': row['nombre-coop'].strip(),
'short_name': row['nombre-corto'].strip(),
'shop': bool(row['es-tienda'].strip()),
'shop_link': row['url'].strip(),
}
coop = Company.objects.create(**coop_data)
logging.info(f"Created Coop: {coop_data}")
coop_counter += 1
coop_user = User.objects.create_user(email=row['email'], company=coop, role='COOP_MANAGER', is_active=False) return Response({'details': f"Created {coop_count} Companies, {user_count} Managing Users"})
# send confirmation email
utils.send_verification_email(request, coop_user)
logging.info(f"Created User: {coop_user}")
user_counter += 1
except Exception as e:
logging.error(f"Could not parse {row}")
return Response()
except Exception as e: except Exception as e:
return Response({"errors": {"details": str(type(e))}}) return Response({"errors": {"details": f'{type(e)}: {e}'}})
@api_view(['GET',]) @api_view(['GET',])

View File

@@ -1,9 +1,9 @@
email,cif,nombre-coop,nombre-corto,url,es-tienda email,cif,nombre-coop,nombre-corto,url,es-tienda,logo-url,telefono,provincia,localidad,cp,direccion
, 1223432214L, FEWQ4FEWQ COOP, fc, tienda1.com, True , 1223432214L, FEWQ4FEWQ COOP, fc, tienda1.com, True,,,,,,
dsfds@mail.com,, FEW2QFEWQ COOP, fc, tienda2.com, True dsfds@mail.com,, FEW2QFEWQ COOP, fc, tienda2.com, True,,,,,,
ghjhg@mail.com, 122343214L,, fc, tienda3.com, True ghjhg@mail.com, 122343214L,, fc, tienda3.com, True,,,,,,
xcv@mail.com, 12343214L, FEWQ2FEWQ COOP,, tienda4.com, True xcv@mail.com, 12343214L, FEWQ2FEWQ COOP,, tienda4.com, True,,,,,,
cvc@mail.com, 1879783214L, 2FEWQFEWQ COOP, fc,, True cvc@mail.com, 1879783214L, 2FEWQFEWQ COOP, fc,, True,,,,,,
bvbc@mail.com, 5653214L, FEW2QFEWQ COOP, fc, tienda6.com, bvbc@mail.com, 5653214L, FEW2QFEWQ COOP, fc, tienda6.com,,,,,,
kjk@mail.com, 54326543H, FE2WQF2EWQ COOP, fc, tienda7.com, True kjk@mail.com, 54326543H, FE2WQF2EWQ COOP, fc, tienda7.com, True,,,,,,
yuyu@mail.com, 12343214L, F2EWQFEWQ COOP, fc, tienda8.com, True yuyu@mail.com, 12343214L, F2EWQFEWQ COOP, fc, tienda8.com, True,,,,,,
1 email email,cif,nombre-coop,nombre-corto,url,es-tienda,logo-url,telefono,provincia,localidad,cp,direccion cif nombre-coop nombre-corto url es-tienda
2 , 1223432214L, FEWQ4FEWQ COOP, fc, tienda1.com, True,,,,,, 1223432214L FEWQ4FEWQ COOP fc tienda1.com True
3 dsfds@mail.com dsfds@mail.com,, FEW2QFEWQ COOP, fc, tienda2.com, True,,,,,, FEW2QFEWQ COOP fc tienda2.com True
4 ghjhg@mail.com ghjhg@mail.com, 122343214L,, fc, tienda3.com, True,,,,,, 122343214L fc tienda3.com True
5 xcv@mail.com xcv@mail.com, 12343214L, FEWQ2FEWQ COOP,, tienda4.com, True,,,,,, 12343214L FEWQ2FEWQ COOP tienda4.com True
6 cvc@mail.com cvc@mail.com, 1879783214L, 2FEWQFEWQ COOP, fc,, True,,,,,, 1879783214L 2FEWQFEWQ COOP fc True
7 bvbc@mail.com bvbc@mail.com, 5653214L, FEW2QFEWQ COOP, fc, tienda6.com,,,,,, 5653214L FEW2QFEWQ COOP fc tienda6.com
8 kjk@mail.com kjk@mail.com, 54326543H, FE2WQF2EWQ COOP, fc, tienda7.com, True,,,,,, 54326543H FE2WQF2EWQ COOP fc tienda7.com True
9 yuyu@mail.com yuyu@mail.com, 12343214L, F2EWQFEWQ COOP, fc, tienda8.com, True,,,,,, 12343214L F2EWQFEWQ COOP fc tienda8.com True

View File

@@ -11,7 +11,7 @@ class ProductSerializer(TaggitSerializer, serializers.ModelSerializer):
tags = TagListSerializerField(required=False) tags = TagListSerializerField(required=False)
category = SingleTagSerializerField(required=False) # main tag category category = SingleTagSerializerField(required=False) # main tag category
attributes = TagListSerializerField(required=False) attributes = TagListSerializerField(required=False)
company = CompanySerializer(read_only=True)
class Meta: class Meta:
model = Product model = Product
fields = '__all__' fields = '__all__'

View File

@@ -311,6 +311,8 @@ class ProductViewSetTest(APITestCase):
# Assert instance has been modified # Assert instance has been modified
for key in data: for key in data:
if key == 'company':
continue
self.assertEqual(data[key], response.data[key]) self.assertEqual(data[key], response.data[key])
def test_auth_user_cannot_modify_other_users_instance(self): def test_auth_user_cannot_modify_other_users_instance(self):
@@ -514,6 +516,12 @@ class ProductSearchTest(TestCase):
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
# load response data # load response data
payload = response.json() payload = response.json()
# check for expected fields in payload
self.assertIsNotNone(payload.get('filters'))
self.assertIsNotNone(payload.get('count'))
self.assertIsNotNone(payload.get('products'))
self.assertIsNotNone(payload.get('prices'))
# check for object creation # check for object creation
self.assertEquals(len(payload['products']), len(expected_instances)) self.assertEquals(len(payload['products']), len(expected_instances))
# check results ordered by rank # check results ordered by rank
@@ -524,6 +532,8 @@ class ProductSearchTest(TestCase):
# check for filters # check for filters
self.assertNotEquals([], payload['filters']['tags']['singles']) self.assertNotEquals([], payload['filters']['tags']['singles'])
self.assertTrue(len(payload['filters']['tags']) >= 2 ) self.assertTrue(len(payload['filters']['tags']) >= 2 )
# check prices
self.assertTrue(payload['prices']['min'] <= payload['prices']['max'])
def test_anon_user_can_paginate_search(self): def test_anon_user_can_paginate_search(self):
expected_instances = [ expected_instances = [
@@ -541,7 +551,7 @@ class ProductSearchTest(TestCase):
q = quote("zapatos rojos") q = quote("zapatos rojos")
limit = 2 limit = 2
url = f"{self.endpoint}?q={q}&limit=2" url = f"{self.endpoint}?q={q}&limit={limit}"
# send in request # send in request
response = self.client.get(url) response = self.client.get(url)

View File

@@ -2,6 +2,7 @@ import logging
from django.db.models import Q from django.db.models import Q
from django.contrib.postgres.search import SearchQuery, SearchRank, SearchVector, TrigramSimilarity from django.contrib.postgres.search import SearchQuery, SearchRank, SearchVector, TrigramSimilarity
from django.db.models import Max, Min
from products.models import Product from products.models import Product
@@ -189,7 +190,12 @@ def find_related_products_v6(keyword, shipping_cost=None, discount=None, categor
if price_max is not None: if price_max is not None:
products_qs = products_qs.filter(price__lt=price_max) products_qs = products_qs.filter(price__lt=price_max)
return set(products_qs) # get min_price and max_price
min_price = products_qs.aggregate(Min('price'))
max_price = products_qs.aggregate(Max('price'))
return set(products_qs), min_price, max_price
def find_related_products_v4(keyword): def find_related_products_v4(keyword):

View File

@@ -1,11 +1,7 @@
import logging import logging
import csv import csv
import datetime import datetime
import operator
from functools import reduce
from django.shortcuts import render
from django.conf import settings
from django.db.models import Q from django.db.models import Q
from django.core import serializers from django.core import serializers
@@ -157,6 +153,16 @@ def product_search(request):
- category: string - category: string
- tags: string - tags: string
- order: string (newest/oldest) - order: string (newest/oldest)
- price_min: int
- price_max: int
In the response:
- filters
- count
- products
- price_min
- price_max
""" """
# capture query params # capture query params
q = request.GET.get('q', None) q = request.GET.get('q', None)
@@ -195,11 +201,21 @@ def product_search(request):
try: try:
# we collect our results here # we collect our results here
result_set = set() result_set = set()
# values for response
prices = {
'min': None,
'max': None,
}
# split query string into single words # split query string into single words
chunks = q.split(' ') chunks = q.split(' ')
for chunk in chunks: for chunk in chunks:
product_set = find_related_products_v6(chunk, shipping_cost, discount, category, tags, price_min, price_max) product_set, min_price, max_price = find_related_products_v6(chunk, shipping_cost, discount, category, tags, price_min, price_max)
# update price values
if prices['min'] is None or min_price['price__min'] < prices['min']:
prices['min'] = min_price['price__min']
if prices['max'] is None or max_price['price__max'] > prices['max']:
prices['max'] = max_price['price__max']
# add to result set # add to result set
result_set.update(product_set) result_set.update(product_set)
# TODO: add search for entire phrase ??? # TODO: add search for entire phrase ???
@@ -218,6 +234,7 @@ def product_search(request):
# order results by RANK # order results by RANK
ordered_products = sorted(result_list, key= lambda rank:rank.rank, reverse=True) ordered_products = sorted(result_list, key= lambda rank:rank.rank, reverse=True)
# extract max and min price values
serializer = SearchResultSerializer(ordered_products, many=True) serializer = SearchResultSerializer(ordered_products, many=True)
product_results = [dict(i) for i in serializer.data] product_results = [dict(i) for i in serializer.data]
total_results = len(product_results) total_results = len(product_results)
@@ -231,6 +248,6 @@ def product_search(request):
limit = int(limit) limit = int(limit)
product_results = product_results[:limit] product_results = product_results[:limit]
return Response(data={"filters": filters, "count": total_results, "products": product_results}) return Response(data={"filters": filters, "count": total_results, "products": product_results, 'prices': prices})
except Exception as e: except Exception as e:
return Response({"errors": {"details": str(e)}}, status=status.HTTP_500_INTERNAL_SERVER_ERROR) return Response({"errors": {"details": str(e)}}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)

View File

@@ -59,7 +59,7 @@ def create_imported_product(info, company, history, user):
'name': info.get('name'), 'name': info.get('name'),
'description': BeautifulSoup(info.get('description', ''), "lxml").text, 'description': BeautifulSoup(info.get('description', ''), "lxml").text,
'sku': info.get('sku'), 'sku': info.get('sku'),
'price': info.get('price'), 'price': None if info.get('price') == '' else info.get('price'),
} }
# alternative method # alternative method
@@ -73,20 +73,27 @@ def create_imported_product(info, company, history, user):
except Exception as e: except Exception as e:
logging.error(f"Could not create product instance: {str(e)}") logging.error(f"Could not create product instance: {str(e)}")
return None return None
try:
# get image if len(info['images']) > 0:
headers={"User-Agent" : "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/51.0.2704.103 Safari/537.36"} try:
image_url = info['images'][0]['src'] # get image
response = requests.get(image_url, stream=True, headers=headers) headers={"User-Agent" : "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/51.0.2704.103 Safari/537.36"}
response.raw.decode_content = True image_url = info['images'][0]['src']
image = Image.open(response.raw) response = requests.get(image_url, stream=True, headers=headers)
# save using File object assert(response.status_code==200)
img_io = BytesIO() response.raw.decode_content = True
image.save(img_io, format='JPEG') image = Image.open(response.raw)
new.image.save(f"{new.name}-{new.sku}.jpg", File(img_io), save=False) # save using File object
new.save() img_io = BytesIO()
except Exception as e: image.save(img_io, format=image.format)
logging.error(f"Could not add image to product: {str(e)}") new.image.save(f"{new.name}-{new.sku}.{image.format.lower()}", File(img_io), save=False)
new.save()
except AssertionError as e:
logging.error(f"Source image [{info['images'][0]['src']}] not reachable: {response.status_code}")
except Exception as e:
logging.error(f"Could not add image to product {new.sku} from [{info['images'][0]['src']}]: {str(e)}")
else:
logging.info(f"No image for Product {new.name}")
return new return new
else: else:
logging.error(f"{serializer.errors}") logging.error(f"{serializer.errors}")