107 lines
5.0 KiB
Python
107 lines
5.0 KiB
Python
from rest_framework.views import APIView
|
|
from rest_framework.response import Response
|
|
import pandas as pd
|
|
from apps.shop.models import Product as ProductModel, ProductCategory as ProductCategoryModel
|
|
from apps.workspace.models import Workspace as WorkspaceModel, City as CityModel
|
|
from apps.customer.models import Customer as CustomerModel
|
|
from apps.customer.models import ResponsibleGroup as ResponsibleGroupModel
|
|
from django.core.exceptions import ValidationError as DjangoValidationError
|
|
from django.db import IntegrityError, transaction
|
|
from rest_framework import exceptions
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from utils import custom_permissions
|
|
from utils.get_user_workspace import get_user_workspace
|
|
from utils.validations.mobile import normalize_mobile
|
|
|
|
|
|
class ImportExcelView(APIView):
|
|
permission_classes = (IsAuthenticated, custom_permissions.WorkspaceAdminPermission)
|
|
|
|
def get(self, request):
|
|
# Keep the legacy URL and filename, but use the same validated importer as Admin.
|
|
from django.core.files import File
|
|
from apps.shop.product_import import parse_workbook, import_products
|
|
try:
|
|
with open('pe.xlsx', 'rb') as source:
|
|
rows = parse_workbook(File(source, name='pe.xlsx'))
|
|
for row in rows:
|
|
row['values'].setdefault('view_only', 'true')
|
|
result = import_products(get_user_workspace(request), rows,
|
|
{'mode': 'upsert', 'create_categories': True, 'errors': 'atomic', 'ordering': 'preserve'})
|
|
except (OSError, DjangoValidationError) as error:
|
|
raise exceptions.ValidationError(getattr(error, 'messages', ['فایل اکسل محصولات خوانده نشد.']))
|
|
if not result['committed']:
|
|
raise exceptions.ValidationError(result)
|
|
return Response(data={'result': 'ok', 'imported': result['created'] + result['updated']})
|
|
|
|
|
|
class ImportExcelCitysView(APIView):
|
|
|
|
def get(self, request):
|
|
|
|
|
|
df = pd.read_excel('cities.xlsx')
|
|
for index, row in df.iterrows():
|
|
city_name = row['name']
|
|
pv_id = row['province_id']
|
|
|
|
try:
|
|
ostan = CityModel.objects.get(pv_id=pv_id, is_ostan=True)
|
|
except Exception as exp:
|
|
print(city_name)
|
|
print(pv_id)
|
|
|
|
city_obj = CityModel()
|
|
city_obj.name = city_name
|
|
city_obj.is_ostan = False
|
|
city_obj.ostan = ostan
|
|
city_obj.save()
|
|
|
|
|
|
return Response(data={"result": 'ok'})
|
|
|
|
|
|
def customer_from_excel(row, workspace):
|
|
mobile = row['mobile']
|
|
if pd.isna(mobile) or pd.isna(row['name']):
|
|
raise exceptions.ValidationError('نام و شماره موبایل مشتری الزامی است.')
|
|
if isinstance(mobile, float) and mobile.is_integer():
|
|
mobile = int(mobile)
|
|
city = CityModel.objects.filter(name=row['shar'], ostan__name=row['ostan'], ostan__is_ostan=True).first()
|
|
city = city or CityModel.objects.filter(pk=1229).first()
|
|
if not city:
|
|
raise exceptions.ValidationError('شهر مشتری در فایل پیدا نشد.')
|
|
group = ResponsibleGroupModel.objects.filter(pk=row['id_mantaghe'], sys_ws=workspace).first()
|
|
if not group:
|
|
raise exceptions.ValidationError('گروه مسئول متعلق به این فضای کاری نیست.')
|
|
customer = CustomerModel(name_family=row['name'], mobile=normalize_mobile(mobile), city=city,
|
|
responsible_group=group, sys_ws=workspace,
|
|
phone=None if pd.isna(row['phone']) else row['phone'],
|
|
address=None if pd.isna(row['address']) else row['address'])
|
|
customer.full_clean()
|
|
customer.save()
|
|
|
|
|
|
class ImportExcelCustomersView(APIView):
|
|
permission_classes = (IsAuthenticated, custom_permissions.WorkspaceAdminPermission)
|
|
|
|
@transaction.atomic
|
|
def get(self, request):
|
|
workspace = get_user_workspace(request)
|
|
try:
|
|
data = pd.read_excel('customersData.xlsx')
|
|
except (OSError, ValueError):
|
|
raise exceptions.ValidationError('فایل اکسل مشتریان خوانده نشد.')
|
|
for index, row in data.iterrows():
|
|
try:
|
|
customer_from_excel(row, workspace)
|
|
except DjangoValidationError as error:
|
|
raise exceptions.ValidationError({'row': int(index)+2, 'errors': error.message_dict})
|
|
except IntegrityError:
|
|
raise exceptions.ValidationError({'row': int(index)+2, 'mobile': 'شماره موبایل مشتری تکراری است.'})
|
|
except exceptions.ValidationError as error:
|
|
raise exceptions.ValidationError({'row': int(index)+2, 'errors': error.detail})
|
|
except (KeyError, TypeError, ValueError):
|
|
raise exceptions.ValidationError({'row': int(index)+2, 'error': 'اطلاعات این ردیف معتبر نیست.'})
|
|
return Response(data={'result': 'ok', 'imported': len(data)})
|