def __init__(self, *args, **kwargs):
super(ProductForm, self).__init__(*args, **kwargs)
if get_current_user().is_partner:
clients = Client.objects.filter(partner_uuid=Partner.objects.get(
user=get_current_user()))
self.fields['client_uuid'].queryset = clients
def __init__(self, *args, **kwargs):
super(WeatherForm, self).__init__(*args, **kwargs)
if get_current_user().is_partner:
self.fields['store_uuid'].queryset = Store.objects.filter(
client_uuid__in=Client.objects.filter(
partner_uuid=Partner.objects.get(
user=get_current_user())).values_list('uuid',
flat=True))
def __init__(self, *args, **kwargs):
super(DemographicForm, self).__init__(*args, **kwargs)
if get_current_user().is_partner:
clients_uuid = Client.objects.filter(
partner_uuid=Partner.objects.get(
user=get_current_user())).values_list('uuid', flat=True)
self.fields['sensor_uuid'].queryset = Sensor.objects.filter(
client_uuid__in=clients_uuid)
def __init__(self, *args, **kwargs):
super(TransactionLinesForm, self).__init__(*args, **kwargs)
if get_current_user().is_partner:
clients_uuid = Client.objects.filter(
partner_uuid=Partner.objects.get(
user=get_current_user())).values_list('uuid', flat=True)
self.fields[
'transaction_uuid'].queryset = Transaction.objects.filter(
client_uuid__in=clients_uuid)
def __init__(self, *args, **kwargs):
super(PointOfSaleForm, self).__init__(*args, **kwargs)
if get_current_user().is_partner:
clients = Client.objects.filter(partner_uuid=Partner.objects.get(
user=get_current_user()))
clients_uuid = clients.values_list('uuid', flat=True)
self.fields['client_uuid'].queryset = clients
self.fields['sensor_uuid'].queryset = Sensor.objects.filter(
client_uuid__in=clients_uuid)
self.fields['store_uuid'].queryset = Store.objects.filter(
client_uuid__in=clients_uuid)
def get_queryset(self, request):
qs = super(ProductAdmin, self).get_queryset(request)
if request.user.is_superuser:
return qs
return qs.filter(client_uuid__in=Client.objects.filter(
partner_uuid=Partner.objects.get(
user=get_current_user())).values_list('uuid', flat=True))
def get_queryset(self, request):
qs = super(DemographicAdmin, self).get_queryset(request)
if request.user.is_superuser:
return qs
clients_uuid = Client.objects.filter(partner_uuid=Partner.objects.get(
user=get_current_user())).values_list('uuid', flat=True)
return qs.filter(sensor_uuid__in=Sensor.objects.filter(
client_uuid__in=clients_uuid).values_list('uuid', flat=True))
def get_queryset(self, request):
qs = super(TransactionLinesAdmin, self).get_queryset(request)
if request.user.is_superuser:
return qs
clients_uuid = Client.objects.filter(partner_uuid=Partner.objects.get(
user=get_current_user())).values_list('uuid', flat=True)
return qs.filter(transaction_uuid__in=Transaction.objects.filter(
client_uuid__in=clients_uuid))
def __init__(self, *args, **kwargs):
self.check_balena_application_uuid = False
super(SensorForm, self).__init__(*args, **kwargs)
if 'instance' in kwargs:
if kwargs['instance'] != None:
if kwargs['instance'].balena_application_uuid and kwargs[
'instance']:
self.fields['balena_application_uuid'].widget.attrs[
'readonly'] = True
self.check_balena_application_uuid = True
self.fields['license'].widget.attrs['readonly'] = True
self.fields['api_key'].widget.attrs['readonly'] = True
# self.fields['balena_uuid'].widget.attrs['readonly'] = True
if get_current_user().is_partner:
clients = Client.objects.filter(partner_uuid=Partner.objects.get(
user=get_current_user()))
self.fields['client_uuid'].queryset = clients
def lookups(self, request, model_admin):
"""
Returns a list of tuples. The first element in each
tuple is the coded value for the option that will
appear in the URL query. The second element is the
human-readable name for the option that will appear
in the right sidebar.
"""
list_of_clients = []
if request.user.is_superuser:
queryset = Client.objects.all()
else:
queryset = Client.objects.filter(partner_uuid=Partner.objects.get(
user=get_current_user()))
for clients in queryset:
list_of_clients.append((str(clients.uuid), clients.name))
return sorted(list_of_clients, key=lambda tp: tp[1])
