python操作svn信息获取 python实现对svn操作及信息获取

"""
SVN状态对照表
"""
class FileState:
    Normal = 0  # 000000 正常在svn管理下的最新的文件
    RemoteLocked = 1  # 000001 云端锁定态
    LocalLocked = 2  # 000010 本地锁定态
    Locked = 3  # 000011 已锁定 state and Locked == True
    LocalMod = 4  # 000100 本地有修改需提交
    RemoteMod = 8  # 001000 远程有修改需要更新
    Conflicked = 12  # 001100 冲突 state and Conflicked == Conflicked
    UnVersioned = 16  # 010000 未提交到库
    Error = 32  # 100000 错误状态

"""
具体实现逻辑
"""
# -*- coding: utf-8 -*-
import os
import pprint
import subprocess
import time
from threading import Thread
from xmltodict import parse as xmlParse
def _doSvnCommandSync(args):
    startupinfo = subprocess.STARTUPINFO()
    startupinfo.dwFlags = subprocess.CREATE_NEW_CONSOLE | subprocess.STARTF_USESHOWWINDOW
    startupinfo.wShowWindow = subprocess.SW_HIDE
    p = subprocess.Popen(
        args,
        stdout=subprocess.PIPE,
        stderr=subprocess.PIPE,
        startupinfo=startupinfo,
        shell=True
    )
    rst, err = p.communicate()
    try:
        rst = str(rst, 'utf-8')
    except:
        rst = str(rst, 'gbk', errors="-ignore")
    try:
        err = str(err, 'utf-8')
    except:
        err = str(err, 'gbk', errors="-ignore")
    return rst, err
def svnCommitNoUnlockSync(path, comment=""):
    rst, err = _doSvnCommandSync("svn commit " + path + " -m \"" + comment + "\"" + " --no-unlock")
    return rst, err
def svnCommitSync(path, comment=""):
    rst, err = _doSvnCommandSync("svn commit " + path + " -m \"" + comment + "\"")
    return rst, err
def _svnStatusSync(path):
    rst, err = _doSvnCommandSync("svn status " + path)
    if err:
        return None, err
    data = rst
    return data, None
def svnLockSync(path):
    rst, err = _doSvnCommandSync("svn lock -m '哈哈哈哈哈哈' " + path)
    return rst, err
def svnAddSync(path):
    data, err = _doSvnCommandSync("svn add " + path)
    return data, err
def svnUnLockSync(path):
    rst, err = _doSvnCommandSync("svn unlock " + path)
    return rst, err
def svnDeleteSync(path):
    return _doSvnCommandSync("svn delete " + path)
def _svnStatusXMLSync(path):
    rst, err = _doSvnCommandSync("svn status " + path + " -u --xml")
    if err:
        return None, err
    data = rst
    data = xmlParse(data)
    return data, None
def syncGetAllFileStatus(rootPath):
    data, info = _svnStatusXMLSync(rootPath)
    returnDict = {}
    lockRole = ""
    state = FileState.Normal
    if info:
        if data is None:
            state = state | FileState.UnVersioned
        else:
            state = state | FileState.Error
        return returnDict
    target = data["status"]["target"]

    if target and "entry" in target:
        iterList = []
        if not isinstance(target["entry"], list):
            iterList.append(target["entry"])
        else:
            iterList = target["entry"]
        for fileStatusItem in iterList:
            state = FileState.Normal
            filePath = fileStatusItem["@path"]
            wc_status = fileStatusItem["wc-status"]
            if "unversioned" == wc_status["@item"]:
                state = state | FileState.UnVersioned
            elif "modified" == wc_status["@item"]:
                state = state | FileState.LocalMod
            elif "repos-status" in fileStatusItem:
                repos_status = fileStatusItem["repos-status"]
                if "lock" in repos_status and "lock" not in wc_status:
                    info = repos_status["lock"]["owner"]
                    lockRole = info
                    state = state | FileState.RemoteLocked
                elif "lock" in wc_status:
                    info = wc_status["lock"]["owner"]
                    lockRole = info
                    state = state | FileState.LocalLocked
                elif "modified" == repos_status["@item"]:
                    state = state | FileState.RemoteMod
                    info = "%s is modified on remote, you need update first" % filePath
                if "modified" == wc_status["@item"]:
                    state = state | FileState.LocalMod
                    info = "%s is modified on local, you need commit first" % filePath
            returnDict[os.path.normcase(filePath)] = [state, info, lockRole]
    return returnDict
def openTortoise():
    pathsStr = "".join("G:\SVNCheckOut\Txt2")
    cmd = "TortoiseProc.exe /command:commit /path %s" % pathsStr
    p = subprocess.Popen(
        cmd,
        stdout=subprocess.PIPE,
        stderr=subprocess.PIPE,
        encoding="utf-8",
        shell=True
    )
# class Process(subprocess.Popen):
#     def register_callback(self, callback, *args, **kwargs):
#         Thread(target=self._poll_completion, args=(callback, args, kwargs)).start()
#
#     def _poll_completion(self, callback, args, kwargs):
#         while self.poll() is None:
#             time.sleep(0.1)
#         callback(*args, **kwargs)
# def openTortoise():
#     pathsStr = "".join("G:\SVNCheckOut\Version1")
#     cmd = "TortoiseProc.exe /command:commit /path %s" % pathsStr
#     handle = Process(cmd)
#     handle.register_callback(MyPrint)
# def MyPrint():
#     print("~~~~~~~~~~~~~~~~~")
openTortoise()
data = syncGetAllFileStatus(r"G:\SVNCheckOut\Txt2")
pprint.pprint(data)
os.system("Pause")