def get_collection(self, name=''):
"""Return Collection"""
if self.has_collection(name):
self.collection = self.database.collection(name)
return self.collection
else:
msg = coll_err.get(1228).format(name)
raise gmap_exceptions.CollectionNotExist(msg)
def search_in_collection(self, collection, search, page=1, per_page=10):
"""Search Document"""
try:
per_page = per_page if per_page <= MAX_PER_PAGE else MAX_PER_PAGE
offset = (page - 1) * per_page
where, bind_vars = self.prepare_search(search)
bind_vars['@collection'] = collection
bind_vars['offset'] = offset
bind_vars['count'] = per_page
full_query = 'FOR doc IN @@collection {} ' \
'LIMIT @offset, @count RETURN doc'.format(where)
LOGGER.debug('Full Query: %s' % full_query)
cursor = self.database.aql.execute(
full_query,
bind_vars=bind_vars,
count=True,
full_count=True,
batch_size=1,
ttl=10,
optimizer_rules=['+all']
)
return cursor
except exceptions.AQLQueryExecuteError as err:
if err.error_code == 1203:
msg = db_err.get(1203).format(collection)
LOGGER.error(msg)
raise gmap_exceptions.CollectionNotExist(msg)
else:
msg = db_err.get(1).format(err.message)
LOGGER.error(msg)
raise gmap_exceptions.DatabaseException(msg)
except Exception as err:
msg = db_err.get(1).format(str(err))
LOGGER.error(msg)
raise gmap_exceptions.DatabaseException(msg)
def delete_collection(self, name=''):
"""Delete Collection """
try:
self.database.delete_collection(name=name)
self.collection = None
return True
except exceptions.CollectionDeleteError as err:
if err.error_code == 1203:
msg = coll_err.get(1228).format(name)
raise gmap_exceptions.CollectionNotExist(msg)
else:
msg = coll_err.get(0).format(name, err.message)
raise gmap_exceptions.CollectionException(msg)
except Exception as err:
msg = coll_err.get(0).format(name, err.message)
raise gmap_exceptions.CollectionException(msg)
def clear_collection(self, collection, search):
"""Clear colection by query"""
try:
where, bind_vars = self.prepare_search(search)
bind_vars['@collection'] = collection
full_query = 'FOR doc IN @@collection {} ' \
'REMOVE doc._key in @@collection'.format(where)
LOGGER.debug('Full Query: %s' % full_query)
cursor = self.database.aql.execute(
full_query,
bind_vars=bind_vars,
count=True,
full_count=True,
batch_size=1,
ttl=10,
optimizer_rules=['+all']
)
return cursor
except exceptions.AQLQueryExecuteError as err:
if err.error_code == 1203:
msg = db_err.get(1203).format(collection)
LOGGER.error(msg)
raise gmap_exceptions.CollectionNotExist(msg)
else:
msg = db_err.get(1).format(err.message)
LOGGER.error(msg)
raise gmap_exceptions.DatabaseException(msg)
except Exception as err:
msg = db_err.get(1).format(str(err))
LOGGER.error(msg)
raise gmap_exceptions.DatabaseException(msg)
def search_in_collections(self, collections, search, page=1, per_page=10):
"""Search Document"""
try:
offset = (page - 1) * per_page
per_page = per_page if per_page <= 100 else 100
union = [
'a', 'b', 'c', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm'
]
search_name = search[0]
partial_query = []
for index_name, name in enumerate(search_name):
where, bind_vars = self.prepare_search([[name]])
bind_vars['offset'] = offset
bind_vars['count'] = per_page
queries = []
for index_coll, collection in enumerate(collections):
idx = '@cl_{}'.format(index_coll)
bind_vars[idx] = collection
query = 'FOR doc IN @{} {} RETURN doc'.format(idx, where)
queries.append(query)
colls = '({})'.format(','.join(queries))
if len(queries) > 1:
colls = 'UNION{}'.format(colls)
union_item = union[index_name]
partial_query.append('FOR {} IN {} ' \
'SORT {}.name LIMIT @offset, @count RETURN {}'.format(
union_item,
colls,
union_item,
union_item
)
)
if len(partial_query) > 1:
full_query = 'FOR full IN UNION({}) RETURN full'.format(
','.join(partial_query))
else:
full_query = partial_query[0]
LOGGER.debug('Full Query: %s' % full_query)
cursor = self.database.aql.execute(full_query,
bind_vars=bind_vars,
count=True,
full_count=True,
ttl=10,
optimizer_rules=['+all'])
return cursor
except exceptions.AQLQueryExecuteError as err:
if err.error_code == 1203:
msg = db_err.get(1203).format(collection)
LOGGER.error(msg)
raise gmap_exceptions.CollectionNotExist(msg)
else:
msg = db_err.get(1).format(err.message)
LOGGER.error(msg)
raise gmap_exceptions.DatabaseException(msg)
except Exception as err:
msg = db_err.get(1).format(str(err))
LOGGER.error(msg)
raise gmap_exceptions.DatabaseException(msg)
