Skip to content

Commit a9bf50a

Browse files
committed
Add version utils
1 parent a72c903 commit a9bf50a

File tree

3 files changed

+29
-0
lines changed

3 files changed

+29
-0
lines changed
Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
from investing_bot_framework.utils.version import get_version
2+
3+
VERSION = (1, 0, 0, 'alpha', 0)
4+

investing_bot_framework/utils/__init__.py

Whitespace-only changes.
Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
1+
def get_version(version=None):
2+
version = get_complete_version(version)
3+
main = get_main_version(version)
4+
return main
5+
6+
7+
def get_main_version(version=None):
8+
"""Return main version (X.Y[.Z]) from VERSION."""
9+
version = get_complete_version(version)
10+
parts = 2 if version[2] == 0 else 3
11+
return '.'.join(str(x) for x in version[:parts])
12+
13+
14+
def get_complete_version(version=None):
15+
"""
16+
Return a tuple of the investing algorithm framework version. If version argument is non-empty,
17+
check for correctness of the tuple provided.
18+
"""
19+
if version is None:
20+
from investing_bot_framework import VERSION as version
21+
else:
22+
assert len(version) == 5
23+
assert version[3] in ('alpha', 'beta', 'rc', 'final')
24+
25+
return version

0 commit comments

Comments
 (0)