bugfix
This commit is contained in:
parent
ed50690920
commit
ea1a9a084f
@ -3,31 +3,37 @@ db = DBPools()
|
|||||||
dbname = get_module_dbname('rag')
|
dbname = get_module_dbname('rag')
|
||||||
async with db.sqlorContext(dbname) as sor:
|
async with db.sqlorContext(dbname) as sor:
|
||||||
orgid = await get_userorgid()
|
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})
|
qs = await sor.sqlExe(sql, {'orgid': orgid})
|
||||||
today = curDateString()
|
today = curDateString()
|
||||||
|
expired_date = strdate_add(today, months=program.term)
|
||||||
if len(qs) == 0:
|
if len(qs) == 0:
|
||||||
if params_kw.program_type == 'free':
|
|
||||||
expired_date = dateadd(today, months=1)
|
|
||||||
else:
|
|
||||||
expired_date = '9999-12-31'
|
|
||||||
ns = {
|
ns = {
|
||||||
"id": uuid(),
|
"id": uuid(),
|
||||||
"orgid": u.userorgid,
|
"orgid": orgid,
|
||||||
"enabled_date": today,
|
"enabled_date": today,
|
||||||
"expired_date": expired_date,
|
"expired_date": expired_date,
|
||||||
'quota': params_kw.quota
|
'quota': program.quota
|
||||||
}
|
}
|
||||||
await sor.C('ragquota', ns)
|
await sor.C('ragquota', ns)
|
||||||
else:
|
else:
|
||||||
lastq = qs[-1]
|
lastq = qs[-1]
|
||||||
lastq.expired_date = today
|
lastq.expired_date = today
|
||||||
await sor.U('ragquote', lastq.copy())
|
await sor.U('ragquota', lastq.copy())
|
||||||
ns = {
|
ns = {
|
||||||
"id": uuid(),
|
"id": uuid(),
|
||||||
"orgid": u.userorgid,
|
"orgid": orgid,
|
||||||
"enabled_date": today,
|
"enabled_date": today,
|
||||||
"expired_date": expired_date,
|
"expired_date": expired_date,
|
||||||
'quota': params_kw.quota
|
'quota': program.quota
|
||||||
}
|
}
|
||||||
await sor.C('ragquota', ns)
|
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}')
|
||||||
|
|||||||
@ -112,7 +112,8 @@
|
|||||||
},
|
},
|
||||||
"options":{
|
"options":{
|
||||||
"params":{
|
"params":{
|
||||||
"selected_program":"{{p.id}}"
|
"selected_program":"{{p.id}}",
|
||||||
|
"quota":{{p.quota}}
|
||||||
},
|
},
|
||||||
"url":"{{entire_url('./program_selected.dspy')}}"
|
"url":"{{entire_url('./program_selected.dspy')}}"
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user