26 lines
719 B
Python
26 lines
719 B
Python
from mods.db import db, Catalog, CatalogIndex
|
|
import re
|
|
|
|
def add_catalog(name):
|
|
with db.atomic():
|
|
catalog = Catalog.create(name=name)
|
|
CatalogIndex.insert(
|
|
{
|
|
CatalogIndex.rowid: catalog.id,
|
|
CatalogIndex.text: name
|
|
}
|
|
).execute()
|
|
return catalog
|
|
|
|
def search_catalogs(q):
|
|
q = re.sub(r'\s+', ' ', q).strip()
|
|
q = ' '.join([ f'{x}*' for x in q.split(' ')])
|
|
with db.atomic():
|
|
return (Catalog.select()
|
|
.join(
|
|
CatalogIndex,
|
|
on=(Catalog.id == CatalogIndex.rowid))
|
|
.where(CatalogIndex.match(q))
|
|
.order_by(CatalogIndex.bm25()))
|
|
|