Fix

[redakcja.git] / apps / api / handlers / library_handlers.py

diff --git a/apps/api/handlers/library_handlers.py b/apps/api/handlers/library_handlers.py
index 39177f5..93d650b 100644 (file)
--- a/apps/api/handlers/library_handlers.py
+++ b/apps/api/handlers/library_handlers.py
@@ -60,8 +60,7 @@ class LibraryHandler(BaseHandler):
 
         documents = {}
         
 
         documents = {}
         

-        for docid in lib.documents():
-            docid = docid.decode('utf-8')
+        for docid in lib.documents():            

             documents[docid] = {
                 'url': reverse('document_view', args=[docid]),
                 'name': docid,
             documents[docid] = {
                 'url': reverse('document_view', args=[docid]),
                 'name': docid,


@@ -102,6 +101,9 @@ class LibraryHandler(BaseHandler):
         else:            
             data = request.FILES['ocr_file'].read().decode('utf-8')
 
         else:            
             data = request.FILES['ocr_file'].read().decode('utf-8')
 

+        if data is None:
+            return response.BadRequest().django_response('You must pass ocr_data or ocr_file.')
+

         if form.cleaned_data['generate_dc']:
             data = librarian.wrap_text(data, unicode(date.today()))
 
         if form.cleaned_data['generate_dc']:
             data = librarian.wrap_text(data, unicode(date.today()))
 


@@ -118,9 +120,10 @@ class LibraryHandler(BaseHandler):
                     doc = doc.quickwrite('xml', data.encode('utf-8'),
                         '$AUTO$ XML data uploaded.', user=request.user.username)
                 except Exception,e:
                     doc = doc.quickwrite('xml', data.encode('utf-8'),
                         '$AUTO$ XML data uploaded.', user=request.user.username)
                 except Exception,e:

+                    import traceback

                     # rollback branch creation
                     lib._rollback()
                     # rollback branch creation
                     lib._rollback()

-                    raise LibraryException("Exception occured:" + repr(e))
+                    raise LibraryException(traceback.format_exc())

 
                 url = reverse('document_view', args=[doc.id])
 
 
                 url = reverse('document_view', args=[doc.id])
 


@@ -133,8 +136,9 @@ class LibraryHandler(BaseHandler):
             finally:
                 lock.release()
         except LibraryException, e:
             finally:
                 lock.release()
         except LibraryException, e:

+            import traceback

             return response.InternalError().django_response(\
             return response.InternalError().django_response(\

-                {'exception': repr(e) })                
+                {'exception': traceback.format_exc()} )

         except DocumentAlreadyExists:
             # Document is already there
             return response.EntityConflict().django_response(\
         except DocumentAlreadyExists:
             # Document is already there
             return response.EntityConflict().django_response(\


@@ -233,7 +237,6 @@ class DocumentHTMLHandler(BaseHandler):
 #
 # Image Gallery
 #
 #
 # Image Gallery
 #

-from django.core.files.storage import FileSystemStorage

 
 class DocumentGalleryHandler(BaseHandler):
     allowed_methods = ('GET')
 
 class DocumentGalleryHandler(BaseHandler):
     allowed_methods = ('GET')


@@ -251,17 +254,20 @@ class DocumentGalleryHandler(BaseHandler):
 
             gallery = {'name': assoc.name, 'pages': []}
             
 
             gallery = {'name': assoc.name, 'pages': []}
             

-            for file in sorted(os.listdir(dirpath), key=natural_order()):
+            for file in os.listdir(dirpath):
+                file = file.decode('utf-8')
+                

                 log.info(file)
                 name, ext = os.path.splitext(os.path.basename(file))
 
                 log.info(file)
                 name, ext = os.path.splitext(os.path.basename(file))
 

-                if ext.lower() not in ['.png', '.jpeg', '.jpg']:
+                if ext.lower() not in [u'.png', u'.jpeg', u'.jpg']:

                     log.info("Ignoring: %s %s", name, ext)
                     continue
 
                     log.info("Ignoring: %s %s", name, ext)
                     continue
 

-                url = settings.MEDIA_URL + assoc.subpath + u'/' + file.decode('utf-8');
+                url = settings.MEDIA_URL + assoc.subpath + u'/' + file;

                 gallery['pages'].append(url)
                 gallery['pages'].append(url)

-                
+
+            gallery['pages'].sort()

             galleries.append(gallery)
 
         return galleries                      
             galleries.append(gallery)
 
         return galleries