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

Fixed linter errors, refactored delete.

Browse source
This commit is contained in:
Simon Prickett 2022-02-15 18:46:50 +00:00
parent 885b24b35b
commit 6d009a0d72
3 changed files with 7 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

14
aredis_om/model/model.py
View file

@ -1114,8 +1114,10 @@ class RedisModel(BaseModel, abc.ABC, metaclass=ModelMeta):
pk = getattr(self, self._meta.primary_key.field.name)
return self.make_primary_key(pk)
async def delete(self):
return await self.db().delete(self.key())
@classmethod
async def delete(cls, pk: Any) -> int:
"""Delete data at this key."""
return await cls.db().delete(cls.make_primary_key(pk))
@classmethod
async def get(cls, pk: Any) -> "RedisModel":
@ -1335,10 +1337,6 @@ class HashModel(RedisModel, abc.ABC):
result = cls.parse_obj(document)
return result
@classmethod
async def delete(cls, pk: Any) -> int:
return await cls.db().delete(cls.make_primary_key(pk))
@classmethod
@no_type_check
def _get_value(cls, *args, **kwargs) -> Any:
@ -1507,10 +1505,6 @@ class JsonModel(RedisModel, abc.ABC):
raise NotFoundError
return cls.parse_raw(document)
@classmethod
async def delete(cls, pk: Any) -> int:
return await cls.db().delete(cls.make_primary_key(pk))
@classmethod
def redisearch_schema(cls):
key_prefix = cls.make_key(cls._meta.primary_key_pattern.format(pk=""))