Source code for bw2data.search.indices

# -*- coding: utf-8 -*-
from __future__ import print_function, unicode_literals
from eight import *

from .. import config, projects
from .schema import bw2_schema
from whoosh import index, query
import os
import shutil



[docs]
class IndexManager(object):
    def __init__(self, database_path, dir_name="whoosh"):

[docs]
        self.path = os.path.join(projects.request_directory("whoosh"), database_path)

        if not os.path.exists(self.path):
            os.mkdir(self.path)


[docs]
    def get(self):
        try:
            return index.open_dir(self.path)
        except index.EmptyIndexError:
            return self.create()



[docs]
    def create(self):
        self.delete_database()
        os.mkdir(self.path)
        return index.create_in(self.path, bw2_schema)



[docs]
    def _format_dataset(self, ds):
        fl = lambda o: o[1].lower() if isinstance(o, tuple) else o.lower()
        return dict(
            name=ds.get(u"name", u"").lower(),
            comment=ds.get(u"comment", u"").lower(),
            product=ds.get(u"reference product", u"").lower(),
            categories=u", ".join(ds.get(u"categories", [])).lower(),
            location=fl(ds.get(u"location", u"")),
            database=ds[u"database"],
            code=ds['code']
        )



[docs]
    def add_dataset(self, ds):
        self.add_datasets([ds])



[docs]
    def add_datasets(self, datasets):
        writer = self.get().writer()
        for ds in datasets:
            writer.add_document(**self._format_dataset(ds))
        writer.commit()



[docs]
    def update_dataset(self, ds):
        writer = self.get().writer()
        writer.update_document(**self._format_dataset(ds))
        writer.commit()



[docs]
    def delete_dataset(self, ds):
        index = self.get()
        index.delete_by_term("code", ds['code'])



[docs]
    def delete_database(self):
        shutil.rmtree(self.path)