lockbot 2.9.0__tar.gz → 2.9.2__tar.gz

{lockbot-2.9.0/python/lockbot.egg-info → lockbot-2.9.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lockbot
-Version: 2.9.0
+Version: 2.9.2
 Summary: Cluster resource management bot for IM platforms
 Author-email: Jianbang Yang <yangjianbang112@gmail.com>
 License: MIT
@@ -53,6 +53,8 @@ Supports both standalone Flask deployment and a full platform mode with FastAPI
 [![PyPI version](https://img.shields.io/pypi/v/lockbot?color=blue)](https://pypi.org/project/lockbot/)
 [![Docker Image](https://img.shields.io/badge/ghcr.io-dynamicheart%2Flockbot-blue?logo=docker)](https://github.com/DynamicHeart/lockbot/pkgs/container/lockbot)
 
+<img src="./docs/demo.gif" width="600" />
+
 ## Features
 
 - **Device Lock Bot** — Lock/unlock individual GPUs or devices on a cluster

{lockbot-2.9.0 → lockbot-2.9.2}/README.md

@@ -10,6 +10,8 @@ Supports both standalone Flask deployment and a full platform mode with FastAPI
 [![PyPI version](https://img.shields.io/pypi/v/lockbot?color=blue)](https://pypi.org/project/lockbot/)
 [![Docker Image](https://img.shields.io/badge/ghcr.io-dynamicheart%2Flockbot-blue?logo=docker)](https://github.com/DynamicHeart/lockbot/pkgs/container/lockbot)
 
+<img src="./docs/demo.gif" width="600" />
+
 ## Features
 
 - **Device Lock Bot** — Lock/unlock individual GPUs or devices on a cluster

{lockbot-2.9.0 → lockbot-2.9.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "lockbot"
-version = "2.9.0"
+version = "2.9.2"
 description = "Cluster resource management bot for IM platforms"
 readme = "README.md"
 license = {text = "MIT"}

{lockbot-2.9.0 → lockbot-2.9.2}/python/lockbot/backend/app/bots/schemas.py

@@ -68,6 +68,10 @@ def _validate_config_overrides(v: dict | None) -> dict | None:
     if show_alias is not None and not isinstance(show_alias, bool):
         errors.append("SHOW_NODE_ALIAS must be a boolean")
 
+    allow_alias = v.get("ALLOW_USER_ALIAS")
+    if allow_alias is not None and not isinstance(allow_alias, bool):
+        errors.append("ALLOW_USER_ALIAS must be a boolean")
+
     max_dur = v.get("MAX_LOCK_DURATION")
     default_dur = v.get("DEFAULT_DURATION")
     if isinstance(max_dur, int) and isinstance(default_dur, int) and max_dur != -1 and default_dur > max_dur:

{lockbot-2.9.0 → lockbot-2.9.2}/python/lockbot/core/base_bot.py

@@ -133,6 +133,74 @@ class BaseLockBot:
         alias = aliases.get(node_key)
         return f"{node_key}「{alias}」" if alias else node_key
 
+    def alias(self, user_id, command):
+        """Set or clear a node alias. Usage: alias <node_key> [alias_value]"""
+        if not self.config.get_val("ALLOW_USER_ALIAS"):
+            return self.show_error(user_id, t("alias.disabled", config=self.config))
+
+        import re
+
+        m = re.match(r"^\s*alias\s+([\w\d_-]+)(?:\s+(.{1,15}))?\s*$", command)
+        if not m:
+            return self.show_error(user_id, t("alias.usage", config=self.config))
+
+        node_key = m.group(1)
+        alias_val = (m.group(2) or "").strip()
+
+        cluster_configs = self.config.get_val("CLUSTER_CONFIGS")
+        if node_key not in cluster_configs:
+            return self.show_error(
+                user_id,
+                t(
+                    "alias.invalid_node",
+                    config=self.config,
+                    node_key=node_key,
+                    valid_keys=str(list(cluster_configs.keys())),
+                ),
+            )
+
+        aliases = self.config.get_val("NODE_ALIASES") or {}
+        if alias_val:
+            aliases[node_key] = alias_val
+        else:
+            aliases.pop(node_key, None)
+        self.config.set_val("NODE_ALIASES", aliases)
+
+        # Persist to DB if running under platform mode
+        self._persist_config_override("NODE_ALIASES", aliases)
+
+        if alias_val:
+            self.logger.info("user=%s set alias %s -> %s", user_id, node_key, alias_val)
+            msg = t("alias.set_ok", config=self.config, node_key=node_key, alias=alias_val)
+        else:
+            self.logger.info("user=%s cleared alias %s", user_id, node_key)
+            msg = t("alias.clear_ok", config=self.config, node_key=node_key)
+        return self.adapter.build_reply(msg, [user_id])
+
+    def _persist_config_override(self, key, value):
+        """Persist a single config_override key to DB (best-effort)."""
+        try:
+            import json
+
+            from lockbot.backend.app.bots.models import Bot
+            from lockbot.backend.app.database import SessionLocal
+
+            bot_id = self.config.get_val("BOT_ID")
+            if not bot_id:
+                return
+            db = SessionLocal()
+            try:
+                bot = db.query(Bot).filter(Bot.id == int(bot_id)).first()
+                if bot:
+                    overrides = json.loads(bot.config_overrides) if bot.config_overrides else {}
+                    overrides[key] = value
+                    bot.config_overrides = json.dumps(overrides, ensure_ascii=False)
+                    db.commit()
+            finally:
+                db.close()
+        except Exception:
+            pass
+
     def _whitelist_hint(self):
         """Return whitelist hint string if BOT_OWNER is configured, else empty string."""
         owner = self.config.get_val("BOT_OWNER")

{lockbot-2.9.0 → lockbot-2.9.2}/python/lockbot/core/device_bot.py

@@ -39,7 +39,7 @@ class DeviceBot(BaseLockBot):
         _loader = staticmethod(create_or_load_device_state)
 
     def supported_commands(self):
-        return ["lock", "slock", "unlock", "free", "kickout", "help", "h", "query"]
+        return ["lock", "slock", "unlock", "free", "kickout", "help", "h", "query", "alias"]
 
     def parse_command(self, user_id, command_key, command, parsing_duration=False):
         """
@@ -493,6 +493,9 @@ class DeviceBot(BaseLockBot):
         text += t("help.query_node_example", config=self.config, node=example_node)
         text += "\n"
 
+        if self.config.get_val("ALLOW_USER_ALIAS"):
+            text += f"6. 备注: alias {example_node} <备注>\n\n"
+
         return text
 
     def _check_and_notify(self) -> float | None:

{lockbot-2.9.0 → lockbot-2.9.2}/python/lockbot/core/handler.py

@@ -55,6 +55,8 @@ def execute_command(msg_data, bot):
             return bot.query(user_id)
     elif cmd in ("help", "h"):
         return bot.print_help(user_id)
+    elif cmd == "alias":
+        return bot.alias(user_id, rcv_info)
     else:
         return bot.print_help(user_id, t("error.unknown_error", config=getattr(bot, "config", None), command=rcv_info))
         return bot.print_help(user_id, t("error.unknown_error", config=getattr(bot, "config", None), command=rcv_info))

{lockbot-2.9.0 → lockbot-2.9.2}/python/lockbot/core/i18n/en.py

@@ -173,6 +173,12 @@ MESSAGES = {
     "opkick.invalid_format": "Invalid opkick format. Usage: opkick <username> [node] [dev x,y]",
     "opkick.notify_kicked": "⚠️ [Operator Release]\n\nNode: {node_key}\n",
     "opkick.notify_kicked_reason": "Reason: {reason}\n\n",
+    # ── Alias command ──
+    "alias.usage": "Usage: alias <node_key> [label]\nExample: alias node0 10.10.10.56\nClear: alias node0",
+    "alias.invalid_node": "Node {node_key} does not exist. Available: {valid_keys}\n",
+    "alias.set_ok": "✅ Set {node_key} alias to「{alias}」\n\n",
+    "alias.clear_ok": "✅ Cleared alias for {node_key}\n\n",
+    "alias.disabled": "User alias command is not enabled for this bot",
     # ── Webhook: bot not running ──
     "webhook.bot_not_running": "⚠️ Bot {bot_name}(ID:{bot_id}) is not running. "  # noqa: E501
     "Please contact the owner @{owner_username} to start it.",

{lockbot-2.9.0 → lockbot-2.9.2}/python/lockbot/core/i18n/zh.py

@@ -166,6 +166,12 @@ MESSAGES = {
     "opkick.invalid_format": "opkick命令格式错误，正确格式: opkick <username> [node] [dev x,y]",
     "opkick.notify_kicked": "⚠️【运维释放】\n\n节点: {node_key}\n",
     "opkick.notify_kicked_reason": "原因: {reason}\n\n",
+    # ── Alias command ──
+    "alias.usage": "用法: alias <节点名> [备注]\n示例: alias node0 10.10.10.56\n清除: alias node0",
+    "alias.invalid_node": "节点 {node_key} 不存在，可用节点: {valid_keys}\n",
+    "alias.set_ok": "✅ 已设置 {node_key} 备注为「{alias}」\n\n",
+    "alias.clear_ok": "✅ 已清除 {node_key} 的备注\n\n",
+    "alias.disabled": "该机器人未开启用户自助设置备注功能",
     # ── Webhook: bot not running ──
     "webhook.bot_not_running": (
         "⚠️ 机器人 {bot_name}(ID:{bot_id}) 尚未启动，请联系管理人 @{owner_username} 启动后再使用。"

{lockbot-2.9.0 → lockbot-2.9.2}/python/lockbot/core/io.py

@@ -154,6 +154,8 @@ def create_or_load_device_state(config=None):
                     clamped = apply_max_duration_limit(device_status["current_users"], max_duration, duration_whitelist)
                     clamped_user_ids.update(clamped)
                     default_device.update(**device_status)
+                    # Always use config's dev_model as source of truth
+                    default_device["dev_model"] = cluster_configs[node_key][dev_id]
 
         if clamped_user_ids:
             save_bot_state_to_file(default_state, config)

{lockbot-2.9.0 → lockbot-2.9.2}/python/lockbot/core/node_bot.py

@@ -37,7 +37,7 @@ class NodeBot(BaseLockBot):
         _loader = staticmethod(create_or_load_node_state)
 
     def supported_commands(self):
-        return ["lock", "slock", "unlock", "free", "kickout", "help", "h", "query"]
+        return ["lock", "slock", "unlock", "free", "kickout", "help", "h", "query", "alias"]
 
     def parse_command(self, user_id, command_key, command, parsing_duration=False):
         """
@@ -358,7 +358,12 @@ class NodeBot(BaseLockBot):
         reply_info += t("help.section4_title", config=self.config)
         reply_info += t("help.section5_title", config=self.config)
         reply_info += t("help.query_at_bot", config=self.config)
-        reply_info += f"    {example_node0}\n\n"
+        reply_info += f"    {example_node0}\n"
+
+        if self.config.get_val("ALLOW_USER_ALIAS"):
+            reply_info += f"6. 备注: alias {example_node0} <备注>\n"
+
+        reply_info += "\n"
         return reply_info
 
     def _check_and_notify(self) -> float | None:

{lockbot-2.9.0 → lockbot-2.9.2}/python/lockbot/core/queue_bot.py

@@ -26,7 +26,7 @@ class QueueBot(NodeBot):
     """
 
     def supported_commands(self):
-        return ["lock", "unlock", "free", "kickout", "kicklock", "help", "h", "book", "take", "query"]
+        return ["lock", "unlock", "free", "kickout", "kicklock", "help", "h", "book", "take", "query", "alias"]
 
     def lock(self, user_id, command):
         """
@@ -345,7 +345,12 @@ class QueueBot(NodeBot):
         parts.append(t("help.section_help_title_queue", config=self.config))
         parts.append(t("help.section_query_title_queue", config=self.config))
         parts.append(t("help.query_at_bot", config=self.config))
-        parts.append(f"    {example_node0}\n\n")
+        parts.append(f"    {example_node0}\n")
+
+        if self.config.get_val("ALLOW_USER_ALIAS"):
+            parts.append(f"8. 备注: alias {example_node0} <备注>\n")
+
+        parts.append("\n")
         return "".join(parts)
 
     def _check_and_notify(self) -> float | None:
@@ -522,11 +527,12 @@ class QueueBot(NodeBot):
             usage_info = ""
             for node_key, node_status in nodes.items():
                 if node_filter is None or node_key == node_filter:
+                    display_name = self._display_node(node_key)
                     if node_status["status"] == "idle":
-                        usage_info += "{:} {}\n".format(node_key, t("status.idle", config=self.config))
+                        usage_info += "{:} {}\n".format(display_name, t("status.idle", config=self.config))
                     else:
                         for user_idx, user_info in enumerate(node_status["current_users"]):
-                            node_name = f"{node_key}" if user_idx == 0 else ""
+                            node_name = f"{display_name}" if user_idx == 0 else ""
                             uid = user_info["user_id"]
                             duration = format_duration(
                                 remaining_duration(user_info["start_time"], user_info["duration"]),

{lockbot-2.9.0 → lockbot-2.9.2/python/lockbot.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lockbot
-Version: 2.9.0
+Version: 2.9.2
 Summary: Cluster resource management bot for IM platforms
 Author-email: Jianbang Yang <yangjianbang112@gmail.com>
 License: MIT
@@ -53,6 +53,8 @@ Supports both standalone Flask deployment and a full platform mode with FastAPI
 [![PyPI version](https://img.shields.io/pypi/v/lockbot?color=blue)](https://pypi.org/project/lockbot/)
 [![Docker Image](https://img.shields.io/badge/ghcr.io-dynamicheart%2Flockbot-blue?logo=docker)](https://github.com/DynamicHeart/lockbot/pkgs/container/lockbot)
 
+<img src="./docs/demo.gif" width="600" />
+
 ## Features
 
 - **Device Lock Bot** — Lock/unlock individual GPUs or devices on a cluster