tmtt/redis-om-python
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Replace execute_command with specific redis functions when possible (#346)

Browse source 
* replace execute_command

* change to root_path
This commit is contained in:
dvora-h 2022-08-22 18:04:10 +03:00 • committed by GitHub
parent 60f102d188
commit 57608d85f4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 8 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

7
aredis_om/model/model.py
View file

@ -31,6 +31,7 @@ from pydantic.fields import ModelField, Undefined, UndefinedType
from pydantic.main import ModelMetaclass, validate_model
from pydantic.typing import NoArgAnyCallable
from pydantic.utils import Representation
from redis.commands.json.path import Path
from typing_extensions import Protocol, get_args, get_origin
from ulid import ULID
@ -1490,7 +1491,7 @@ class JsonModel(RedisModel, abc.ABC):
db = self._get_db(pipeline)
# TODO: Wrap response errors in a custom exception?
await db.execute_command("JSON.SET", self.key(), ".", self.json())
await db.json().set(self.key(), Path.root_path(), json.loads(self.json()))
return self
@classmethod
@ -1535,8 +1536,8 @@ class JsonModel(RedisModel, abc.ABC):
@classmethod
async def get(cls, pk: Any) -> "JsonModel":
document = await cls.db().execute_command("JSON.GET", cls.make_primary_key(pk))
if not document:
document = json.dumps(await cls.db().json().get(cls.make_key(pk)))
if document == "null":
raise NotFoundError
return cls.parse_raw(document)