nodeocc 1.0.1__py3-none-any.whl

view/styles.py

@@ -0,0 +1,76 @@
+def _format_to(st, width, dir='left'):
+        ts = str(st)[:width]
+        if dir == 'left':
+            return ts + ' ' * (width - len(ts))
+        else:
+            return ' ' * (width - len(ts)) + ts
+
+def cmdstyle(color, string):
+    """
+    style formatting function for unix terminals
+    """
+    strings = string.split('\n')
+    styles = {
+        'RED' : lambda x: '\033[31m' + str(x) + '\033[0m',
+        'YELLOW' : lambda x: '\033[33m' + str(x) + '\033[0m',
+        'GREEN' : lambda x: '\033[32m' + str(x) + '\033[0m',
+        'MAGENTA' : lambda x: '\033[35m' + str(x) + '\033[0m',
+        'BLUE' : lambda x: '\033[34m' + str(x) + '\033[0m',
+
+        'LRED' : lambda x: '\033[91m' + str(x) + '\033[0m',
+        'LYELLOW' : lambda x: '\033[93m' + str(x) + '\033[0m',
+        'LGREEN' : lambda x: '\033[92m' + str(x) + '\033[0m',
+        'LMAGENTA' : lambda x: '\033[95m' + str(x) + '\033[0m',
+        'LBLUE' : lambda x: '\033[94m' + str(x) + '\033[0m',
+        'LCYAN' : lambda x: '\033[96m' + str(x) + '\033[0m',
+        'LWHITE' : lambda x: '\033[97m' + str(x) + '\033[0m',
+
+        'CYAN' : lambda x: '\033[36m' + str(x) + '\033[0m',
+        'BG_CYAN': lambda x: '\033[33m' + str(x) + '\033[0m',
+        'WHITE' : lambda x: '\033[37m' + str(x) + '\033[0m',
+
+        'BG_RED': lambda x: '\033[41m\033[37m' + str(x) + '\033[0m',
+        'BG_GREEN': lambda x: '\033[42m\033[37m' + str(x) + '\033[0m',
+        'BG_YELLOW': lambda x: '\033[43m' + str(x) + '\033[0m',
+        'BG_MAGENTA': lambda x: '\033[45m\033[37m' + str(x) + '\033[0m'
+
+    }
+    return '\n'.join([styles[color](x) for x in strings])
+
+def crsstyler(color, string):
+    """
+    custom style formatting function for curses applications
+    """
+    strings = string.split('\n')
+    styles = {
+        'RED' : lambda x: '<*3~' + str(x) + ':*>',
+        'YELLOW' : lambda x: '<*4~' + str(x) + ':*>',
+        'GREEN' : lambda x: '<*5~' + str(x) + ':*>',
+        'MAGENTA' : lambda x: '<*6~' + str(x) + ':*>',
+        'BLUE' : lambda x: '<*7~' + str(x) + ':*>',
+
+        'LRED' : str,
+        'LYELLOW' : str,
+        'LGREEN' : str,
+        'LMAGENTA' : str,
+        'LBLUE' : str,
+        'LCYAN' : str,
+        'LWHITE' : str,
+        'LBG_CYAN' : str,
+
+        'CYAN' : lambda x: '<*8~' + str(x) + ':*>',
+        'BG_CYAN': lambda x: '<*15~' + str(x) + ':*>',
+        'WHITE' : lambda x: '<*2~' + str(x) + ':*>',
+
+        'BG_RED':       lambda x: '<*10~' + str(x) + ':*>',
+        'BG_GREEN':     lambda x: '<*11~' + str(x) + ':*>',
+        'BG_YELLOW':    lambda x: '<*12~' + str(x) + ':*>',
+        'BG_MAGENTA':   lambda x: '<*13~' + str(x) + ':*>'
+    }
+    return '\n'.join([styles[color](x) for x in strings])
+
+def mockstyler(color, string):
+    """
+    mock style formatting function for debug purposes
+    """
+    return string

view/utils.py

@@ -0,0 +1,60 @@
+import numpy as np
+import re
+
+
+def is_student_viz(user):
+    return user.user_group in ('studenti', 'tesisti') and not is_cvcs_viz(user)
+
+
+def is_cvcs_viz(user):
+    return re.search(r'cvcs|ai4a2023', user.account.lower()) is not None
+
+
+def is_dev(job):
+    return job.partition == 'all_serial'
+
+
+def is_prod(job):
+    return job.partition == 'all_usr_prod'
+
+
+def to_datetime(time_str):
+    return np.datetime64(re.match(r'\d+\s+(.*)\n', str(time_str)).group(1))
+
+
+def to_datetime(time_str):
+    return np.datetime64(re.match(r'\d+\s+(.*)\n', str(time_str)).group(1))
+
+
+def maintenance_status(infrastructure):
+    onmain = False
+    waitString = ''
+    if len(infrastructure.maintenances):
+        # time format: '0   2023-12-04 14:00:00\nName: StartTime, dtype: datetime64[ns]'
+        next_maintenance = sorted(infrastructure.maintenances, key=lambda x: to_datetime(x['start_time']))[0]
+        start_time = to_datetime(next_maintenance['start_time'])
+        end_time = to_datetime(next_maintenance['end_time'])
+        time_to_maintenance = (start_time - np.datetime64('now')).astype(float)
+        # TODO: fix timezone
+        if time_to_maintenance < 0 and (end_time - np.datetime64('now')) > 0:
+            onmain = True
+        else:
+            tt_d = int(time_to_maintenance / (60 * 60 * 24))
+            tt_h = int(time_to_maintenance % (60 * 60 * 24) / (60 * 60))
+            tt_m = int(time_to_maintenance % (60 * 60) / 60)
+            if tt_d > 0:
+                waitString = '%dd' % tt_d
+            elif tt_h > 0:
+                waitString = '%dh' % tt_h
+            else:
+                waitString = '%dm' % tt_m
+    return onmain, waitString
+
+
+numfont = '⁰¹²³⁴⁵⁶⁷⁸⁹'
+
+
+def to_font(num):
+    if num < 0:
+        return '⁻' + to_font(-num)
+    return ''.join([numfont[int(i)] for i in str(num)])