mirror of
https://github.com/captn3m0/Scripts.git
synced 2024-09-27 22:22:53 +00:00
203 lines
5.4 KiB
Python
Executable File
203 lines
5.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Source: https://github.com/rkashapov/i3blocks-pomodoro
|
|
#
|
|
# MIT License
|
|
#
|
|
# Copyright (c) 2019 Rustam Kashapov
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
# of this software and associated documentation files (the "Software"), to deal
|
|
# in the Software without restriction, including without limitation the rights
|
|
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
# copies of the Software, and to permit persons to whom the Software is
|
|
# furnished to do so, subject to the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be included in all
|
|
# copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
# SOFTWARE.
|
|
|
|
"""
|
|
Simple Pomodoro app inspired by spt project.
|
|
The app designed to be used with i3blocks.
|
|
"""
|
|
|
|
import errno
|
|
import os
|
|
import socket
|
|
from itertools import cycle
|
|
from subprocess import call
|
|
from time import sleep
|
|
from threading import Thread
|
|
|
|
|
|
def notify(text):
|
|
call(["notify-send", "Pomodoro", text])
|
|
|
|
|
|
def daemonize():
|
|
if os.fork() > 0:
|
|
exit()
|
|
os.setsid()
|
|
if os.fork() > 0:
|
|
exit()
|
|
|
|
|
|
class AlredyStartedError(Exception):
|
|
pass
|
|
|
|
|
|
class Cycle:
|
|
def __init__(self, items):
|
|
self._items = items
|
|
self._iter = None
|
|
self.reset()
|
|
|
|
def reset(self):
|
|
self._iter = cycle(self._items)
|
|
|
|
def __iter__(self):
|
|
return self
|
|
|
|
def __next__(self):
|
|
return next(self._iter)
|
|
|
|
|
|
class Pomodoro:
|
|
def __init__(self, cycle):
|
|
self._cycle = cycle
|
|
self._state = "Paused"
|
|
self._remaining = 0
|
|
self._do_reset = False
|
|
self._running = False
|
|
|
|
@property
|
|
def state(self):
|
|
state = self._state
|
|
if not self._running:
|
|
state = "Paused"
|
|
minutes, seconds = divmod(self._remaining, 60)
|
|
return "{0} {1:02d}:{2:02d}".format(state, minutes, seconds)
|
|
|
|
def toggle(self):
|
|
self._running = not self._running
|
|
|
|
def reset(self):
|
|
self._do_reset = True
|
|
self._cycle.reset()
|
|
|
|
def start(self):
|
|
for timeout, state, message in self._cycle:
|
|
if self._do_reset:
|
|
self._do_reset = False
|
|
|
|
if self._running:
|
|
self._state = state
|
|
notify(message)
|
|
|
|
self._remaining = timeout * 60
|
|
while self._remaining:
|
|
if self._do_reset:
|
|
break
|
|
sleep(1)
|
|
if self._running:
|
|
self._remaining -= 1
|
|
self._state = state
|
|
|
|
|
|
class App:
|
|
ADDRESS = '/tmp/pomodoro.sock'
|
|
PAUSE = b"pause"
|
|
RESET = b"reset"
|
|
DISPLAY = b"display"
|
|
|
|
def __init__(self, pomodoro):
|
|
self._thread = Thread(target=self._handle_connections)
|
|
self._sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
|
self._pomodoro = pomodoro
|
|
|
|
def run(self):
|
|
try:
|
|
self._sock.bind(self.ADDRESS)
|
|
except socket.error as e:
|
|
if e.errno != errno.EADDRINUSE:
|
|
raise e
|
|
|
|
try:
|
|
self._sock.connect(self.ADDRESS)
|
|
except socket.error as err:
|
|
if err.errno != errno.ECONNREFUSED:
|
|
raise e
|
|
os.unlink(self.ADDRESS)
|
|
self._sock.bind(self.ADDRESS)
|
|
else:
|
|
raise AlredyStartedError()
|
|
|
|
daemonize()
|
|
|
|
Thread(target=self._pomodoro.start).start()
|
|
self._handle_connections()
|
|
|
|
def display(self):
|
|
self._send(self.DISPLAY)
|
|
|
|
def toggle(self):
|
|
self._send(self.PAUSE)
|
|
|
|
def reset(self):
|
|
self._send(self.RESET)
|
|
|
|
def _send(self, command):
|
|
self._sock.sendall(command)
|
|
print(self._sock.recv(32).decode("utf-8"))
|
|
|
|
def _handle_connections(self):
|
|
self._sock.listen()
|
|
while True:
|
|
conn, _ = self._sock.accept()
|
|
command = conn.recv(16)
|
|
if command == self.PAUSE:
|
|
self._pomodoro.toggle()
|
|
elif command == self.RESET:
|
|
self._pomodoro.reset()
|
|
|
|
conn.sendall(self._pomodoro.state.encode("utf-8"))
|
|
conn.close()
|
|
|
|
|
|
app = App(
|
|
Pomodoro(
|
|
Cycle(
|
|
[
|
|
(25, "Working", "Time to start working!"),
|
|
(5, "Resting", "Time to start resting!"),
|
|
(25, "Working", "Time to start working!"),
|
|
(5, "Resting", "Time to start resting!"),
|
|
(25, "Working", "Time to start working!"),
|
|
(5, "Resting", "Time to start resting!"),
|
|
(25, "Working", "Time to start working!"),
|
|
(15, "Long Break", "Time to take some nap!"),
|
|
]
|
|
)
|
|
)
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
app.run()
|
|
except AlredyStartedError:
|
|
button = os.getenv("BLOCK_BUTTON", "").lower()
|
|
if button == "1":
|
|
app.toggle()
|
|
elif button == "3":
|
|
app.reset()
|
|
else:
|
|
app.display()
|