wip: 2026-08-11 工作树快照(docs/memory/h5.py/scripts 等 447 项未评审改动,安全提交到 feat 分支)
This commit is contained in:
@@ -0,0 +1,78 @@
|
||||
"""Upload large file to server via plink PTY base64 chunks."""
|
||||
import subprocess
|
||||
import base64
|
||||
import hashlib
|
||||
import sys
|
||||
import os
|
||||
|
||||
LOCAL_FILE = r"D:\资料\03-项目开发\wecom_it_smart_desk\tmp-h5-dist.tar.gz"
|
||||
REMOTE_FILE = "/tmp/h5-dist.tar.gz"
|
||||
CHUNK_SIZE = 4000 # chars per chunk
|
||||
|
||||
PLINK = r"C:\Users\simon\.workbuddy\skills\jumpserver-ops\scripts\plink.exe"
|
||||
HOST = "sxn@10.212.189.210"
|
||||
PORT = "2222"
|
||||
PASSWORD = os.environ.get("JMS_PASSWORD", "") # May be cached
|
||||
|
||||
def run_plink(commands):
|
||||
"""Run commands via plink PTY."""
|
||||
cmd = [PLINK, "-P", str(PORT), "-batch", HOST]
|
||||
stdin_data = "\n".join(commands) + "\nexit\n"
|
||||
result = subprocess.run(
|
||||
cmd,
|
||||
input=stdin_data,
|
||||
capture_output=True,
|
||||
text=True,
|
||||
timeout=300
|
||||
)
|
||||
return result.stdout + result.stderr
|
||||
|
||||
def main():
|
||||
# Read file
|
||||
with open(LOCAL_FILE, "rb") as f:
|
||||
data = f.read()
|
||||
|
||||
local_md5 = hashlib.md5(data).hexdigest()
|
||||
print(f"File: {LOCAL_FILE}")
|
||||
print(f"Size: {len(data)} bytes")
|
||||
print(f"MD5: {local_md5}")
|
||||
|
||||
# Base64 encode
|
||||
b64 = base64.b64encode(data).decode("ascii")
|
||||
total_chunks = (len(b64) + CHUNK_SIZE - 1) // CHUNK_SIZE
|
||||
print(f"Base64 length: {len(b64)} chars, {total_chunks} chunks")
|
||||
|
||||
# Clear remote file
|
||||
print("Clearing remote file...")
|
||||
run_plink([f"> {REMOTE_FILE}.b64"])
|
||||
|
||||
# Send chunks
|
||||
for i in range(total_chunks):
|
||||
start = i * CHUNK_SIZE
|
||||
end = min(start + CHUNK_SIZE, len(b64))
|
||||
chunk = b64[start:end]
|
||||
cmd = f"echo '{chunk}' >> {REMOTE_FILE}.b64"
|
||||
run_plink([cmd])
|
||||
if (i + 1) % 10 == 0 or i == total_chunks - 1:
|
||||
print(f" Sent chunk {i+1}/{total_chunks}")
|
||||
|
||||
# Decode and verify
|
||||
print("Decoding and verifying...")
|
||||
verify_cmds = [
|
||||
f"base64 -d {REMOTE_FILE}.b64 > {REMOTE_FILE}",
|
||||
f"wc -c < {REMOTE_FILE}",
|
||||
f"md5sum {REMOTE_FILE}",
|
||||
f"rm -f {REMOTE_FILE}.b64",
|
||||
]
|
||||
output = run_plink(verify_cmds)
|
||||
print(f"Server output:\n{output}")
|
||||
|
||||
if local_md5 in output:
|
||||
print(f"\n✅ MD5 match! Upload successful.")
|
||||
return 0
|
||||
else:
|
||||
print(f"\n❌ MD5 mismatch! Expected: {local_md5}")
|
||||
return 1
|
||||
|
||||
if __name__ == "__main__":
|
||||
sys.exit(main())
|
||||
Reference in New Issue
Block a user