52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
"""
|
|
Lexicon query interface
|
|
"""
|
|
|
|
import re
|
|
|
|
from sqlalchemy import select, func
|
|
|
|
from amanuensis.db import DbContext, Lexicon
|
|
from amanuensis.errors import ArgumentError
|
|
|
|
|
|
RE_ALPHANUM_DASH_UNDER = re.compile(r'^[A-Za-z0-9-_]*$')
|
|
|
|
|
|
def create(
|
|
db: DbContext,
|
|
name: str,
|
|
title: str,
|
|
prompt: str) -> Lexicon:
|
|
"""
|
|
Create a new lexicon.
|
|
"""
|
|
# Verify name
|
|
if not isinstance(name, str):
|
|
raise ArgumentError('Lexicon name must be a string')
|
|
if not name.strip():
|
|
raise ArgumentError('Lexicon name must not be blank')
|
|
if not RE_ALPHANUM_DASH_UNDER.match(name):
|
|
raise ArgumentError('Lexicon name may only contain alphanumerics, dash, and underscore')
|
|
|
|
# Verify title
|
|
if title is not None and not isinstance(name, str):
|
|
raise ArgumentError('Lexicon name must be a string')
|
|
|
|
# Verify prompt
|
|
if not isinstance(prompt, str):
|
|
raise ArgumentError('Lexicon prompt must be a string')
|
|
|
|
# Query the db to make sure the lexicon name isn't taken
|
|
if db.session.query(func.count(Lexicon.id)).filter(Lexicon.name == name).scalar() > 0:
|
|
raise ArgumentError('Lexicon name is already taken')
|
|
|
|
new_lexicon = Lexicon(
|
|
name=name,
|
|
title=title,
|
|
prompt=prompt,
|
|
)
|
|
db.session.add(new_lexicon)
|
|
db.session.commit()
|
|
return new_lexicon
|