62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
from __future__ import absolute_import
|
|
|
|
import os.path
|
|
|
|
try:
|
|
import pkg_resources
|
|
except ImportError:
|
|
# pkg_resource is not available on Google App Engine
|
|
pkg_resources = None
|
|
|
|
from raven._compat import text_type
|
|
from .exceptions import InvalidGitRepository
|
|
|
|
__all__ = ('fetch_git_sha', 'fetch_package_version')
|
|
|
|
|
|
def fetch_git_sha(path, head=None):
|
|
"""
|
|
>>> fetch_git_sha(os.path.dirname(__file__))
|
|
"""
|
|
if not head:
|
|
head_path = os.path.join(path, '.git', 'HEAD')
|
|
if not os.path.exists(head_path):
|
|
raise InvalidGitRepository(
|
|
'Cannot identify HEAD for git repository at %s' % (path,))
|
|
|
|
with open(head_path, 'r') as fp:
|
|
head = text_type(fp.read()).strip()
|
|
|
|
if head.startswith('ref: '):
|
|
revision_file = os.path.join(
|
|
path, '.git', *head.rsplit(' ', 1)[-1].split('/')
|
|
)
|
|
else:
|
|
return head
|
|
else:
|
|
revision_file = os.path.join(path, '.git', 'refs', 'heads', head)
|
|
|
|
if not os.path.exists(revision_file):
|
|
if not os.path.exists(os.path.join(path, '.git')):
|
|
raise InvalidGitRepository(
|
|
'%s does not seem to be the root of a git repository' % (path,))
|
|
raise InvalidGitRepository(
|
|
'Unable to find ref to head "%s" in repository' % (head,))
|
|
|
|
fh = open(revision_file, 'r')
|
|
try:
|
|
return text_type(fh.read()).strip()
|
|
finally:
|
|
fh.close()
|
|
|
|
|
|
def fetch_package_version(dist_name):
|
|
"""
|
|
>>> fetch_package_version('sentry')
|
|
"""
|
|
if pkg_resources is None:
|
|
raise NotImplementedError('pkg_resources is not available '
|
|
'on this Python install')
|
|
dist = pkg_resources.get_distribution(dist_name)
|
|
return dist.version
|