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
|
__all__ = ['Pygit2GitODB', 'Pygit2GitDB', 'Pygit2CompatibilityGitDB']
from git.db.py.complex import PureGitODB
from git.db.py.base import (
PureRepositoryPathsMixin,
PureConfigurationMixin,
PureIndexDB,
)
from git.db.py.resolve import PureReferencesMixin
from git.db.py.transport import PureTransportDB
from git.db.py.submodule import PureSubmoduleDB
from git.db.cmd.complex import CmdHighLevelRepository, GitCommandMixin
from git.db.compat import RepoCompatibilityInterface
from pygit2 import Repository as Pygit2Repo
from git.base import OInfo, OStream
from git.fun import type_id_to_type_map, type_to_type_id_map
from git.util import hex_to_bin
from cStringIO import StringIO
import os
class Pygit2GitODB(PureGitODB):
"""A full fledged database to read and write object files from all kinds of sources."""
def __init__(self, objects_root):
"""Initalize this instance"""
PureGitODB.__init__(self, objects_root)
if hasattr(self, 'git_dir'):
wd = self.git_dir
else:
wd = os.path.dirname(objects_root)
#END try to figure out good entry for pygit2 - it needs the .gitdir
print objects_root
print wd
self._py2_repo = Pygit2Repo(wd)
def __getattr__(self, attr):
try:
# supply LazyMixin with this call first
return super(Pygit2GitODB, self).__getattr__(attr)
except AttributeError:
# now assume its on the pygit2 repository ... for now
return getattr(self._py2_repo, attr)
#END handle attr
#{ Object DBR
def info(self, binsha):
type_id, uncomp_data = self._py2_repo.read(binsha)
return OInfo(binsha, type_id_to_type_map[type_id], len(uncomp_data))
def stream(self, binsha):
type_id, uncomp_data = self._py2_repo.read(binsha)
return OStream(binsha, type_id_to_type_map[type_id], len(uncomp_data), StringIO(uncomp_data))
# #}END object dbr
#
# #{ Object DBW
def store(self, istream):
# TODO: remove this check once the required functionality was merged in pygit2
if hasattr(self._py2_repo, 'write'):
istream.binsha = hex_to_bin(self._py2_repo.write(type_to_type_id_map[istream.type], istream.read()))
return istream
else:
return super(Pygit2GitODB, self).store(istream)
#END handle write support
#}END object dbw
class Pygit2GitDB( PureRepositoryPathsMixin, PureConfigurationMixin,
PureReferencesMixin, PureSubmoduleDB,
PureIndexDB,
PureTransportDB, # not fully implemented
GitCommandMixin,
CmdHighLevelRepository,
Pygit2GitODB): # must come last, as it doesn't pass on __init__ with super
def __init__(self, root_path):
"""Initialize ourselves on the .git directory, or the .git/objects directory."""
PureRepositoryPathsMixin._initialize(self, root_path)
super(Pygit2GitDB, self).__init__(self.objects_dir)
class Pygit2CompatibilityGitDB(RepoCompatibilityInterface, Pygit2GitDB):
"""Basic pygit2 compatibility database"""
pass
|
