haikal/inventory/mixins.py
Marwan Alwali 53d7db0bfc update
2025-02-10 14:42:07 +03:00

66 lines
2.2 KiB
Python

from django import forms
from django.utils.translation import get_language
from django.urls import reverse, reverse_lazy
class AddClassMixin:
"""
Mixin for adding classes to a model.
"""
def add_class_to_fields(self):
"""
Adds the class to the fields of the model.
:return: class names form-control or form-select
"""
for field_name, field in self.fields.items():
if isinstance(field.widget, forms.Select):
existing_classes = field.widget.attrs.get('class', '')
field.widget.attrs['class'] = f"{existing_classes} form-select form-select-sm".strip()
else:
existing_classes = field.widget.attrs.get('class', '')
field.widget.attrs['class'] = f"{existing_classes} form-control form-control-sm".strip()
class LocalizedNameMixin:
"""
Mixin to provide a reusable get_localized_name method.
"""
def get_local_name(self):
"""
Returns the localized name based on the current language.
"""
if get_language() == 'ar':
return getattr(self, 'arabic_name', None)
return getattr(self, 'name', None)
# class AddDealerInstanceMixin:
# def form_valid(self, form):
# if form.is_valid():
# form.instance.dealer = self.request.user.dealer.get_root_dealer
# form.save()
# return super().form_valid(form)
# else:
# return form.errors
class BaseDateNavigationUrlMixIn:
BASE_DATE_URL_KWARGS = (
'entity_slug',
'unit_slug',
'ledger_pk',
'account_pk',
'coa_slug'
)
def get_context_data(self, **kwargs):
context = super(BaseDateNavigationUrlMixIn, self).get_context_data(**kwargs)
self.get_base_date_nav_url(context)
return context
def get_base_date_nav_url(self, context, **kwargs):
view_name = context['view'].request.resolver_match.url_name
view_name_base = '-'.join(view_name.split('-')[:2])
context['date_navigation_url'] = reverse(
view_name_base,
kwargs={
k: v for k, v in self.kwargs.items() if
k in self.BASE_DATE_URL_KWARGS
})