Newer
Older
Lluis Gifre Renom
committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
from redis.client import Redis
from .tools._Entity import _Entity
from .tools.EntityAttributes import EntityAttributes
from .tools.EntityCollection import EntityCollection
from .tools.Mutex import Mutex
from .Keys import KEY_CONTEXT, KEY_TOPOLOGIES
from .Topology import Topology
VALIDATORS = {}
class Context(_Entity):
def __init__(self, context_uuid : str, redis_client: Redis):
self.__redis_client = redis_client
self.__mutex = Mutex(self.__redis_client)
self.__acquired = False
self.__owner_key = None
self.__entity_key = KEY_CONTEXT.format(context_uuid=context_uuid)
super().__init__(context_uuid, self)
self.context_uuid = self.get_uuid()
self.attributes = EntityAttributes(self, KEY_CONTEXT, validators=VALIDATORS)
self.topologies = EntityCollection(self, KEY_TOPOLOGIES)
def get_parent(self): return(self)
def get_context(self): return(self)
def get_redis_client(self): return(self.__redis_client)
def __enter__(self):
self.__acquired,self.__owner_key = self.__mutex.acquire(self.__entity_key, owner_key=self.__owner_key)
if not self.__acquired: raise Exception('Unable to acquire {}'.format(self.__entity_key))
return self
def __exit__(self, exc_type, exc_val, exc_tb):
if not self.__acquired: return
self.__mutex.release(self.__entity_key, self.__owner_key)
def topology(self, topology_uuid : str) -> Topology:
return Topology(topology_uuid, self)