From 6bd9dcdb814b2f03f053ff5f5bf9a20ffe092582 Mon Sep 17 00:00:00 2001
From: René 'Necoro' Neumann <necoro@necoro.net>
Date: Mon, 27 Feb 2017 01:18:09 +0100
Subject: Inserting Documents

---
 archivist/cli.py | 77 ++++++++++++++++++++++++++++++++++++++++++++++++--------
 1 file changed, 67 insertions(+), 10 deletions(-)

(limited to 'archivist/cli.py')

diff --git a/archivist/cli.py b/archivist/cli.py
index 0b8abe1..8238a30 100644
--- a/archivist/cli.py
+++ b/archivist/cli.py
@@ -61,6 +61,12 @@ def list_prefixes():
     for p in Prefix.select():
         print("  * %s (builtin: %s; pseudo: %s)" % (p.name, p.builtin, p.pseudo))
 
+def prefix_tag_name(name, prefix=None):
+    if prefix:
+        return "%s:%s" % (prefix, name)
+    else:
+        return name
+
 class PrefixTag:
     def __init__(self, tag, prefix = None):
         self.tag = tag
@@ -72,6 +78,16 @@ class PrefixTag:
         else:
             return self.tag
 
+    def __iter__(self):
+        yield self.prefix
+        yield self.tag
+
+    def prefixed_name(self):
+        return prefix_tag_name(self.tag, self.prefix)
+
+    def __str__(self):
+        return self.prefixed_name()
+
 class PrefixTagType(click.ParamType):
     name = 'prefixed tag'
 
@@ -87,12 +103,8 @@ class PrefixTagType(click.ParamType):
 
 TAG = PrefixTagType()
 
-@tag.command('add')
-@click.argument('name', type = TAG)
-@click.argument('description', required = False)
-def add_tag(name, description):
+def create_tag(name, description=None):
     from .model import Tag, Prefix, db
-
     with db.atomic():
         if name.prefix:
             prefix, created = Prefix.get_or_create(name = name.prefix)
@@ -100,14 +112,20 @@ def add_tag(name, description):
         if not created and prefix.pseudo:
             raise click.UsageError("Prefix '%s' is not allowed to carry additional tags." % name.prefix)
 
-        tag, created = Tag.create_or_get(name = name.tag, prefix = name.prefix, description = description)
-        if not created:
-            print("Tag already existed:", tag)
+        return Tag.create_or_get(name = name.tag, prefix = name.prefix, description = description)
+
+@tag.command('add')
+@click.argument('name', type = TAG)
+@click.argument('description', required = False)
+def add_tag(name, description):
+    created, tag = create_tag(name, description)
+    if not created:
+        print("Tag already existed:", tag)
 
 @tag.command('edit')
-@click.option('--description')
 @click.argument('name', type = TAG)
-def edit_tag(description, name):
+@click.option('--description')
+def edit_tag(name, description):
     from .model import Tag, Prefix, db
 
     try:
@@ -118,3 +136,42 @@ def edit_tag(description, name):
     if description:
         tag.description = description
         tag.save()
+
+@cli.group()
+def doc():
+    """Document handling"""
+    pass
+
+@doc.command('add')
+@click.argument('file', type=click.File(mode = 'rb'))
+@click.argument('tags', type = TAG, nargs = -1)
+@click.option('--create-tags', '-c', is_flag = True)
+@click.option('--ignore-missing-tags', '-i', is_flag = True)
+def add_doc(file, tags, create_tags, ignore_missing_tags):
+    """Add a new document together with the given tags."""
+
+    from .model import Document, Tag, db, DocumentTag
+    from .peewee_ext import sqlite_tuple_in
+    import magic
+
+    with db.atomic():
+        if tags:
+            if create_tags:
+                tags = [create_tag(tag)[0] for tag in tags]
+            else:
+                fetched_tags = Tag.select().where(sqlite_tuple_in((Tag.prefix, Tag.name), tags)).execute()
+                if len(fetched_tags) < len(tags):
+                    print("Some tags were not present:", ', '.join(set(map(str, tags)).difference(map(str, fetched_tags))))
+                    if not ignore_missing_tags:
+                        raise click.ClickException("Not all tags present")
+                tags = fetched_tags
+
+        mimetype = magic.from_file(file.name, mime=True)
+
+        doc = Document.create(content = file.read(),
+                file_type = mimetype,
+                original_path = file.name,
+                direction = Document.Direction.IN)
+
+        for t in tags:
+            DocumentTag.create(document = doc, tag = t)
-- 
cgit v1.2.3-54-g00ecf