Ubuntu
126 lines
4.8 KiB
Python
126 lines
4.8 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""Step C: 一键部署 — 本地 commit+push → 服务器 git pull 同步
|
||
|
||
工作流（根治 git 交叉）：
|
||
本地 git add/commit/push → 服务器 git pull --ff-only
|
||
|
||
服务器只做 pull 不做 commit —— pull 只移动引用，不产生新 commit。
|
||
这是交叉问题的根源修复：旧流程是 SFTP 上传后服务器再 commit 一次，
|
||
同一改动被提交两次，author/committer 时间戳不同 → 同 tree 两个 commit ID
|
||
→ ahead 1 behind 1。
|
||
|
||
依赖（已配置，非交互）：
|
||
本地: credential.helper=store, ~/.git-credentials
|
||
服务器: credential.helper=store, ~/.git-credentials, remote → 127.0.0.1:3000
|
||
(Gitea 与服务器同机，走 loopback 避免外部网络波动)
|
||
|
||
用法: python deploy.py "提交信息"
|
||
python deploy.py # 用默认信息
|
||
"""
|
||
import os
|
||
import subprocess
|
||
import sys
|
||
|
||
ROOT = os.path.dirname(os.path.abspath(__file__))
|
||
os.chdir(ROOT)
|
||
|
||
# .gitignore 已排除本机脚本与备份目录，git add -A 会自动跳过，无需额外 pathspec。
|
||
# （pathspec 排除 ':!x.py' 在文件已被 .gitignore 忽略时会报错：
|
||
# "The following paths are ignored by one of your .gitignore files"）
|
||
|
||
DEFAULT_MSG = 'chore: 部署更新'
|
||
SSH_BASE = ("ssh -q -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null "
|
||
"-o ConnectTimeout=10 ubuntu@106.55.169.92")
|
||
REMOTE_DIR = "/var/www/qianshi"
|
||
TIMEOUT = 60
|
||
|
||
ENV = dict(os.environ, GIT_TERMINAL_PROMPT='0') # 禁凭据交互，缺凭据立即失败而非挂起
|
||
|
||
|
||
def sh(args, timeout=TIMEOUT):
|
||
"""跑本地命令，返回 (exit, stdout)。args 为 list，shell=False。
|
||
|
||
Windows 上 shell=True 交给 cmd.exe，引号语义与 POSIX 不同，
|
||
提交信息等参数若用 shlex.quote 反而会出错 —— 统一用 list 直传。
|
||
"""
|
||
r = subprocess.run(args, capture_output=True, text=True,
|
||
timeout=timeout, env=ENV)
|
||
return r.returncode, (r.stdout + r.stderr).strip()
|
||
|
||
|
||
def log(msg):
|
||
print(' %s' % msg, flush=True)
|
||
|
||
|
||
def remote(cmd, timeout=TIMEOUT):
|
||
"""在服务器上执行命令，返回 (exit, stdout)
|
||
|
||
ssh 远程命令必须作为单个 argv 元素传（shell=False），
|
||
否则 Windows 的 cmd.exe 会先解析掉引号，ssh 拿不到完整命令。
|
||
cd 与 cmd 的拼接在 Python 里完成，不走任何 shell 解析。
|
||
"""
|
||
args = SSH_BASE.split() + ['cd %s && %s' % (REMOTE_DIR, cmd)]
|
||
r = subprocess.run(args, capture_output=True, text=True,
|
||
timeout=timeout, env=ENV)
|
||
return r.returncode, (r.stdout + r.stderr).strip()
|
||
|
||
|
||
def deploy(msg=DEFAULT_MSG):
|
||
log('[1/4] 检查改动')
|
||
_, st = sh(['git', 'status', '--porcelain'])
|
||
if not st:
|
||
log('工作区干净，无需提交')
|
||
else:
|
||
log('[2/4] 提交')
|
||
# .gitignore 已排除本机脚本与备份目录，git add -A 无需额外 pathspec
|
||
c, out = sh(['git', 'add', '-A'])
|
||
if c:
|
||
log('add 失败: %s' % out)
|
||
return False
|
||
_, staged = sh(['git', 'diff', '--cached', '--name-only'])
|
||
if not staged:
|
||
log('排除项之外无改动')
|
||
else:
|
||
log(' %d 个文件' % len(staged.splitlines()))
|
||
c, out = sh(['git', 'commit', '-m', msg])
|
||
if c:
|
||
log('commit 失败: %s' % out)
|
||
return False
|
||
_, commit = sh(['git', 'rev-parse', '--short', 'HEAD'])
|
||
log('本地 HEAD: %s' % commit)
|
||
|
||
log('[3/4] push origin main')
|
||
c, out = sh(['git', 'push', 'origin', 'main'])
|
||
if c:
|
||
log('push 失败: %s' % out)
|
||
return False
|
||
log(' pushed')
|
||
|
||
log('[4/4] 服务器 git pull --ff-only')
|
||
c, out = remote('GIT_TERMINAL_PROMPT=0 git pull --ff-only origin main')
|
||
if c:
|
||
log('pull 失败（若报 diverging，说明服务器产生过本地 commit，需人工处理）')
|
||
log(out)
|
||
return False
|
||
log(' %s' % out.splitlines()[-1] if out else ' ok')
|
||
|
||
# 校验两端一致（取完整 hash 的最末纯 hash 行，避免 pull 进度信息干扰）
|
||
_, local = sh(['git', 'rev-parse', 'HEAD'])
|
||
_, srv = remote('git rev-parse HEAD', 30)
|
||
srv_hash = next((l for l in srv.splitlines() if len(l) == 40 and all(
|
||
c in '0123456789abcdef' for c in l)), '?')
|
||
if local != srv_hash:
|
||
log('!! 本地 %s != 服务器 %s，同步失败' % (local[:7], srv_hash[:7]))
|
||
return False
|
||
log(' 服务器 HEAD: %s' % srv_hash[:7])
|
||
log('')
|
||
log('=== 部署完成 ===')
|
||
log('提交: %s %s' % (local[:7], msg))
|
||
log('本地/服务器 已同步，工作流无 commit 交叉')
|
||
return True
|
||
|
||
|
||
if __name__ == '__main__':
|
||
msg = sys.argv[1] if len(sys.argv) > 1 else DEFAULT_MSG
|
||
sys.exit(0 if deploy(msg) else 1)
|