mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2025-04-02 13:45:10 -05:00
74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
# Generated by Django 3.1.2 on 2020-10-29 14:29
|
|
import os
|
|
|
|
from django.db import migrations
|
|
|
|
from django.conf import settings
|
|
|
|
|
|
def make_index(apps, schema_editor):
|
|
Document = apps.get_model("documents", "Document")
|
|
documents = Document.objects.all()
|
|
print()
|
|
try:
|
|
print(" --> Creating document index...")
|
|
from whoosh.writing import AsyncWriter
|
|
from documents import index
|
|
ix = index.open_index(recreate=True)
|
|
with AsyncWriter(ix) as writer:
|
|
for document in documents:
|
|
index.update_document(writer, document)
|
|
except ImportError:
|
|
# index may not be relevant anymore
|
|
print(" --> Cannot create document index.")
|
|
|
|
|
|
def restore_filenames(apps, schema_editor):
|
|
Document = apps.get_model("documents", "Document")
|
|
for doc in Document.objects.all():
|
|
file_name = "{:07}.{}".format(doc.pk, doc.file_type)
|
|
if doc.storage_type == "gpg":
|
|
file_name += ".gpg"
|
|
|
|
if not doc.filename == file_name:
|
|
try:
|
|
print("file was renamed, restoring {} to {}".format(doc.filename, file_name))
|
|
os.rename(os.path.join(settings.ORIGINALS_DIR, doc.filename),
|
|
os.path.join(settings.ORIGINALS_DIR, file_name))
|
|
except PermissionError:
|
|
pass
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
|
|
def initialize_document_classifier(apps, schema_editor):
|
|
try:
|
|
print("Initalizing document classifier...")
|
|
from documents.classifier import DocumentClassifier
|
|
classifier = DocumentClassifier()
|
|
try:
|
|
classifier.train()
|
|
classifier.save_classifier()
|
|
except Exception as e:
|
|
print("Classifier error: {}".format(e))
|
|
except ImportError:
|
|
print("Document classifier not found, skipping")
|
|
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
|
|
dependencies = [
|
|
('documents', '0023_document_current_filename'),
|
|
]
|
|
|
|
operations = [
|
|
migrations.RunPython(make_index, migrations.RunPython.noop),
|
|
migrations.RunPython(restore_filenames),
|
|
migrations.RunPython(initialize_document_classifier, migrations.RunPython.noop),
|
|
migrations.RemoveField(
|
|
model_name='document',
|
|
name='filename',
|
|
),
|
|
]
|