mirror of
https://github.com/MISP/misp-galaxy.git
synced 2024-12-02 11:47:18 +00:00
29 lines
704 B
Python
29 lines
704 B
Python
|
from typing import TYPE_CHECKING
|
||
|
|
|
||
|
|
if TYPE_CHECKING:
|
||
|
|
from redis.asyncio.client import Pipeline, Redis
|
||
|
|
|
||
|
|
|
||
|
|
def from_url(url, **kwargs):
|
||
|
|
"""
|
||
|
|
Returns an active Redis client generated from the given database URL.
|
||
|
|
|
||
|
|
Will attempt to extract the database id from the path url fragment, if
|
||
|
|
none is provided.
|
||
|
|
"""
|
||
|
|
from redis.asyncio.client import Redis
|
||
|
|
|
||
|
|
return Redis.from_url(url, **kwargs)
|
||
|
|
|
||
|
|
|
||
|
|
class pipeline:
|
||
|
|
def __init__(self, redis_obj: "Redis"):
|
||
|
|
self.p: "Pipeline" = redis_obj.pipeline()
|
||
|
|
|
||
|
|
async def __aenter__(self) -> "Pipeline":
|
||
|
|
return self.p
|
||
|
|
|
||
|
|
async def __aexit__(self, exc_type, exc_value, traceback):
|
||
|
|
await self.p.execute()
|
||
|
|
del self.p
|