Skip to content

Commit b0f4da8

Browse files
committed
make some ContextWebSocket private
1 parent 37c2a22 commit b0f4da8

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

template/server/messaging.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,8 @@ def __init__(self, in_background: bool = False):
4848
class ContextWebSocket:
4949
_ws: Optional[WebSocketClientProtocol] = None
5050
_receive_task: Optional[asyncio.Task] = None
51-
global_env_vars: Optional[Dict[StrictStr, str]] = None
52-
global_env_vars_set = False
51+
_global_env_vars: Optional[Dict[StrictStr, str]] = None
52+
_global_env_vars_set = False
5353
_cleanup_task: Optional[asyncio.Task] = None
5454

5555
def __init__(
@@ -165,9 +165,9 @@ def _reset_env_vars_code(self, env_vars: Dict[StrictStr, str]) -> str:
165165

166166
for key in env_vars:
167167
# Check if this var exists in global env vars
168-
if self.global_env_vars and key in self.global_env_vars:
168+
if self._global_env_vars and key in self._global_env_vars:
169169
# Reset to global value
170-
value = self.global_env_vars[key]
170+
value = self._global_env_vars[key]
171171
command = self._set_env_var_snippet(key, value)
172172
else:
173173
# Remove the variable
@@ -305,11 +305,11 @@ async def execute(
305305
# Build the complete code snippet with env vars
306306
complete_code = code
307307

308-
if not self.global_env_vars:
309-
self.global_env_vars = await get_envs()
308+
if not self._global_env_vars:
309+
self._global_env_vars = await get_envs()
310310

311-
if not self.global_env_vars_set and self.global_env_vars:
312-
env_setup_code = self._set_env_vars_code(self.global_env_vars)
311+
if not self._global_env_vars_set and self._global_env_vars:
312+
env_setup_code = self._set_env_vars_code(self._global_env_vars)
313313
if env_setup_code:
314314
indented_env_code = self._indent_code_with_level(env_setup_code, code_indent)
315315
complete_code = f"{indented_env_code}\n{complete_code}"

0 commit comments

Comments
 (0)