Cutlery/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-11-03 19:17:13 -05:00
Code Issues Packages Projects Releases Wiki Activity

added migration step to create initial classifier, silences consumption warnings

Browse Source
This commit is contained in:
Jonas Winkler 2020-11-03 14:10:53 +01:00
parent 28ba634e6a
commit ebac10bdfb
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/documents/migrations/1000_update_paperless.py
View File

@ -41,6 +41,21 @@ def restore_filenames(apps, schema_editor):
pass 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): class Migration(migrations.Migration):
dependencies = [ dependencies = [
@ -50,6 +65,7 @@ class Migration(migrations.Migration):
operations = [ operations = [
migrations.RunPython(make_index, migrations.RunPython.noop), migrations.RunPython(make_index, migrations.RunPython.noop),
migrations.RunPython(restore_filenames), migrations.RunPython(restore_filenames),
migrations.RunPython(initialize_document_classifier, migrations.RunPython.noop),
migrations.RemoveField( migrations.RemoveField(
model_name='document', model_name='document',
name='filename', name='filename',