init

sumasen_easyocr/admin.py

@@ -0,0 +1,4 @@
+from django.contrib import admin
+from .models import Document
+
+admin.site.register(Document)

sumasen_easyocr/apps.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class SumasenEasyocrConfig(AppConfig):
+    default_auto_field = 'django.db.models.BigAutoField'
+    name = 'sumasen_easyocr'

sumasen_easyocr/migrations/0001_initial.py

@@ -0,0 +1,31 @@
+# Generated by Django 3.2.9 on 2021-12-02 10:05
+
+import datetime
+from django.conf import settings
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Document',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.CharField(max_length=512, verbose_name='name')),
+                ('file', models.FileField(upload_to='%y/%m/%d')),
+                ('uploaded_date', models.DateField(default=datetime.date.today)),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL)),
+            ],
+            options={
+                'verbose_name_plural': 'Documents',
+            },
+        ),
+    ]

sumasen_easyocr/models.py

@@ -0,0 +1,51 @@
+from django.db import models
+from django.db.models.deletion import CASCADE
+from django.utils.translation import gettext_lazy as _
+from django.contrib.auth.models import User
+import datetime
+import os
+from django.db.models.signals import post_save, post_delete
+from django.dispatch import receiver
+import easyocr
+
+
+class Document(models.Model):
+    name=models.CharField(_('name'), max_length=512)
+    user = models.ForeignKey(User, on_delete=CASCADE)
+    file = models.FileField(upload_to='%y/%m/%d', blank=False)
+    uploaded_date = models.DateField(default=datetime.date.today)
+
+    class Meta:
+        verbose_name_plural = "Documents"
+
+    def __str__(self):
+        return self.name
+
+
+@receiver(post_save, sender=Document)
+def publish_date(sender, instance, created, **kwargs):
+    file = instance.file.path
+    file_format = os.path.basename(file).split('.')[-1]
+    file_name = os.path.basename(file).split('.')[0]
+    file_path = os.path.dirname(file)
+    name = instance.name
+    
+    #os.remove(file)
+
+    try:
+        reader = easyocr.Reader(['ja','en'])
+        result = reader.readtext(file)
+        print('@@@@@@@@@@')
+        print(result)
+        print('@@@@@@@@@@')
+        
+    except Exception as e:
+        pass
+        
+
+
+@receiver(post_delete, sender=Document)
+def delete_layer(sender, instance, **kwargs):
+    if instance.file:
+        if os.path.isfile(instance.file.path):
+            os.remove(instance.file.path)

sumasen_easyocr/tests.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

sumasen_easyocr/urls.py

@@ -0,0 +1,7 @@
+from django.contrib import admin
+from django.urls import path, re_path
+from .views import FileUploadView
+
+urlpatterns = [
+    re_path(r'^upload/', FileUploadView.as_view())
+]

sumasen_easyocr/views.py

@@ -0,0 +1,28 @@
+from rest_framework.views import APIView
+from rest_framework.parsers import MultiPartParser
+from rest_framework.response import Response
+from rest_framework import status
+import easyocr
+
+class FileUploadView(APIView):
+    parser_classes = [MultiPartParser,]
+
+    def post(self, request, format='jpg'):
+        up_file = request.FILES['file']
+        destination = open('/Users/mohamednouffer/workspace/akira_san/sumasen_ocr/sumasen_easyocr/uploaded/' + up_file.name, 'wb+')
+        for chunk in up_file.chunks():
+            destination.write(chunk)
+        destination.close()
+
+        try:
+            reader = easyocr.Reader(['ja','en'])
+            result = reader.readtext('/Users/mohamednouffer/workspace/akira_san/sumasen_ocr/sumasen_easyocr/uploaded/' + up_file.name)
+            #print('@@@@@@@@@@')
+            #print(result)
+            #print('@@@@@@@@@@')
+            return Response(result, status.HTTP_201_CREATED)
+            #return Response({''}, status.HTTP_201_CREATED)
+        
+        except Exception as e:
+            print(e)
+            return Response({'Error': "Error occured"}, status.HTTP_201_CREATED)