Skip to content

Feat/swanlab-logout #597

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 2 commits into from
Jun 1, 2024
Merged

Feat/swanlab-logout #597

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
53c7353
cli-swanlab-logout
Zeyi-Lin May 30, 2024
308b904
update confirm
Zeyi-Lin May 30, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
25 changes: 24 additions & 1 deletion swanlab/cli/main.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,10 +12,11 @@
from .utils import is_valid_ip, is_valid_port, is_valid_root_dir, URL
from swanlab.package import version_limit, get_package_version, is_login
from swanlab.api.auth import terminal_login
from ..env import get_server_host, get_server_port, get_swanlog_dir
from ..env import get_server_host, get_server_port, get_swanlog_dir, get_swanlab_folder
from ..db import connect
from ..utils import FONT
import time
import shutil


@click.group(invoke_without_command=True)
Expand Down Expand Up @@ -145,6 +146,28 @@ def login(api_key: str, relogin: bool, **kwargs):
print(FONT.swanlab("Login successfully. Hi, " + FONT.bold(FONT.default(login_info.username))) + "!")


# ---------------------------------- 退出登录命令 ----------------------------------
@cli.command()
def logout(**kwargs):
"""Logout to the swanlab cloud."""
command = FONT.bold("swanlab login")
if is_login():
# 如果已经是登录状态，那么则询问用户是否确认，如果确认则删除token文件夹
confirm = input(FONT.swanlab("Are you sure you want to logout? (y/N): "))
if confirm.lower() == "y":
try:
shutil.rmtree(get_swanlab_folder())
return print(FONT.swanlab("Logout successfully. You can use `" + command + "` to login again."))
except Exception as e:
return print(FONT.swanlab("Logout failed. Please check if you have file operation permissions."))
else:
return print(FONT.swanlab("Logout canceled."))

# 如果还未登录，则不做任何处理，并告知用户如何登录
tip = FONT.swanlab("You are not logged in. If you want to login in, please use `" + command + "` to login.")
return print(tip)


# ---------------------------------- 转换命令，用于转换其他实验跟踪工具 ----------------------------------
@cli.command()
@click.option(
Expand Down