-
Notifications
You must be signed in to change notification settings - Fork 233
/
session.py
35 lines (24 loc) · 1.03 KB
/
session.py
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
import uuid
from optimus.server.functions import run, code, engine, session, create_session, delete_session, features
from .engine import Engine
defaultEngineConfiguration = { "engine": "dask" }
class Session:
def __init__(self, config=None, id=None, **kwargs):
self.config = config or kwargs
self.id = id or str(uuid.uuid4())
self._default_engine = None
if "engineConfiguration" not in self.config:
self.config.update({"engineConfiguration": defaultEngineConfiguration})
def engine(self, config=None, **kwargs):
config = config or kwargs or self.config["engineConfiguration"]
e = Engine(config, session=self)
if self._default_engine is None:
self._default_engine = e
return e
@property
def default_engine(self):
if self._default_engine is None:
return self.engine()
return self._default_engine
def request(self, body=None, **kwargs):
return self.default_engine.request(body, **kwargs)