GitPython/test/test_quick_doc.py at main · pushfoo/GitPython · GitHub

Repository navigation

History

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

193

194

195

196

197

198

199

200

201

202

203

204

205

206

207

208

209

210

211

212

213

214

215

216

217

218

219

from test.lib import TestBase

from test.lib.helper import with_rw_directory

class QuickDoc(TestBase):

def tearDown(self):

import gc

gc.collect()

@with_rw_directory

def test_init_repo_object(self, path_to_dir):

# [1-test_init_repo_object]

# $ git init <path/to/dir>

from git import Repo

repo = Repo.init(path_to_dir)

# ![1-test_init_repo_object]

# [2-test_init_repo_object]

repo = Repo(path_to_dir)

# ![2-test_init_repo_object]

del repo # Avoids "assigned to but never used" warning. Doesn't go in the docs.

@with_rw_directory

def test_cloned_repo_object(self, local_dir):

from git import Repo

# code to clone from url

# [1-test_cloned_repo_object]

# $ git clone <url> <local_dir>

repo_url = "https://github.com/gitpython-developers/QuickStartTutorialFiles.git"

repo = Repo.clone_from(repo_url, local_dir)

# ![1-test_cloned_repo_object]

# code to add files

# [2-test_cloned_repo_object]

# We must make a change to a file so that we can add the update to git

update_file = "dir1/file2.txt" # we'll use local_dir/dir1/file2.txt

with open(f"{local_dir}/{update_file}", "a") as f:

f.write("\nUpdate version 2")

# ![2-test_cloned_repo_object]

# [3-test_cloned_repo_object]

# $ git add <file>

add_file = [update_file] # relative path from git root

repo.index.add(add_file) # notice the add function requires a list of paths

# ![3-test_cloned_repo_object]

# code to commit - not sure how to test this

# [4-test_cloned_repo_object]

# $ git commit -m <message>

repo.index.commit("Update to file2")

# ![4-test_cloned_repo_object]

# [5-test_cloned_repo_object]

# $ git log <file>

# relative path from git root

repo.iter_commits(all=True, max_count=10, paths=update_file) # gets the last 10 commits from all branches

# Outputs: <generator object Commit._iter_from_process_or_stream at 0x7fb66c186cf0>

# ![5-test_cloned_repo_object]

# [6-test_cloned_repo_object]

commits_for_file_generator = repo.iter_commits(all=True, max_count=10, paths=update_file)

commits_for_file = list(commits_for_file_generator)

commits_for_file

# Outputs: [<git.Commit "SHA1-HEX_HASH-2">,

# <git.Commit "SHA1-HEX-HASH-1">]

# ![6-test_cloned_repo_object]

# Untracked files - create new file

# [7-test_cloned_repo_object]

f = open(f"{local_dir}/untracked.txt", "w") # creates an empty file

f.close()

# ![7-test_cloned_repo_object]

# [8-test_cloned_repo_object]

repo.untracked_files

# Output: ['untracked.txt']

# ![8-test_cloned_repo_object]

# Modified files

# [9-test_cloned_repo_object]

# Let's modify one of our tracked files

with open(f"{local_dir}/Downloads/file3.txt", "w") as f:

f.write("file3 version 2") # overwrite file 3

# ![9-test_cloned_repo_object]

# [10-test_cloned_repo_object]

repo.index.diff(None) # compares staging area to working directory

# Output: [<git.diff.Diff object at 0x7fb66c076e50>,

# <git.diff.Diff object at 0x7fb66c076ca0>]

# ![10-test_cloned_repo_object]

# [11-test_cloned_repo_object]

diffs = repo.index.diff(None)

for d in diffs:

print(d.a_path)

# Output

# Downloads/file3.txt

# ![11-test_cloned_repo_object]

# compares staging area to head commit

# [11.1-test_cloned_repo_object]

diffs = repo.index.diff(repo.head.commit)

for d in diffs:

print(d.a_path)

# Output

# ![11.1-test_cloned_repo_object]

# [11.2-test_cloned_repo_object]

# lets add untracked.txt

repo.index.add(["untracked.txt"])

diffs = repo.index.diff(repo.head.commit)

for d in diffs:

print(d.a_path)

# Output

# untracked.txt

# ![11.2-test_cloned_repo_object]

# Compare commit to commit

# [11.3-test_cloned_repo_object]

first_commit = list(repo.iter_commits(all=True))[-1]

diffs = repo.head.commit.diff(first_commit)

for d in diffs:

print(d.a_path)

# Output

# dir1/file2.txt

# ![11.3-test_cloned_repo_object]

"""Trees and Blobs"""

# Latest commit tree

# [12-test_cloned_repo_object]

tree = repo.head.commit.tree

# ![12-test_cloned_repo_object]

# Previous commit tree

# [13-test_cloned_repo_object]

prev_commits = list(repo.iter_commits(all=True, max_count=10)) # last 10 commits from all branches

tree = prev_commits[0].tree

# ![13-test_cloned_repo_object]

# Iterating through tree

# [14-test_cloned_repo_object]

files_and_dirs = [(entry, entry.name, entry.type) for entry in tree]

files_and_dirs

# Output

# [(< git.Tree "SHA1-HEX_HASH" >, 'Downloads', 'tree'),

# (< git.Tree "SHA1-HEX_HASH" >, 'dir1', 'tree'),

# (< git.Blob "SHA1-HEX_HASH" >, 'file4.txt', 'blob')]

# ![14-test_cloned_repo_object]

# [15-test_cloned_repo_object]

def print_files_from_git(root, level=0):

for entry in root:

print(f'{"-" * 4 * level}| {entry.path}, {entry.type}')

if entry.type == "tree":

print_files_from_git(entry, level + 1)

# ![15-test_cloned_repo_object]

# [16-test_cloned_repo_object]

print_files_from_git(tree)

# Output

# | Downloads, tree

# ----| Downloads / file3.txt, blob

# | dir1, tree

# ----| dir1 / file1.txt, blob

# ----| dir1 / file2.txt, blob

# | file4.txt, blob

# # ![16-test_cloned_repo_object]

# Printing text files

# [17-test_cloned_repo_object]

print_file = "dir1/file2.txt"

tree[print_file] # the head commit tree

# Output <git.Blob "SHA1-HEX-HASH">

# ![17-test_cloned_repo_object]

# print latest file

# [18-test_cloned_repo_object]

blob = tree[print_file]

print(blob.data_stream.read().decode())

# Output

# file 2 version 1

# Update version 2

# ![18-test_cloned_repo_object]

# print previous tree

# [18.1-test_cloned_repo_object]

commits_for_file = list(repo.iter_commits(all=True, paths=print_file))

tree = commits_for_file[-1].tree # gets the first commit tree

blob = tree[print_file]

print(blob.data_stream.read().decode())

# Output

# file 2 version 1

# ![18.1-test_cloned_repo_object]

GitPython/test/test_quick_doc.py at main · pushfoo/GitPython · GitHub

Repository navigation

Expand file tree

Breadcrumbs

test_quick_doc.py

Latest commit

History

Breadcrumbs

test_quick_doc.py

File metadata and controls