Skip to content

Commit

Permalink
Added cvat.utils with get_version implementation
Browse files Browse the repository at this point in the history
  • Loading branch information
Nikita Manovich committed Jan 23, 2019
1 parent a2dc274 commit 7b8b9f8
Showing 1 changed file with 59 additions and 0 deletions.
59 changes: 59 additions & 0 deletions cvat/utils/version.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,59 @@
# Copyright (C) 2019 Intel Corporation
#
# SPDX-License-Identifier: MIT
#
# Note: It is slightly re-implemented Django version of code. We cannot use
# get_version from django.utils.version module because get_git_changeset will
# always return empty value (cwd=repo_dir isn't correct). Also it gives us a
# way to define the version as we like.

import datetime
import os
import subprocess

def get_version(version):
"""Return a PEP 440-compliant version number from VERSION."""
# Now build the two parts of the version number:
# main = X.Y[.Z]
# sub = .devN - for pre-alpha releases
# | {a|b|rc}N - for alpha, beta, and rc releases

main = get_main_version(version)

sub = ''
if version[3] == 'alpha' and version[4] == 0:
git_changeset = get_git_changeset()
if git_changeset:
sub = '.dev%s' % git_changeset

elif version[3] != 'final':
mapping = {'alpha': 'a', 'beta': 'b', 'rc': 'rc'}
sub = mapping[version[3]] + str(version[4])

return main + sub

def get_main_version(version):
"""Return main version (X.Y[.Z]) from VERSION."""
parts = 2 if version[2] == 0 else 3
return '.'.join(str(x) for x in version[:parts])

def get_git_changeset():
"""Return a numeric identifier of the latest git changeset.
The result is the UTC timestamp of the changeset in YYYYMMDDHHMMSS format.
This value isn't guaranteed to be unique, but collisions are very unlikely,
so it's sufficient for generating the development version numbers.
"""
repo_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
git_log = subprocess.Popen(
'git log --pretty=format:%ct --quiet -1 HEAD',
stdout=subprocess.PIPE, stderr=subprocess.PIPE,
shell=True, cwd=repo_dir, universal_newlines=True,
)
timestamp = git_log.communicate()[0]
try:
timestamp = datetime.datetime.utcfromtimestamp(int(timestamp))
except ValueError:
return None
return timestamp.strftime('%Y%m%d%H%M%S')

0 comments on commit 7b8b9f8

Please sign in to comment.