1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117
|
import os
import pytest
import shutil
import subprocess
from .repo import Repo
from ..site import HAVE_GIT
GIT_ENV = {
'GIT_AUTHOR_DATE': '1320966000 +0200',
'GIT_AUTHOR_NAME': 'tomjon',
'GIT_AUTHOR_EMAIL': 'tom@jon.com',
'GIT_COMMITTER_DATE': '1320966000 +0200',
'GIT_COMMITTER_NAME': 'tomjon',
'GIT_COMMITTER_EMAIL': 'tom@jon.com'
}
class Git(Repo):
def __init__(self, directory, subdir):
if not HAVE_GIT:
pytest.skip("git is not available")
self.submodules = {}
super(Git, self).__init__(directory, subdir)
def create(self, directory):
self.copy_directory(directory, self.repo)
subprocess.call(['git', 'init', '.'], env=GIT_ENV, cwd=self.repo)
subprocess.call(
['git', 'checkout', '-b', 'master'],
env=GIT_ENV,
cwd=self.repo
)
subprocess.call(['git', 'add', '.'], env=GIT_ENV, cwd=self.repo)
subprocess.call(['git', 'commit', '-m', 'Initial commit'], env=GIT_ENV, cwd=self.repo)
return self.latest_commit()
def add_tag(self, tag):
subprocess.call(['git', 'tag', tag], env=GIT_ENV, cwd=self.repo)
def add_commit(self):
subprocess.call(['git', 'commit', '--allow-empty', '-m', 'Additional commit'],
env=GIT_ENV, cwd=self.repo)
return self.latest_commit()
def add_file(self, filename):
shutil.copy(filename, self.repo)
subprocess.call(['git', 'add', os.path.basename(filename)], env=GIT_ENV, cwd=self.repo)
subprocess.call([
'git', 'commit', '-m', 'Added {}'.format(os.path.basename(filename))
], env=GIT_ENV, cwd=self.repo)
return self.latest_commit()
def modify_file(self, new_file, path):
shutil.copy(new_file, os.path.join(self.repo, path))
subprocess.call([
'git', 'commit', path, '-m', 'Modified {}'.format(os.path.basename(path))
], env=GIT_ENV, cwd=self.repo)
return self.latest_commit()
def add_submodule(self, subdir, url=None, checkout=None):
submodule = {}
if checkout is not None:
submodule['checkout'] = checkout
if url is not None:
submodule['url'] = url
self.submodules[subdir] = submodule
subprocess.call(['git', 'submodule', 'add', url, subdir], env=GIT_ENV, cwd=self.repo)
subprocess.call(['git', 'commit', '-m', 'Added the submodule'], env=GIT_ENV, cwd=self.repo)
return self.latest_commit()
# This can also be used to a file or a submodule
def remove_path(self, path):
subprocess.call(['git', 'rm', path], env=GIT_ENV, cwd=self.repo)
subprocess.call(['git', 'commit', '-m', 'Removing {}'.format(path)], env=GIT_ENV, cwd=self.repo)
return self.latest_commit()
def source_config(self, ref=None, checkout_submodules=None):
config = {
'kind': 'git',
'url': 'file://' + self.repo,
'track': 'master'
}
if ref is not None:
config['ref'] = ref
if checkout_submodules is not None:
config['checkout-submodules'] = checkout_submodules
if self.submodules:
config['submodules'] = dict(self.submodules)
return config
def latest_commit(self):
output = subprocess.check_output([
'git', 'rev-parse', 'HEAD'
], env=GIT_ENV, cwd=self.repo)
return output.decode('UTF-8').strip()
def branch(self, branch_name):
subprocess.call(['git', 'checkout', '-b', branch_name], env=GIT_ENV, cwd=self.repo)
def delete_tag(self, tag_name):
subprocess.call(['git', 'tag', '-d', tag_name],
env=GIT_ENV, cwd=self.repo)
def checkout(self, commit):
subprocess.call(['git', 'checkout', commit],
env=GIT_ENV, cwd=self.repo)
def add_annotated_tag(self, tag, message):
subprocess.call(['git', 'tag', '-a', tag, '-m', message],
env=GIT_ENV, cwd=self.repo)
return self.latest_commit()
|