import os
from itertools import islice
import cPickle as pickle
from datetime import datetime
import git
import pylons
import pymongo.bson
from ming.orm.mapped_class import MappedClass
from ming.orm.property import FieldProperty
from ming.utils import LazyProperty
from pyforge.model import Repository, ArtifactReference, User
class GitRepository(Repository):
class __mongometa__:
name='git-repository'
def index(self):
result = Repository.index(self)
result.update(
type_s='GitRepository')
return result
def commit(self, rev):
return GitCommit(rev, self)
def iter_commits(self, *args, **kwargs):
return (GitCommit.from_git(c, self)
for c in self._impl.iter_commits(*args, **kwargs))
@LazyProperty
def _impl(self):
return git.Repo(os.path.join(self.path, self.name))
def __getattr__(self, name):
return getattr(self._impl, name)
@property
def tags(self):
'''Override Artifact.tags'''
return self._impl.tags
class MockQuery(object):
def __init__(self, cls):
self._cls = cls
def get(self, _id):
return self._cls(_id, repo=pylons.c.app.repo)
class GitCommit(object):
type_s='GitCommit'
def __init__(self, id, repo):
self._id = id
self._repo = repo
@classmethod
def from_git(cls, c, repo):
result = cls(id=c.sha, repo=repo)
result.__dict__['_impl'] = c
return result
def dump_ref(self):
'''Return a pickle-serializable reference to an artifact'''
try:
d = ArtifactReference(dict(
project_id=pylons.c.project._id,
mount_point=pylons.c.app.config.options.mount_point,
artifact_type=pymongo.bson.Binary(pickle.dumps(self.__class__)),
artifact_id=self._id))
return d
except AttributeError:
return None
def url(self):
return self._repo.url() + self._id
def primary(self, *args):
return self
def shorthand_id(self):
return '[%s]' % self._id[:6]
@LazyProperty
def _impl(self):
return self._repo._impl.commit(self._id)
def __getattr__(self, name):
return getattr(self._impl, name)
@LazyProperty
def authored_datetime(self):
return datetime.fromtimestamp(self.authored_date+self.author_tz_offset)
@LazyProperty
def committed_datetime(self):
return datetime.fromtimestamp(self.authored_date+self.author_tz_offset)
@LazyProperty
def author_url(self):
u = User.by_email_address(self.author.email)
if u: return u.url()
@LazyProperty
def committer_url(self):
u = User.by_email_address(self.committer.email)
if u: return u.url()
@LazyProperty
def parents(self):
return tuple(GitCommit.from_git(c, self._repo) for c in self._impl.parents)
GitCommit.query = MockQuery(GitCommit)
MappedClass.compile_all()