initial commit

config/lemonbar/i3_workspaces.py

@@ -0,0 +1,102 @@
+#!/usr/bin/env python
+# 
+# Print i3 workspaces on every change.
+# 
+# Format: 
+#   For every workspace (x = workspace name)
+#       - "FOCx" -> Focused workspace
+#       - "INAx" -> Inactive workspace
+#       - "ACTx" -> Ative workspace
+#       - "URGx" -> Urgent workspace
+#
+# Uses i3py.py -> https://github.com/ziberna/i3-py
+# Based in wsbar.py en examples dir
+#
+# 16 feb 2015 - Electro7
+
+
+import sys
+import time
+import subprocess
+
+import i3
+
+class State(object):
+    # workspace states
+    focused = 'FOC'
+    active = 'ACT'
+    inactive = 'INA'
+    urgent = 'URG'
+    
+    def get_state(self, workspace, output):
+        if workspace['focused']:
+            if output['current_workspace'] == workspace['name']:
+                return self.focused
+            else:
+                return self.active
+        if workspace['urgent']:
+            return self.urgent
+        else:
+            return self.inactive
+
+
+class i3ws(object):
+    ws_format = '%s%s '
+    end_format = 'WSP%s'
+    state = State()
+    
+    def __init__(self, state=None):
+        if state:
+            self.state = state
+        # socket
+        self.socket = i3.Socket()
+        # Output to console
+        workspaces = self.socket.get('get_workspaces')
+        outputs = self.socket.get('get_outputs')
+        self.display(self.format(workspaces, outputs))
+        # Subscribe to an event
+        callback = lambda data, event, _: self.change(data, event)
+        self.subscription = i3.Subscription(callback, 'workspace')
+    
+    def change(self, event, workspaces):
+        # Receives event and workspace data
+        if 'change' in event:
+            outputs = self.socket.get('get_outputs')
+            text = self.format(workspaces, outputs)
+            self.display(text)
+    
+    def format(self, workspaces, outputs):
+        # Formats the text according to the workspace data given.
+        out = '' 
+        for workspace in workspaces:
+            output = None
+            for output_ in outputs:
+                if output_['name'] == workspace['output']:
+                    output = output_
+                    break
+            if not output:
+                continue
+            st = self.state.get_state(workspace, output)
+            name = workspace['name']
+            item= self.ws_format % (st, name)
+            out += item
+        return self.end_format % out
+    
+    def display(self, text):
+        # Displays the text in stout
+        print(text)
+        sys.stdout.flush()
+    
+    def quit(self):
+        # Quits the i3ws; closes the subscription and terminates
+        self.subscription.close()
+
+if __name__ == '__main__':
+    ws = i3ws()
+    try:
+        while True:
+            time.sleep(1)
+    except KeyboardInterrupt:
+        print('')  # force new line
+    finally:
+        ws.quit()