# -*- coding: utf-8 -*-

class ArangoDb(object):
    def __init__(self, database):
        super(ArangoDb, self).__init__()
        self.database = database

    def get_or_create_collection(self, collection_name):
        if self.database.hasCollection(collection_name):
            return self.database[collection_name]
        else:
            return self.database.createCollection(name=collection_name)

    def AQLQuery(self, query, bind_vars={}):
        return self.database.AQLQuery(query, bindVars=bind_vars)