Open a lesana directory

5 files changed, 62 insertions, 8 deletions

diff --git a/lesana/__init__.py b/lesana/__init__.py
index e69de29..969ac3f 100644
--- a/lesana/__init__.py
+++ b/lesana/__init__.py
@@ -0,0 +1 @@
+from .collection import Collection
diff --git a/lesana/collection.py b/lesana/collection.py
new file mode 100644
index 0000000..df38e89
--- /dev/null
+++ b/lesana/collection.py
@@ -0,0 +1,29 @@
+import os
+
+import ruamel.yaml
+import xapian
+
+
+class Collection(object):
+    """
+    """
+
+    def __init__(self, directory=None):
+        self.basedir = directory or os.getcwd()
+        try:
+            with open(os.path.join(self.basedir, 'schema.yaml')) as fp:
+                self.schema = ruamel.yaml.load(fp, ruamel.yaml.RoundTripLoader)
+        except FileNotFoundError:
+            self.schema = ruamel.yaml.load("")
+        os.makedirs(os.path.join(self.basedir, '.lesana'), exist_ok=True)
+        self.cache = xapian.WritableDatabase(
+            os.path.join(self.basedir, '.lesana/xapian'),
+            xapian.DB_CREATE_OR_OPEN
+            )
+
+    def update_cache(self, files=None):
+        """
+        Update the xapian db with the data in files.
+
+        If no files have been passed, add everything.
+        """
diff --git a/tests/data/empty/.gitignore b/tests/data/empty/.gitignore
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/tests/data/empty/.gitignore
diff --git a/tests/data/simple/schema.yaml b/tests/data/simple/schema.yaml
new file mode 100644
index 0000000..177aa92
--- /dev/null
+++ b/tests/data/simple/schema.yaml
@@ -0,0 +1,12 @@
+name: "Simple lesana collection"
+lang: 'english'
+fields:
+    - name: name
+      type: string
+      index: free
+    - name: description
+      type: text
+      index: free
+    - name: position
+      type: string
+      index: facet
diff --git a/tests/test_collection.py b/tests/test_collection.py
index 63820d5..64dd293 100644
--- a/tests/test_collection.py
+++ b/tests/test_collection.py
@@ -1,10 +1,22 @@
-from __future__ import division
-from __future__ import absolute_import
-from __future__ import print_function
-from __future__ import unicode_literals
-
+import os.path
+import shutil
 import unittest
 
-class testCollection(unittest.TestCase):
-    def test_noting(self):
-        pass
+import lesana
+
+
+class testCollectionLoading(unittest.TestCase):
+    def tearDown(self):
+        shutil.rmtree(os.path.join(self.collection.basedir, '.lesana'))
+
+    def test_empty(self):
+        self.collection = lesana.Collection('tests/data/empty')
+        self.assertIsNone(self.collection.schema)
+        self.assertIsNotNone(self.collection.cache)
+
+    def test_simple(self):
+        self.collection = lesana.Collection('tests/data/simple')
+        self.assertIsNotNone(self.collection.schema)
+        self.assertEqual(self.collection.schema['name'], "Simple lesana collection")
+        self.assertEqual(len(self.collection.schema['fields']), 3)
+        self.assertIsNotNone(self.collection.cache)