New command: update a field in the results of a query

4 files changed, 71 insertions, 0 deletions

diff --git a/lesana/collection.py b/lesana/collection.py
index 4bec594..25b86e5 100644
--- a/lesana/collection.py
+++ b/lesana/collection.py
@@ -452,6 +452,18 @@ class Collection(object):
             repo.index.remove([f_path])
         os.remove(f_path)
 
+    def update_field(self, query, field, value):
+        self.start_search(query)
+        changed = []
+        for e in self.get_all_search_results():
+            e.data[field] = value
+            changed.append(e)
+        self.save_entries(changed)
+        self.git_add_files([
+            os.path.join(self.itemdir, e.fname) for e in changed
+        ])
+        self.update_cache([e.fname for e in changed])
+
     def get_template(self, template_fname, searchpath='.'):
         env = jinja2.Environment(
             loader=jinja2.FileSystemLoader(
diff --git a/lesana/command.py b/lesana/command.py
index f2f3e56..c2bdf28 100644
--- a/lesana/command.py
+++ b/lesana/command.py
@@ -331,3 +331,29 @@ class Remove(Command):
     def main(self):
         collection = self.collection_class(self.args.collection)
         collection.remove_entries(eids=self.args.entries)
+
+
+class Update(Command):
+    arguments = [
+        (['--collection', '-c'], dict(
+            help='The collection to work on (default .)',
+            )),
+        (['--field', '-f'], dict(
+            help='The field to change',
+            )),
+        (['--value', '-t'], dict(
+            help='The value to set',
+            )),
+        (['query'], dict(
+            help='Xapian query to search in the collection',
+            nargs='+'
+            )),
+        ]
+
+    def main(self):
+        collection = self.collection_class(self.args.collection)
+        collection.update_field(
+            ' '.join(self.args.query),
+            field=self.args.field,
+            value=self.args.value
+        )
diff --git a/scripts/lesana b/scripts/lesana
index 408aa2c..5591b3b 100755
--- a/scripts/lesana
+++ b/scripts/lesana
@@ -15,6 +15,7 @@ class Lesana():
         ('show', lesana.command.Show()),
         ('index', lesana.command.Index()),
         ('search', lesana.command.Search()),
+        ('update', lesana.command.Update()),
         ('export', lesana.command.Export()),
         ('init', lesana.command.Init()),
         ('rm', lesana.command.Remove()),
diff --git a/tests/test_collection.py b/tests/test_collection.py
index e97e912..eb57fb0 100644
--- a/tests/test_collection.py
+++ b/tests/test_collection.py
@@ -154,6 +154,38 @@ class testCollection(unittest.TestCase):
         self.assertIn('11189ee4: Another item', res)
 
 
+class testSimpleCollection(unittest.TestCase):
+    def setUp(self):
+        self.tmpdir = tempfile.mkdtemp()
+        shutil.copytree('tests/data/simple', self.tmpdir, dirs_exist_ok=True)
+        self.collection = lesana.Collection(self.tmpdir)
+
+    def tearDown(self):
+        shutil.rmtree(self.tmpdir)
+
+    def test_update(self):
+        self.collection.update_field('Item', field="position", value="new_pos")
+        with open(os.path.join(
+            self.collection.basedir,
+            'items',
+            '11189ee47ddf4796b718a483b379f976.yaml'
+        )) as fp:
+            self.assertIn("new_pos", fp.read())
+            pass
+        self.assertEqual(
+            self.collection.entry_from_eid(
+                "11189ee47ddf4796b718a483b379f976"
+            ).data['position'],
+            "new_pos"
+        )
+
+        self.assertIsNone(
+            self.collection.entry_from_eid(
+                "8b69b063b2a64db7b5714294a69255c7"
+            ).data['position']
+        )
+
+
 class testEntries(unittest.TestCase):
     def setUp(self):
         self.collection = lesana.Collection('tests/data/simple')