mirror of
https://gitee.com/xmz7891/ykxmz.git
synced 2026-10-18 20:40:43 +00:00
84 lines
2.6 KiB
Python
84 lines
2.6 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""随身 WiFi（Debian）远程执行小工具。
|
||
|
||
用法：
|
||
python _build/devssh.py "命令" # 跑一条命令
|
||
python _build/devssh.py --script 文件 # 跑一段脚本
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import sys
|
||
import time
|
||
from pathlib import Path
|
||
|
||
import paramiko
|
||
|
||
HOST = "192.168.2.156"
|
||
USER = "root"
|
||
PASSWORD = "1313144"
|
||
|
||
|
||
def run(command: str, timeout: int = 600, quiet: bool = False) -> tuple:
|
||
client = paramiko.SSHClient()
|
||
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
||
client.connect(HOST, username=USER, password=PASSWORD, timeout=15,
|
||
banner_timeout=20, auth_timeout=20,
|
||
look_for_keys=False, allow_agent=False)
|
||
try:
|
||
_stdin, stdout, stderr = client.exec_command(command, timeout=timeout)
|
||
out = stdout.read().decode("utf-8", "replace")
|
||
err = stderr.read().decode("utf-8", "replace")
|
||
code = stdout.channel.recv_exit_status()
|
||
if not quiet:
|
||
if out.strip():
|
||
print(out.rstrip())
|
||
if err.strip():
|
||
print("[stderr] " + err.rstrip())
|
||
return code, out, err
|
||
finally:
|
||
client.close()
|
||
|
||
|
||
def upload(local: str, remote: str, mode: int = 0o644) -> None:
|
||
client = paramiko.SSHClient()
|
||
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
||
client.connect(HOST, username=USER, password=PASSWORD, timeout=15,
|
||
look_for_keys=False, allow_agent=False)
|
||
try:
|
||
sftp = client.open_sftp()
|
||
sftp.put(local, remote)
|
||
sftp.chmod(remote, mode)
|
||
sftp.close()
|
||
print(f" 已上传 {Path(local).name} -> {remote}")
|
||
finally:
|
||
client.close()
|
||
|
||
|
||
def download(remote: str, local: str) -> None:
|
||
client = paramiko.SSHClient()
|
||
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
||
client.connect(HOST, username=USER, password=PASSWORD, timeout=15,
|
||
look_for_keys=False, allow_agent=False)
|
||
try:
|
||
sftp = client.open_sftp()
|
||
sftp.get(remote, local)
|
||
sftp.close()
|
||
print(f" 已下载 {remote} -> {local}")
|
||
finally:
|
||
client.close()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
if len(sys.argv) < 2:
|
||
print(__doc__)
|
||
sys.exit(1)
|
||
if sys.argv[1] == "--script":
|
||
run(Path(sys.argv[2]).read_text(encoding="utf-8"), timeout=600)
|
||
else:
|
||
args = sys.argv[1:]
|
||
timeout = 600
|
||
if args and args[0] == '--timeout':
|
||
timeout = int(args[1])
|
||
args = args[2:]
|
||
run(" ".join(args), timeout=timeout)
|