autoMaster/git_api.py at master · alekseiko/autoMaster · GitHub

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
#!/usr/bin/env python

__author__ = "aleksei.kornev@gmail.com (Aleksei Kornev)"


import config
import subprocess
import sys

from logger import get_logger

LOGGER = get_logger("git_api")

class GitEngine:

	def __init__(self, repo = config.repo_dir):
		self.__repo = repo

	def __exec(self, command):
		LOGGER.debug("Exec command %s" % command)
		proc = subprocess.Popen(command, cwd = self.__repo, \
					 stderr = subprocess.PIPE, \
	                                 stdout = subprocess.PIPE, \
					 shell = True)
		try:
			stdout_value = proc.stdout.read().rstrip()
			stderr_value = proc.stderr.read().rstrip()
			status = proc.wait()
		finally:
			proc.stdout.close()
			proc.stderr.close()

		LOGGER.debug("FAIL STATUS %s" % status)
		if status != 0:
			LOGGER.error("Command: %s status: %s err: %s out: %s" % (command, status, \
					stderr_value, stdout_value))
			raise GitEngineError(status, stderr_value)

		return stdout_value

	def push(self, remote_path, local_branch, remote_branch):
		command = "git push %s %s:%s" % (remote_path, local_branch, remote_branch)

		self.__exec(command)

	def fetch(self, remote_path, refs_name):
		command = "git fetch " + remote_path
		# add refs definition
		command += " +refs/heads/*:refs/remotes/%s/*" % refs_name

		self.__exec(command)

	def cherry_pick(self, from_sha):
		command = "git cherry-pick %s" % from_sha

		self.__exec(command)


	def search(self, regexp, master_branch = None, slave_branch = None):
		""" return list of lists sha-> commit time in UNIX timestamp """
		command = "git log"

		if master_branch  is not None and slave_branch is not None:
			command += " %s..%s" % (master_branch, slave_branch)

		command += " --grep='" + regexp +"' --pretty=format:%H_%at"
		out = self.__exec(command)
		# if commits aren't found
		if out == "":
			return []
		# split commit shas to list
		return [ line.split("_") \
				for line in out.split("\n")]


	def checkout(self, sha, branch_name = None):
		command = "git checkout"

		if branch_name is not None:
			command += " -b " + branch_name

		command += " " + sha

		self.__exec(command)

	def reset(self, sha, isHard = False):
		command = "git reset"
		if isHard:
			command = command + " --hard"
		command = command + " " + sha

		self.__exec(command)

class GitEngineError(Exception):

	def __init__(self, status, error):
		self.status = status
		self.error = error

	def __str__(self):
		return repr(self.status + " " + self.error)

#if __name__ == "__main__":
#	engine = GitEngine()
#	engine.reset("HEAD^")
#	print engine.search("1")
#	engine.checkout(engine.search("h")[1], "test1")
#	engine.checkout("test/master", "test3000")
#	engine.cherry_pick("test1")
#	engine.fetch("git://github.com/teamurko/Cube.git", "cube")