TrueNAS/scale-build
1
0
Fork 0
mirror of https://github.com/truenas/scale-build.git synced 2025-12-20 02:49:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

add get_env_variable function

Browse Source
This commit is contained in:
caleb
2022-04-04 07:47:13 -04:00
parent bc0a85345c
commit 7f6da092d0
1 changed files with 37 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
scale_build/config.py
View File

@@ -1,25 +1,41 @@
import os
import time
from os import environ, cpu_count
from time import time
from datetime import datetime
BUILD_TIME = int(time.time())
BUILD_TIME_OBJ = datetime.fromtimestamp(BUILD_TIME)
BUILDER_DIR = os.getenv('BUILDER_DIR', './')
BRANCH_OUT_NAME = os.getenv('NEW_BRANCH_NAME')
BRANCH_OVERRIDES = {k[:-(len('_OVERRIDE'))]: v for k, v in os.environ.items() if k.endswith('_OVERRIDE')}
FORCE_CLEANUP_WITH_EPOCH_CHANGE = bool(int(os.getenv('FORCE_CLEANUP_WITH_EPOCH_CHANGE') or '0'))
GITHUB_TOKEN = os.getenv('GITHUB_TOKEN')
PARALLEL_BUILD = int(os.getenv('PARALLEL_BUILDS', max(os.cpu_count(), 8) / 4))
PKG_DEBUG = bool(int(os.getenv('PKG_DEBUG') or '0'))
SIGNING_KEY = os.getenv('SIGNING_KEY')
SIGNING_PASSWORD = os.getenv('SIGNING_PASSWORD')
SKIP_SOURCE_REPO_VALIDATION = bool(int(os.getenv('SKIP_SOURCE_REPO_VALIDATION') or '0'))
TRAIN = os.getenv('TRUENAS_TRAIN')
TRUENAS_BRANCH_OVERRIDE = os.getenv('TRUENAS_BRANCH_OVERRIDE')
TRY_BRANCH_OVERRIDE = os.getenv('TRY_BRANCH_OVERRIDE')
if os.getenv('TRUENAS_VERSION'):
VERSION = os.getenv('TRUENAS_VERSION')
def get_env_variable(key, default_value, _type):
value = environ.get(key)
if value:
if _type == bool:
if value.isdigit():
return _type(value)
elif value.lower()[0] == 'y':
return True
elif value.lower()[0] == 'n':
return False
else:
VERSION = f'22.12-MASTER-{BUILD_TIME_OBJ.strftime("%Y%m%d-%H%M%S")}'
# the variable is set but is something randon (i.e. VAR='please')
# so just assume it's set
return True
else:
return _type(value)
else:
return _type(default_value)
BUILD_TIME = int(time())
BUILD_TIME_OBJ = datetime.fromtimestamp(BUILD_TIME)
BUILDER_DIR = get_env_variable('BUILDER_DIR', './', str)
BRANCH_OUT_NAME = get_env_variable('NEW_BRANCH_NAME', '', str)
BRANCH_OVERRIDES = {k[:-(len('_OVERRIDE'))]: v for k, v in environ.items() if k.endswith('_OVERRIDE')}
FORCE_CLEANUP_WITH_EPOCH_CHANGE = get_env_variable('FORCE_CLEANUP_WITH_EPOCH_CHANGE', 0, bool)
GITHUB_TOKEN = get_env_variable('GITHUB_TOKEN', '', str)
PARALLEL_BUILD = get_env_variable('PARALLEL_BUILDS', (max(cpu_count(), 8) / 4), int)
PKG_DEBUG = get_env_variable('PKG_DEBUG', 0, bool)
SIGNING_KEY = get_env_variable('SIGNING_KEY', '', str)
SIGNING_PASSWORD = get_env_variable('SIGNING_PASSWORD', '', str)
SKIP_SOURCE_REPO_VALIDATION = get_env_variable('SKIP_SOURCE_REPO_VALIDATION', 0, bool)
TRAIN = get_env_variable('TRUENAS_TRAIN', '', str)
TRUENAS_BRANCH_OVERRIDE = get_env_variable('TRUENAS_BRANCH_OVERRIDE', '', str)
TRY_BRANCH_OVERRIDE = get_env_variable('TRY_BRANCH_OVERRIDE', '', str)
VERSION = get_env_variable('TRUENAS_VERSION', f'22.12-MASTER-{BUILD_TIME_OBJ.strftime("%Y%m%d-%H%M%S")}', str)