badnet 0.1.0__py3-none-any.whl

badnet-0.1.0.dist-info/METADATA

@@ -0,0 +1,4 @@
+Metadata-Version: 2.4
+Name: badnet
+Version: 0.1.0
+Requires-Dist: openai

badnet-0.1.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+badnet.py,sha256=5NKG8wDY6CsyZtrJ8YtlfPN3IaXxvjs5m9KV7mrBdz8,7832
+badnet-0.1.0.dist-info/METADATA,sha256=wKdjrqFRpagA7omPm72ExaQoAlQeJm7KEDQ-6YoRthg,72
+badnet-0.1.0.dist-info/WHEEL,sha256=mOdXR9C3wMVov7cSY8PY-XCzo3SmmWCHvGQr5csJoao,91
+badnet-0.1.0.dist-info/entry_points.txt,sha256=2ZqpIHF9esjp0QN63FU2d3edU6gV8cU1u_iWHZmrjak,39
+badnet-0.1.0.dist-info/top_level.txt,sha256=ymTDn4a8M8Owgc-s6xxWVLg0VIyjk4d6C5mWjxxl8RU,7
+badnet-0.1.0.dist-info/RECORD,,

badnet-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (84.2.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

badnet-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+badnet = badnet:main

badnet-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+badnet

badnet.py

@@ -0,0 +1,210 @@
+#!/usr/bin/env python3
+"""badnet - консольный ассистент DeepSeek с доступом к терминалу."""
+import json
+import os
+import subprocess
+from pathlib import Path
+
+from openai import OpenAI
+
+CONFIG_DIR = Path.home() / ".badnet"
+CONFIG_FILE = CONFIG_DIR / "config.json"
+BASE_URL = "https://api.deepseek.com"
+DEFAULT_MODEL = "deepseek-chat"
+MODELS = ["deepseek-chat", "deepseek-reasoner"]
+MAX_OUTPUT = 8000
+
+SYSTEM_PROMPT = (
+    "Ты badnet, консольный ассистент с доступом к терминалу пользователя. "
+    "Когда нужно что-то проверить или сделать, вызывай инструмент run_shell. "
+    "Отвечай по-русски, кратко и по делу."
+)
+
+TOOLS = [{
+    "type": "function",
+    "function": {
+        "name": "run_shell",
+        "description": "Выполнить shell-команду в терминале и вернуть вывод",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "command": {"type": "string", "description": "Команда для выполнения"}
+            },
+            "required": ["command"],
+        },
+    },
+}]
+
+# Буквы из блоков: каждая строка одной буквы имеет одинаковую ширину (5 символов)
+LETTERS = {
+    "B": ["████ ", "█   █", "████ ", "█   █", "████ "],
+    "A": [" ███ ", "█   █", "█████", "█   █", "█   █"],
+    "D": ["████ ", "█   █", "█   █", "█   █", "████ "],
+    "N": ["█   █", "██  █", "█ █ █", "█  ██", "█   █"],
+    "E": ["█████", "█    ", "████ ", "█    ", "█████"],
+    "T": ["█████", "  █  ", "  █  ", "  █  ", "  █  "],
+}
+
+
+def print_banner():
+    word = "BADNET"
+    for row in range(5):
+        print("  " + "  ".join(LETTERS[ch][row] for ch in word))
+    print("  консольный ассистент DeepSeek  |  /help - список команд\n")
+
+
+def load_config():
+    if CONFIG_FILE.exists():
+        return json.loads(CONFIG_FILE.read_text(encoding="utf-8"))
+    return {}
+
+
+def save_config(cfg):
+    CONFIG_DIR.mkdir(parents=True, exist_ok=True)
+    CONFIG_FILE.write_text(json.dumps(cfg, ensure_ascii=False, indent=2), encoding="utf-8")
+
+
+def run_shell(command):
+    try:
+        p = subprocess.run(
+            command, shell=True, capture_output=True, text=True,
+            timeout=120, cwd=os.getcwd(),
+        )
+        out = (p.stdout + p.stderr).strip() or "(нет вывода)"
+    except subprocess.TimeoutExpired:
+        return "Ошибка: команда превысила таймаут 120 секунд"
+    if len(out) > MAX_OUTPUT:
+        out = out[:MAX_OUTPUT] + "\n...[вывод обрезан]"
+    print(out)
+    return f"exit={p.returncode}\n{out}"
+
+
+def ask_model(client, cfg, messages, state):
+    while True:
+        resp = client.chat.completions.create(
+            model=cfg["model"], messages=messages, tools=TOOLS
+        )
+        msg = resp.choices[0].message
+        messages.append(msg.model_dump(exclude_none=True))
+
+        if not msg.tool_calls:
+            print(msg.content or "")
+            return
+
+        if msg.content:
+            print(msg.content)
+
+        for call in msg.tool_calls:
+            args = json.loads(call.function.arguments or "{}")
+            command = args.get("command", "")
+            print(f"\n$ {command}")
+
+            if state["auto"]:
+                result = run_shell(command)
+            else:
+                answer = input("Выполнить? [y - да / a - да и всегда / n - нет]: ").strip().lower()
+                if answer in ("a", "а"):
+                    state["auto"] = True
+                    print("Автовыполнение включено (/auto - выключить)")
+                    answer = "y"
+                if answer in ("y", "д", "да", "yes"):
+                    result = run_shell(command)
+                else:
+                    result = "Пользователь отклонил выполнение команды."
+                    print(result)
+
+            messages.append({"role": "tool", "tool_call_id": call.id, "content": result})
+
+
+HELP = """Команды:
+  /put КЛЮЧ      сохранить API-ключ DeepSeek
+  /mod [МОДЕЛЬ]  выбрать модель (без аргумента - показать список)
+  /clear         очистить контекст диалога
+  /run КОМАНДА   выполнить команду в терминале без участия модели
+  /cd ПУТЬ       сменить рабочую папку
+  /cwd           показать текущую папку
+  /auto          вкл/выкл автовыполнение команд модели
+  /help          эта справка
+  /exit          выход"""
+
+
+def main():
+    cfg = load_config()
+    cfg.setdefault("model", DEFAULT_MODEL)
+    client = OpenAI(api_key=cfg["api_key"], base_url=BASE_URL) if cfg.get("api_key") else None
+
+    messages = [{"role": "system", "content": SYSTEM_PROMPT}]
+    state = {"auto": False}
+
+    print_banner()
+
+    while True:
+        try:
+            user = input(f"[{cfg['model']}] badnet> ").strip()
+        except (EOFError, KeyboardInterrupt):
+            print()
+            break
+
+        if not user:
+            continue
+
+        if user.startswith("/"):
+            cmd, _, arg = user.partition(" ")
+            arg = arg.strip()
+
+            if cmd == "/exit":
+                break
+            elif cmd == "/help":
+                print(HELP)
+            elif cmd == "/put":
+                if not arg:
+                    print("Использование: /put КЛЮЧ")
+                    continue
+                cfg["api_key"] = arg
+                save_config(cfg)
+                client = OpenAI(api_key=arg, base_url=BASE_URL)
+                print("Ключ сохранён в", CONFIG_FILE)
+            elif cmd == "/mod":
+                if not arg:
+                    print("Текущая модель:", cfg["model"])
+                    print("Доступно:", ", ".join(MODELS))
+                else:
+                    cfg["model"] = arg
+                    save_config(cfg)
+                    print("Модель:", arg)
+            elif cmd == "/clear":
+                messages = [{"role": "system", "content": SYSTEM_PROMPT}]
+                print("Контекст очищен")
+            elif cmd == "/run":
+                if not arg:
+                    print("Использование: /run КОМАНДА")
+                else:
+                    run_shell(arg)
+            elif cmd == "/cd":
+                try:
+                    os.chdir(os.path.expanduser(arg or "~"))
+                    print("Папка:", os.getcwd())
+                except OSError as e:
+                    print("Ошибка:", e)
+            elif cmd == "/cwd":
+                print(os.getcwd())
+            elif cmd == "/auto":
+                state["auto"] = not state["auto"]
+                print("Автовыполнение:", "вкл" if state["auto"] else "выкл")
+            else:
+                print(f"Неизвестная команда: {cmd}. Список - /help")
+            continue
+
+        if client is None:
+            print("Сначала задай ключ: /put КЛЮЧ")
+            continue
+
+        messages.append({"role": "user", "content": user})
+        try:
+            ask_model(client, cfg, messages, state)
+        except Exception as e:
+            print(f"Ошибка API: {e}")
+
+
+if __name__ == "__main__":
+    main()