do_branding command added to setup

1 files changed, 15 insertions, 0 deletions

diff --git a/pkg/branding/__init__.py b/pkg/branding/__init__.py
new file mode 100644
index 00000000..0bd6befb
--- /dev/null
+++ b/pkg/branding/__init__.py
@@ -0,0 +1,15 @@
+from .config import APP_BASE_NAME, APP_PREFIX, BRANDED_BUILD, BRANDED_OPTS
+
+
+def get_name():
+    if BRANDED_BUILD is True:
+        return APP_PREFIX + BRANDED_OPTS.get('short_name', 'name_unknown')
+    else:
+        return APP_BASE_NAME
+
+
+def get_shortname():
+    if BRANDED_BUILD is True:
+        return BRANDED_OPTS.get('short_name', 'name_unknown')
+
+__all__ = ['get_name']