tmp
This commit is contained in:
parent
29a73c3c47
commit
c379ee7d8f
|
@ -0,0 +1,64 @@
|
|||
"""
|
||||
Index query interface
|
||||
"""
|
||||
|
||||
import re
|
||||
|
||||
from amanuensis.db import DbContext, ArticleIndex, IndexType
|
||||
from amanuensis.errors import ArgumentError
|
||||
|
||||
|
||||
def create(
|
||||
db: DbContext,
|
||||
lexicon_id: int,
|
||||
index_type: IndexType,
|
||||
pattern: str,
|
||||
logical_order: int,
|
||||
display_order: int,
|
||||
capacity: int) -> ArticleIndex:
|
||||
"""
|
||||
Create a new index in a lexicon.
|
||||
"""
|
||||
# Verify argument types are correct
|
||||
if not isinstance(lexicon_id, int):
|
||||
raise ArgumentError('lexicon_id')
|
||||
if not isinstance(index_type, IndexType):
|
||||
raise ArgumentError('index_type')
|
||||
if not isinstance(pattern, str):
|
||||
raise ArgumentError('pattern')
|
||||
if not isinstance(logical_order, int):
|
||||
raise ArgumentError('logical_order')
|
||||
if not isinstance(display_order, int):
|
||||
raise ArgumentError('display_order')
|
||||
if not isinstance(capacity, int):
|
||||
raise ArgumentError('capacity')
|
||||
|
||||
# Verify the pattern is valid for the index type:
|
||||
if index_type == IndexType.CHAR:
|
||||
if len(pattern) < 1:
|
||||
raise ArgumentError(f'Pattern "{pattern}" too short for index type {index_type}')
|
||||
elif index_type == IndexType.RANGE:
|
||||
range_def = re.match(r'^(.)-(.)$', pattern)
|
||||
if not range_def:
|
||||
raise ArgumentError(f'Pattern "{pattern}" is not a valid range format')
|
||||
start_char, end_char = range_def.group(1), range_def.group(2)
|
||||
if start_char >= end_char:
|
||||
raise ArgumentError(f'Range start "{start_char}" is not before range end "{end_char}"')
|
||||
elif index_type == IndexType.PREFIX:
|
||||
if len(pattern) < 1:
|
||||
raise ArgumentError(f'Pattern "{pattern}" too short for index type {index_type}')
|
||||
elif index_type == IndexType.ETC:
|
||||
if len(pattern) < 1:
|
||||
raise ArgumentError(f'Pattern "{pattern}" too short for index type {index_type}')
|
||||
|
||||
new_index = ArticleIndex(
|
||||
lexicon_id=lexicon_id,
|
||||
index_type=index_type,
|
||||
pattern=pattern,
|
||||
logical_order=logical_order,
|
||||
display_order=display_order,
|
||||
capacity=capacity,
|
||||
)
|
||||
db.session.add(new_index)
|
||||
db.session.commit()
|
||||
return new_index
|
Loading…
Reference in New Issue