staff.py 2.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182
  1. from dbsynce.models import Company
  2. from dbsynce.models import Relationship
  3. from dbsynce.models import Permissions
  4. from django.contrib.auth.decorators import login_required
  5. from django.http import JsonResponse
  6. from django.utils.translation import gettext as _
  7. from django_tables2 import SingleTableView
  8. from django.views.generic import TemplateView, DetailView
  9. from sharix_admin.tables import PartnersTable
  10. from sharix_admin.tables import PermissionsTable
  11. from sharix_admin.tables import RelationshipTable
  12. from sharix_admin.utils import group_required
  13. from .base import BaseView
  14. from django.contrib.auth import get_user_model
  15. from django.contrib.auth.mixins import UserPassesTestMixin
  16. from django.contrib.auth.models import Group
  17. from django.utils.translation import gettext as _
  18. #TODO This is abslutely incorrect, custom page needed!
  19. class AccessControlListView(UserPassesTestMixin, SingleTableView):
  20. table_class = PermissionsTable
  21. queryset = Permissions.objects.all()
  22. template_name = 'sharix_admin/staff/access_control.html'
  23. def get_context_data(self, **kwargs):
  24. context = super().get_context_data(**kwargs)
  25. context.update({
  26. 'title': _('User roles Management'),
  27. 'object_list': context['object_list'],
  28. 'groups': Group.objects.all()
  29. })
  30. return context
  31. def test_func(self) -> bool or None:
  32. group_names = ('PLATFORM-ADMIN')
  33. if bool(self.request.user.groups.filter(name=group_names)) or self.request.user.is_superuser:
  34. return True
  35. return False
  36. class RelationshipListView(BaseView, SingleTableView):
  37. page_title = _('Relationship')
  38. page_name = 'relationship'
  39. table_class = RelationshipTable
  40. queryset = Relationship.objects.all()
  41. template_name = 'sharix_admin/staff/relationship.html'
  42. def get_context_data(self, **kwargs):
  43. context = super().get_context_data(**kwargs)
  44. context.update({
  45. 'object_list': context['object_list'],
  46. })
  47. return context
  48. def test_func(self):
  49. group_names = ('PLATFORM-ADMIN')
  50. if bool(self.request.user.groups.filter(name=group_names)) or self.request.user.is_superuser:
  51. return True
  52. return False
  53. @login_required
  54. @group_required('PLATFORM-ADMIN')
  55. def change_relationship_status(request):
  56. if request.method == 'POST':
  57. relationship_id = request.POST.get('relationship')
  58. new_status = request.POST.get('new_status')
  59. relationship = SXRelationship.objects.get(pk=relationship)
  60. relationship.status = new_status
  61. relationship.save()
  62. return JsonResponse({'status': 'success'})
  63. else:
  64. return JsonResponse({'status': 'error'})