mirror of
https://github.com/inventree/InvenTree
synced 2024-08-30 18:33:04 +00:00
PEP style fixes
This commit is contained in:
parent
8e1d9c281a
commit
c2a8cfd183
@ -41,8 +41,6 @@ from .forms import DeleteForm, EditUserForm, SetPasswordForm
|
||||
from .forms import SettingCategorySelectForm
|
||||
from .helpers import str2bool
|
||||
|
||||
from rest_framework import views
|
||||
|
||||
|
||||
def auth_request(request):
|
||||
"""
|
||||
@ -57,84 +55,6 @@ def auth_request(request):
|
||||
return HttpResponse(status=403)
|
||||
|
||||
|
||||
class TreeSerializer(views.APIView):
|
||||
""" JSON View for serializing a Tree object.
|
||||
|
||||
Turns a 'tree' model into a JSON object compatible with bootstrap-treview.
|
||||
|
||||
Ref: https://github.com/jonmiles/bootstrap-treeview
|
||||
"""
|
||||
|
||||
@property
|
||||
def root_url(self):
|
||||
""" Return the root URL for the tree. Implementation is class dependent.
|
||||
|
||||
Default implementation returns #
|
||||
"""
|
||||
|
||||
return '#'
|
||||
|
||||
def itemToJson(self, item):
|
||||
|
||||
data = {
|
||||
'pk': item.id,
|
||||
'text': item.name,
|
||||
'href': item.get_absolute_url(),
|
||||
'tags': [item.item_count],
|
||||
}
|
||||
|
||||
if item.has_children:
|
||||
nodes = []
|
||||
|
||||
for child in item.children.all().order_by('name'):
|
||||
nodes.append(self.itemToJson(child))
|
||||
|
||||
data['nodes'] = nodes
|
||||
|
||||
return data
|
||||
|
||||
def get_items(self):
|
||||
|
||||
return self.model.objects.all()
|
||||
|
||||
def generate_tree(self):
|
||||
|
||||
nodes = []
|
||||
|
||||
items = self.get_items()
|
||||
|
||||
# Construct the top-level items
|
||||
top_items = [i for i in items if i.parent is None]
|
||||
|
||||
top_count = 0
|
||||
|
||||
# Construct the top-level items
|
||||
top_items = [i for i in items if i.parent is None]
|
||||
|
||||
for item in top_items:
|
||||
nodes.append(self.itemToJson(item))
|
||||
top_count += item.item_count
|
||||
|
||||
self.tree = {
|
||||
'pk': None,
|
||||
'text': self.title,
|
||||
'href': self.root_url,
|
||||
'nodes': nodes,
|
||||
'tags': [top_count],
|
||||
}
|
||||
|
||||
def get(self, request, *args, **kwargs):
|
||||
""" Respond to a GET request for the Tree """
|
||||
|
||||
self.generate_tree()
|
||||
|
||||
response = {
|
||||
'tree': [self.tree]
|
||||
}
|
||||
|
||||
return JsonResponse(response, safe=False)
|
||||
|
||||
|
||||
class InvenTreeRoleMixin(PermissionRequiredMixin):
|
||||
"""
|
||||
Permission class based on user roles, not user 'permissions'.
|
||||
|
@ -6,7 +6,6 @@ Provides a JSON API for the Part app
|
||||
from __future__ import unicode_literals
|
||||
|
||||
from django.conf.urls import url, include
|
||||
from django.urls import reverse
|
||||
from django.http import JsonResponse
|
||||
from django.db.models import Q, F, Count, Min, Max, Avg
|
||||
from django.db import transaction
|
||||
@ -43,7 +42,6 @@ from build.models import Build
|
||||
|
||||
from . import serializers as part_serializers
|
||||
|
||||
from InvenTree.views import TreeSerializer
|
||||
from InvenTree.helpers import str2bool, isNull
|
||||
from InvenTree.api import AttachmentMixin
|
||||
|
||||
|
@ -7,17 +7,14 @@ from __future__ import unicode_literals
|
||||
|
||||
from datetime import datetime, timedelta
|
||||
|
||||
from django.utils.translation import ugettext_lazy as _
|
||||
|
||||
from django.conf.urls import url, include
|
||||
from django.urls import reverse
|
||||
from django.http import JsonResponse
|
||||
from django.db.models import Q
|
||||
|
||||
from rest_framework import status
|
||||
from rest_framework.serializers import ValidationError
|
||||
from rest_framework.response import Response
|
||||
from rest_framework import generics, filters, permissions
|
||||
from rest_framework import generics, filters
|
||||
|
||||
from django_filters.rest_framework import DjangoFilterBackend
|
||||
from django_filters import rest_framework as rest_filters
|
||||
@ -42,7 +39,6 @@ import common.models
|
||||
|
||||
import stock.serializers as StockSerializers
|
||||
|
||||
from InvenTree.views import TreeSerializer
|
||||
from InvenTree.helpers import str2bool, isNull
|
||||
from InvenTree.api import AttachmentMixin
|
||||
from InvenTree.filters import InvenTreeOrderingFilter
|
||||
|
Loading…
Reference in New Issue
Block a user