From 9454870d5c34a778d839ab91ac082cc95d1ab0b0 Mon Sep 17 00:00:00 2001 From: wqy Date: Sun, 18 Sep 2022 00:38:25 +0800 Subject: [PATCH] Let client know project open count --- api/__init__.py | 2 +- api/project.py | 5 +++++ tjnetwork.py | 3 +++ 3 files changed, 9 insertions(+), 1 deletion(-) diff --git a/api/__init__.py b/api/__init__.py index b5a6b78..17556e1 100644 --- a/api/__init__.py +++ b/api/__init__.py @@ -1,5 +1,5 @@ from .project import have_project, create_project, delete_project -from .project import is_project_open, open_project, close_project +from .project import is_project_open, get_project_open_count, open_project, close_project from .project import copy_project from .change_set import ChangeSet diff --git a/api/project.py b/api/project.py index d1d9e60..30743aa 100644 --- a/api/project.py +++ b/api/project.py @@ -34,6 +34,11 @@ def open_project(name: str) -> None: def is_project_open(name: str) -> bool: return name in conn +def get_project_open_count(name: str) -> int: + if is_project_open(name): + return _project_open_count[name] + return 0 + def close_project(name: str) -> None: if name in _project_open_count: _project_open_count[name] -= 1 diff --git a/tjnetwork.py b/tjnetwork.py index f031b15..88d150e 100644 --- a/tjnetwork.py +++ b/tjnetwork.py @@ -50,6 +50,9 @@ def delete_project(name: str) -> None: def is_project_open(name: str) -> bool: return api.is_project_open(name) +def get_project_open_count(name: str) -> int: + return api.get_project_open_count(name) + def open_project(name: str) -> None: return api.open_project(name)