# -*- coding: utf-8 -*-
from os import chmod, listdir, remove
from os.path import abspath, isdir, isfile, join
from platform import system
from re import compile, findall
from shutil import rmtree
from subprocess import PIPE, STDOUT, run, check_output, CalledProcessError
def remove_index_lock(directory):
lock = join(directory, 'index.lock')
if isfile(lock):
remove(lock)
class SCM(object):
__shell = system() != 'Windows'
def __init__(self, local:str=None, remote:str=None):
self.local = local
self.remote = remote
if self.remote is None:
self.remote = self.__get_remote_from_local()
assert self.remote is not None
@property
def type(self)->str:
local_type = self.__get_type_from_local()
remote_type = self.__get_type_from_remote()
if local_type and remote_type:
assert local_type == remote_type
return local_type or remote_type
def __get_remote_from_local(self)->str:
if self.local:
if self.type == 'svn':
p = run(f'svn info "{self.local}"', stdout=PIPE, stderr=STDOUT, shell=SCM.__shell)
if p.returncode == 0:
return findall(compile('URL: (.+)\s'), p.stdout.decode('gbk'))[0].strip()
elif self.type == 'git':
p = run(f'git remote -v', cwd=self.local, stdout=PIPE, stderr=STDOUT, shell=SCM.__shell)
if p.returncode == 0:
return findall(compile(f'origin\s(.+)\s\(fetch\)'), p.stdout.decode())[0]
return None
def __get_type_from_local(self)->str:
if self.local and isdir(self.local):
p = run(f'svn info "{self.local}"', stdout=PIPE, stderr=STDOUT, shell=SCM.__shell)
if p.returncode == 0:
return 'svn'
p = run('git rev-parse --short HEAD', cwd=self.local, stdout=PIPE, stderr=STDOUT, shell=SCM.__shell)
if p.returncode == 0:
return 'git'
return None
def __get_type_from_remote(self)->str:
if self.remote:
if '/svn/' in self.remote:
return 'svn'
elif self.remote.endswith('.git'):
return 'git'
return None
def checkout(self)->bool:
if self.type == 'svn':
p = run(f'svn checkout "{self.remote}" "{self.local}"', shell=SCM.__shell, check=True)
elif self.type == 'git':
p = run(f'git clone "{self.remote}" "{self.local}"', shell=SCM.__shell, check=True)
return p.returncode == 0
def switch(self, branch:str)->bool:
if branch is None:
return False
if self.type == 'svn':
p = run(f'svn switch {branch}', cwd=self.local, shell=SCM.__shell, check=True)
elif self.type == 'git':
p = run('git fetch --progress -v "origin"', cwd=self.local, stdout=PIPE, stderr=STDOUT, shell=SCM.__shell)
p = run(f'git switch {branch}', cwd=self.local, shell=SCM.__shell, check=True)
return p.returncode == 0
def revert(self, path:str=''):
if self.type == 'svn':
run(f'svn cleanup "{self.local}"', stdout=PIPE, stderr=STDOUT, shell=SCM.__shell, check=True) # use cleanup to release lock
run(f'svn revert -R "{abspath(join(self.local, path))}"', shell=SCM.__shell, check=True)
elif self.type == 'git':
work_dir = join(self.local, path)
run('git merge --abort', cwd=work_dir, stderr=PIPE, shell=SCM.__shell) # revert conflict
run('git restore .', cwd=work_dir, shell=SCM.__shell, check=True) # revert modified
run('git restore --staged .', cwd=work_dir, shell=SCM.__shell, check=True) # revert added
def fetch(self):
match self.type:
case 'git':
run('git fetch --progress -v "origin"', cwd=self.local, stdout=PIPE, stderr=STDOUT, shell=SCM.__shell)
def pull(self):
match self.type:
case 'git':
run('git pull --progress -v --no-rebase "origin"', cwd=self.local, shell=SCM.__shell, check=True)
def cleanup(self):
if self.type == 'svn':
run(f'svn cleanup --remove-unversioned "{self.local}"', shell=SCM.__shell, check=True)
elif self.type == 'git':
run('git clean -dfx', cwd=self.local, shell=SCM.__shell, check=True)
def update(self, revision:str=''):
revision = revision or '' # 防止None值
revision = revision.strip()
if self.type == 'svn':
if revision:
run(f'svn update "{self.local}" -r {revision}', shell=SCM.__shell, check=True)
else:
run(f'svn update "{self.local}"', shell=SCM.__shell, check=True)
elif self.type == 'git':
if revision:
self.fetch()
run(f'git reset --hard {revision}', cwd=self.local, shell=SCM.__shell, check=True)
else:
self.pull()
def add(self, file:str):
if self.type == 'svn':
p = run(f'svn add {file} --force', cwd=self.local, shell=SCM.__shell)
p = run('cmd.exe /c svn status | find "!"', cwd=self.local, stdout=PIPE, stderr=STDOUT, shell=SCM.__shell)
missing_files = p.stdout.decode().strip().replace('! ', '').splitlines()
for missing_file in missing_files:
p = run(f'svn delete {missing_file}', cwd=self.local, shell=SCM.__shell)
elif self.type == 'git':
run(f'git add {file}', cwd=self.local, shell=SCM.__shell, check=True)
else:
pass
def commit(self, msg:str):
if self.type == 'svn':
run(f'svn commit -m "{msg}"', cwd=self.local, shell=SCM.__shell, check=True)
proc = run('svn status', cwd=self.local, stdout=PIPE, stderr=STDOUT, shell=SCM.__shell, check=False)
_stdout = proc.stdout.decode('utf-8')
assert len(findall(r'M\s+(.+)', _stdout)) == 0, "修改未提交"
assert len(findall(r'!\s+(.+)', _stdout)) == 0, "未删除文件"
elif self.type == 'git':
run(f'git commit -m "{msg}"', cwd=self.local, shell=SCM.__shell, check=True)
run('git push', cwd=self.local, shell=SCM.__shell, check=True)
else:
pass
@property
def revision(self)->str:
if self.type == 'svn':
p = run(f'svn info "{self.local}"', stdout=PIPE, stderr=STDOUT, shell=SCM.__shell)
try:
return findall(compile('Last Changed Rev: ([0-9]+)'), str(p.stdout))[0]
except:
return findall(compile('最后修改的版本: ([0-9]+)'), p.stdout.decode('utf-8'))[0]
elif self.type == 'git':
p = run('git rev-parse --short HEAD', cwd=self.local, stdout=PIPE, stderr=STDOUT, shell=SCM.__shell)
return p.stdout.decode().strip()
else:
return ' '
def get_latest_revision(self, branch:str)->str:
if self.type == 'svn':
p = run(f'svn info "{branch}"', stdout=PIPE, stderr=STDOUT, shell=SCM.__shell, check=True)
try:
return findall(compile('Last Changed Rev: ([0-9]+)'), str(p.stdout))[0]
except:
return findall(compile('最后修改的版本: ([0-9]+)'), p.stdout.decode('utf-8'))[0]
elif self.type == 'git':
p = run(f'git ls-remote "{self.remote}" refs/heads/{branch}', cwd=self.local, stdout=PIPE, stderr=STDOUT, shell=SCM.__shell)
return findall(compile(f'([A-Za-z0-9]+)\s+refs/heads/{branch}'), p.stdout.decode())[0][0:9]
else:
return ' '
def get_commits_between(self, start:str, end:str)->list:
if self.type == 'svn':
pass
elif self.type == 'git':
try:
# 获取从指定标签到当前 HEAD 之间的提交
log_format =
