Cleanups, unused import, unused arguments removal

This commit is contained in:
Roberto Rosario
2014-11-02 23:21:14 -04:00
parent d65cd68aec
commit f52123cd7a
6 changed files with 14 additions and 121 deletions

View File

@@ -1,104 +0,0 @@
# better_yaml.py
"""
Customized YAML serializer, with more condensed and readable output.
Rather than producing a flat list of objects with the same three attributes:
- fields: {...}
model: modelname
pk: 123
This serializer nests the data, grouping by model name, then indexing by
primary key. For example, instead of this output, as produced by the default
YAML serializer:
- fields: {name: blue}
model: app.firstmodel
pk: 3
- fields: {name: red}
model: app.firstmodel
pk: 1
- fields: {name: green}
model: app.firstmodel
pk: 2
- fields: {name: crumbly}
model: app.secondmodel
pk: 2
- fields: {name: squishy}
model: app.secondmodel
pk: 1
You'll get this output:
app.firstmodel:
1: {name: red}
2: {name: green}
3: {name: blue}
app.secondmodel:
1: {name: squishy}
2: {name: crumbly}
To use this customized serializer and deserializer, save this file
somewhere in your Django project, then add this to your settings.py:
SERIALIZATION_MODULES = {
'yaml': 'path.to.better_yaml',
}
Note that this serializer is NOT compatible with the default Django
YAML serializer; this one uses nested dictionaries, while the default
one uses a flat list of object dicts.
Requires PyYaml (http://pyyaml.org/), of course.
"""
from StringIO import StringIO
import yaml
from django.core.serializers.python import Deserializer as PythonDeserializer
from django.core.serializers.pyyaml import Serializer as YamlSerializer
from django.utils.encoding import smart_unicode
class Serializer (YamlSerializer):
"""
Serialize database objects as nested dicts, indexed first by
model name, then by primary key.
"""
def start_serialization(self):
self._current = None
self.objects = {}
def end_object(self, obj):
model = smart_unicode(obj._meta)
pk = obj._get_pk_val()
if model not in self.objects:
self.objects[model] = {}
self.objects[model][pk] = self._current
self._current = None
def Deserializer(stream_or_string, **options):
"""
Deserialize a stream or string of YAML data,
as written by the Serializer above.
"""
if isinstance(stream_or_string, basestring):
stream = StringIO(stream_or_string)
else:
stream = stream_or_string
# Reconstruct the flat object list as PythonDeserializer expects
# NOTE: This could choke on large data sets, since it
# constructs the flattened data list in memory
data = []
for model, objects in yaml.load(stream).iteritems():
# Add the model name back into each object dict
for pk, fields in objects.iteritems():
data.append({'model': model, 'pk': pk, 'fields': fields})
# Deserialize the flattened data
for obj in PythonDeserializer(data, **options):
yield obj

View File

@@ -174,7 +174,8 @@ def validate_path(path):
# If doesn't exist try to create it
try:
os.mkdir(path)
except:
except Exception as exception:
logger.debug('unhandled exception: %s', exception)
return False
# Check if it is writable
@@ -182,7 +183,8 @@ def validate_path(path):
fd, test_filepath = tempfile.mkstemp(dir=path)
os.close(fd)
os.unlink(test_filepath)
except:
except Exception as exception:
logger.debug('unhandled exception: %s', exception)
return False
return True

View File

@@ -11,8 +11,6 @@ from documents.models import Document, DocumentType
from .managers import IndexManager
from .settings import AVAILABLE_INDEXING_FUNCTIONS
available_indexing_functions_string = (_(u'Available functions: %s') % u','.join([u'%s()' % name for name, function in AVAILABLE_INDEXING_FUNCTIONS.items()])) if AVAILABLE_INDEXING_FUNCTIONS else u''
class Index(models.Model):
name = models.CharField(unique=True, max_length=64, verbose_name=_(u'Name'), help_text=_(u'Internal name used to reference this index.'))
@@ -43,7 +41,7 @@ class Index(models.Model):
def save(self, *args, **kwargs):
"""Automatically create the root index template node"""
super(Index, self).save(*args, **kwargs)
index_template_node_root, created = IndexTemplateNode.objects.get_or_create(parent=None, index=self)
IndexTemplateNode.objects.get_or_create(parent=None, index=self)
def get_document_types_names(self):
return u', '.join([unicode(document_type) for document_type in self.document_types.all()] or [u'All'])
@@ -70,7 +68,6 @@ class IndexTemplateNode(MPTTModel):
parent = TreeForeignKey('self', null=True, blank=True)
index = models.ForeignKey(Index, verbose_name=_(u'Index'), related_name='node_templates')
expression = models.CharField(max_length=128, verbose_name=_(u'Indexing expression'), help_text=_(u'Enter a python string expression to be evaluated.'))
# % available_indexing_functions_string)
enabled = models.BooleanField(default=True, verbose_name=_(u'Enabled'), help_text=_(u'Causes this node to be visible and updated when document data changes.'))
link_documents = models.BooleanField(default=False, verbose_name=_(u'Link documents'), help_text=_(u'Check this option to have this node act as a container for documents and not as a parent for further nodes.'))

View File

@@ -110,14 +110,14 @@ class APIDocumentFolderListView(generics.ListAPIView):
class APIFolderDocumentView(views.APIView):
def delete(self, request, *args, **kwargs):
def delete(self, request):
"""Remove a document from the selected folder."""
folder = get_object_or_404(Folder, pk=self.kwargs['pk'])
try:
Permission.objects.check_permissions(self.request.user, [PERMISSION_FOLDER_REMOVE_DOCUMENT])
Permission.objects.check_permissions(request.user, [PERMISSION_FOLDER_REMOVE_DOCUMENT])
except PermissionDenied:
AccessEntry.objects.check_access(PERMISSION_FOLDER_REMOVE_DOCUMENT, self.request.user, folder)
AccessEntry.objects.check_access(PERMISSION_FOLDER_REMOVE_DOCUMENT, request.user, folder)
document = get_object_or_404(Document, pk=self.kwargs['document_pk'])
folder.documents.remove(document)
@@ -128,9 +128,9 @@ class APIFolderDocumentView(views.APIView):
folder = get_object_or_404(Folder, pk=self.kwargs['pk'])
try:
Permission.objects.check_permissions(self.request.user, [PERMISSION_FOLDER_ADD_DOCUMENT])
Permission.objects.check_permissions(request.user, [PERMISSION_FOLDER_ADD_DOCUMENT])
except PermissionDenied:
AccessEntry.objects.check_access(PERMISSION_FOLDER_ADD_DOCUMENT, self.request.user, folder)
AccessEntry.objects.check_access(PERMISSION_FOLDER_ADD_DOCUMENT, request.user, folder)
document = get_object_or_404(Document, pk=self.kwargs['document_pk'])
folder.documents.add(document)

View File

@@ -1,7 +1,5 @@
from __future__ import absolute_import
import tempfile
from django.shortcuts import get_object_or_404
from converter.exceptions import UnkownConvertError, UnknownFileFormat

View File

@@ -111,9 +111,9 @@ class APIDocumentTagView(views.APIView):
document = get_object_or_404(Document, pk=self.kwargs['document_pk'])
try:
Permission.objects.check_permissions(self.request.user, [PERMISSION_TAG_REMOVE])
Permission.objects.check_permissions(request.user, [PERMISSION_TAG_REMOVE])
except PermissionDenied:
AccessEntry.objects.check_access(PERMISSION_TAG_REMOVE, self.request.user, document)
AccessEntry.objects.check_access(PERMISSION_TAG_REMOVE, request.user, document)
tag = get_object_or_404(Tag, pk=self.kwargs['pk'])
tag.documents.remove(document)
@@ -126,9 +126,9 @@ class APIDocumentTagView(views.APIView):
document = get_object_or_404(Document, pk=self.kwargs['document_pk'])
try:
Permission.objects.check_permissions(self.request.user, [PERMISSION_TAG_ATTACH])
Permission.objects.check_permissions(request.user, [PERMISSION_TAG_ATTACH])
except PermissionDenied:
AccessEntry.objects.check_access(PERMISSION_TAG_ATTACH, self.request.user, document)
AccessEntry.objects.check_access(PERMISSION_TAG_ATTACH, request.user, document)
tag = get_object_or_404(Tag, pk=self.kwargs['pk'])
tag.documents.add(document)