import sys from enum import Enum import i3_lemonbar_config as conf import i3_lemonbar_common as common displays = '' workspaces = '' win_title = '' sys_load = '' net_load = '' volume = '' brightness = '' battery = '' date = '' language = '' time = '' response = '' blank = ' ' sr = conf.sep_right slr = conf.sep_l_right sl = conf.sep_left sll = conf.sep_l_left def block(fg = None, bg = None, font = None, click = None, append=''): # Remeber that clickable blocks need to be closed rtn = [] if click is not None: if click == '': rtn.append('A') else: rtn.append('A:'+click+':') if fg is not None: rtn.append('F'+fg) if bg is not None: rtn.append('B'+bg) if font is not None: rtn.append('T'+font) return ''.join(['%{', ' '.join(rtn), '}', append]) reset = block(fg='-', bg='-') reset_power = block(fg='-', bg=conf.color_poweropts) class COLOR_SCHEME(Enum): A1 = (conf.color_sec_b1, conf.color_fore, conf.color_icon) A2 = (conf.color_sec_b2, conf.color_fore, conf.color_icon) A1_INA = (conf.color_sec_b1, conf.color_disable, conf.color_disable) A2_INA = (conf.color_sec_b2, conf.color_disable, conf.color_disable) CPU_ALERT= (conf.color_cpu, conf.color_back, conf.color_back) NET_ALERT= (conf.color_net, conf.color_back, conf.color_back) SPECIAL = (conf.color_head, conf.color_back, conf.color_back) def __init__(self, back_color, text_color, icon_color): self.back_color = back_color self.text_color = text_color self.icon_color = icon_color def single_sect(icon='', text='', click=None, alt=COLOR_SCHEME.A1): # If there is an action we need to close the block close = block(click='') if click is not None else '' if icon != '': icon = ' ' + icon return ' '.join([block(fg=alt.back_color, append=sl) , block(click=click, fg=alt.icon_color, bg=alt.back_color , font='2', append=icon) , block(fg=alt.text_color, font='1', append=text) , close]) def double_sect(text1 = '', text2 = '', icon1 = '', icon2 = '', click = None , alt=COLOR_SCHEME.A1): # If text_color None then icon_color will apply close = block(click='') if click is not None else '' return ''.join([block(fg=alt.back_color, append=sl), ' ' , block(click=click, fg=alt.icon_color, bg=alt.back_color, font='2') , ' ', icon1, block(fg=alt.text_color, font='1'), ' ', text1 , block(fg=alt.icon_color), sll, block(font='2') , icon2, block(fg=alt.text_color, font='1'), ' ', text2, ' ', close]) def control_sect(sec_color=None, head='', buttons=[], actions=[]): rtn = head for button, action in zip(buttons, actions): # If there is an action we need to close the block close = block(click='') if action is not None else '' rtn += ' '.join([block(click=action), button, close]) return rtn # Constants power_opts = ''.join([block(fg=conf.color_fore, bg=conf.color_poweropts) , ' Abort (Esc) | System (l) lock, (e) logout, (s) suspend, (h) hibernate' , ', (r) reboot, (Shift+s) shutdown']) controls = ' '.join([block(fg=conf.color_head, bg=conf.color_sec_b2) , conf.sep_right, block(fg=conf.color_head, bg=conf.color_sec_b2, click='mode cycle') , conf.icon_prog , block(fg=conf.color_sec_b2, bg='-', click='') , control_sect(head='', buttons=['on', 'off'] , actions=['bluetooth power on', 'bluetooth power off']) , control_sect(head='PXC 550', buttons=['conn.', 'disc.'] , actions=['bluetooth connect pxc550', 'bluetooth disconnect pxc550']) ]) def update_response(data): global response resp = ' '.join(data) #common.logger.debug('Got response {}'.format(resp)) response= single_sect(text=resp) def update_displays(data): global displays dsp_array = data[0].split(':') parsed_list = [block(click='displays', font='2')] for dsp in dsp_array[1:]: if dsp == 'eDP1': col_head = conf.color_head elif dsp == 'DP1': col_head = conf.color_vga elif dsp == 'HDMI2': col_head = conf.color_hdmi else: col_head = '#00000000' # Undefined parsed_list.append(block(fg=conf.color_back, bg=col_head)) parsed_list.append(conf.icon_wsp) parsed_list.append(block(click='')) displays = ' '.join(parsed_list) def update_workspaces(data): global workspaces prefix = block(font='1', fg=conf.color_back, bg=conf.color_head) prefix_foc = ''.join([block(fg = conf.color_head, bg=conf.color_wsp) , conf.sep_right , block(fg=conf.color_back, bg=conf.color_wsp, font='1')]) prefix_ina = block(fg=conf.color_back, bg=conf.color_head, font='1') wspces = [] for entry in data: # entry for example FOC5___terms status = entry[0:3] # FOC or INA num = entry[3] name = entry[7:] full_name = ' '.join(['', num, name]) current = ''.join([block(click=('i3-msg workspace' + full_name)) , full_name, block(click='')]) if status == "FOC": wspces.append(''.join([prefix_foc, current])) else: wspces.append(''.join([prefix_ina, current])) workspaces = ''.join([prefix, ' '.join(wspces)]) def update_lang(data): global language language = single_sect(icon=conf.icon_lang, click='lang', text=data[0] , alt=COLOR_SCHEME.A2) def update_bright(data): global brightness brtxt = str(int(float(data[0]))) brightness = single_sect(icon=conf.icon_bright, click='adj_br' , text=(brtxt+'%'), alt=COLOR_SCHEME.A1) def update_net(data): global net_load if data[0] == 'down': # wlan (wland_v, wlanu_v) = ('x', 'x') wlan_alt = COLOR_SCHEME.A2_INA else: (wland_v, wlanu_v) = (data[0],data[1]) if max(float(wland_v), float(wlanu_v)) > float(conf.net_alert): wlan_alt = COLOR_SCHEME.NET_ALERT else: wlan_alt = COLOR_SCHEME.A2 if data[2] == 'down': # eth (ethd_v, ethu_v) = ('x', 'x') eth_alt = COLOR_SCHEME.A1_INA else: (ethd_v, ethu_v) = (data[2],data[3]) if max(float(ethd_v), float(ethu_v)) > float(conf.net_alert): eth_alt = COLOR_SCHEME.NET_ALERT else: eth_alt = COLOR_SCHEME.A1 net_load = ''.join([ double_sect(click = 'wlan', alt=wlan_alt, text1 = wland_v, text2 = wlanu_v, icon1 = (conf.icon_wlan + conf.icon_dl), icon2 = conf.icon_ul) , double_sect(click = 'eth', alt=eth_alt, text1 = ethd_v, text2 = ethu_v, icon1 = (conf.icon_eth + conf.icon_dl), icon2 = conf.icon_ul) ]) # wlan_d wlan_u eth_d eth_u def update_title(data): global win_title win_title = ' '.join([block(fg=conf.color_head, bg=conf.color_sec_b2) , conf.sep_right, block(fg=conf.color_head, bg=conf.color_sec_b2, click='mode cycle') , conf.icon_prog , block(fg=conf.color_sec_b2, bg='-') , ' '.join(data)]) def format_line(): # Need to end with a reset block, otherwise the color fills the %{r} spacer if common.mode == common.bar_mode.normal: return ''.join(['%{l}', reset, displays, workspaces, win_title, '%{r}', sys_load, net_load , volume, brightness, battery, date, language, time, reset]) elif common.mode == common.bar_mode.power: return ''.join(['%{l}', reset, power_opts, '%{r}', sys_load, net_load , volume, brightness, battery, date, language, time, reset_power]) elif common.mode == common.bar_mode.control: return ''.join(['%{l}', reset, displays, workspaces, controls, '%{r}', response, time, reset]) else: return ''.join(['%{l}', reset, displays, workspaces, 'Not normal', '%{r}', time, reset]) parsers_dict = { 'WSP':update_workspaces ,'LANG':update_lang ,'BRIGHT':update_bright ,'WIN':update_title ,'DISP':update_displays ,'RESP':update_response } def parse_line(line_in): ''' Lines are CNK_FAST Fri 21 Sep 19:45:18 VolXXX wlan_d wlan_u eth_d eth_u CNK_SLOW cpu mem disk_root disk_home batt bri lang disp WSPINA1___main INA2___web FOC5___terms INA6___stats ''' try: for key,func in parsers_dict.items(): l = len(key) if line_in[:l] == key: func(line_in[l:].split()) break except: print('Exception occured\n Line in: {}\n Data: {}'.format(line_in, line_in[l:].split())) raise formatted_line = format_line() return formatted_line if __name__ == "__main__": for line in sys.stdin: print(parse_line(line))