retoor
8bacd6aa3f
refactor: reorganize imports and improve error handling in application fix: correct typo in ip2location middleware chore: add author headers to source files
39 lines
1.0 KiB
Python
39 lines
1.0 KiB
Python
# retoor <retoor@molodetz.nl>
|
|
|
|
import json
|
|
|
|
from snek.system.service import BaseService
|
|
|
|
|
|
class UserPropertyService(BaseService):
|
|
mapper_name = "user_property"
|
|
|
|
async def set(self, user_uid, name, value):
|
|
self.mapper.db["user_property"].upsert(
|
|
{
|
|
"user_uid": user_uid,
|
|
"name": name,
|
|
"value": json.dumps(value, default=str),
|
|
},
|
|
["user_uid", "name"],
|
|
)
|
|
self.mapper.db.commit()
|
|
|
|
async def get(self, user_uid, name):
|
|
try:
|
|
return json.loads(
|
|
(await super().get(user_uid=user_uid, name=name))["value"]
|
|
)
|
|
except Exception as ex:
|
|
print(ex)
|
|
return None
|
|
|
|
async def search(self, query, **kwargs):
|
|
query = query.strip().lower()
|
|
if not query:
|
|
raise []
|
|
results = []
|
|
async for result in self.find(name={"ilike": "%" + query + "%"}, **kwargs):
|
|
results.append(result)
|
|
return results
|