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

6
aredis_om/model/migrations/migrator.py
View file

@ -47,7 +47,7 @@ async def create_index(conn: redis.Redis, index_name, schema, current_hash):
f"You attempted to create an index in database {db_number}"
)
try:
await conn.execute_command(f"ft.info {index_name}")
await conn.ft(index_name).info()
except redis.ResponseError:
await conn.execute_command(f"ft.create {index_name} {schema}")
# TODO: remove "type: ignore" when type stubs will be fixed
@ -85,7 +85,7 @@ class IndexMigration:
async def drop(self):
try:
await self.conn.execute_command(f"FT.DROPINDEX {self.index_name}")
await self.conn.ft(self.index_name).dropindex()
except redis.ResponseError:
log.info("Index does not exist: %s", self.index_name)
@ -115,7 +115,7 @@ class Migrator:
current_hash = hashlib.sha1(schema.encode("utf-8")).hexdigest() # nosec
try:
await conn.execute_command("ft.info", cls.Meta.index_name)
await conn.ft(cls.Meta.index_name).info()
except redis.ResponseError:
self.migrations.append(
IndexMigration(