This commit is contained in:
wangmeihua 2025-08-27 15:06:45 +08:00
parent ed50690920
commit ea1a9a084f
2 changed files with 18 additions and 11 deletions

View File

@ -3,31 +3,37 @@ db = DBPools()
dbname = get_module_dbname('rag')
async with db.sqlorContext(dbname) as sor:
orgid = await get_userorgid()
sql="select * from ragquote where orgid = ${orgid}$ order by enabled_date"
programs = await sor.R('ragprogram', {'id': params_kw.selected_program})
if len(programs) < 1:
e = Exception(f'selected program({params_kw.selected_program}) not exists')
exception(f'{e}')
raise e
program = programs[0]
sql="select * from ragquota where orgid = ${orgid}$ order by enabled_date"
qs = await sor.sqlExe(sql, {'orgid': orgid})
today = curDateString()
expired_date = strdate_add(today, months=program.term)
if len(qs) == 0:
if params_kw.program_type == 'free':
expired_date = dateadd(today, months=1)
else:
expired_date = '9999-12-31'
ns = {
"id": uuid(),
"orgid": u.userorgid,
"orgid": orgid,
"enabled_date": today,
"expired_date": expired_date,
'quota': params_kw.quota
'quota': program.quota
}
await sor.C('ragquota', ns)
else:
lastq = qs[-1]
lastq.expired_date = today
await sor.U('ragquote', lastq.copy())
await sor.U('ragquota', lastq.copy())
ns = {
"id": uuid(),
"orgid": u.userorgid,
"orgid": orgid,
"enabled_date": today,
"expired_date": expired_date,
'quota': params_kw.quota
'quota': program.quota
}
await sor.C('ragquota', ns)
return UiMessage(title="set quota", message="set quota success")
e = db.e_except
return UiError(title="set quota", message=f'set quota error:{e}')

View File

@ -112,7 +112,8 @@
},
"options":{
"params":{
"selected_program":"{{p.id}}"
"selected_program":"{{p.id}}",
"quota":{{p.quota}}
},
"url":"{{entire_url('./program_selected.dspy')}}"
}