150 lines
4.2 KiB
Python
150 lines
4.2 KiB
Python
# Safe Eyes is a utility to remind you to take break frequently
|
|
# to protect your eyes from eye strain.
|
|
|
|
# Copyright (C) 2017 Gobinath
|
|
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import gi
|
|
gi.require_version('Gdk', '3.0')
|
|
from gi.repository import Gdk, GLib
|
|
import babel.dates, os, errno, re, subprocess, threading, logging, locale
|
|
import pyaudio, wave
|
|
|
|
bin_directory = os.path.dirname(os.path.realpath(__file__))
|
|
home_directory = os.path.expanduser('~')
|
|
|
|
"""
|
|
Play the alert.mp3
|
|
"""
|
|
def play_notification():
|
|
logging.info("Playing audible alert")
|
|
CHUNK = 1024
|
|
|
|
try:
|
|
# Open the sound file
|
|
path = os.path.join(bin_directory, 'resource','alert.wav')
|
|
sound = wave.open(path, 'rb')
|
|
|
|
# Create a sound stream
|
|
wrapper = pyaudio.PyAudio()
|
|
stream = wrapper.open(format=wrapper.get_format_from_width(sound.getsampwidth()),
|
|
channels=sound.getnchannels(),
|
|
rate=sound.getframerate(),
|
|
output=True)
|
|
|
|
# Write file data into the sound stream
|
|
data = sound.readframes(CHUNK)
|
|
while data != '':
|
|
stream.write(data)
|
|
data = sound.readframes(CHUNK)
|
|
|
|
# Close steam
|
|
stream.stop_stream()
|
|
stream.close()
|
|
wrapper.terminate()
|
|
|
|
except Exception as e:
|
|
logging.warning('Unable to play audible alert')
|
|
logging.exception(e)
|
|
|
|
"""
|
|
Get system idle time in minutes.
|
|
Return the idle time if xprintidle is available, otherwise return 0.
|
|
"""
|
|
def system_idle_time():
|
|
try:
|
|
return int(subprocess.check_output(['xprintidle']).decode('utf-8')) / 60000 # Convert to minutes
|
|
except:
|
|
return 0
|
|
|
|
|
|
"""
|
|
Execute the function in a separate thread.
|
|
"""
|
|
def start_thread(target_function, **args):
|
|
thread = threading.Thread(target=target_function, kwargs=args)
|
|
thread.start()
|
|
|
|
|
|
"""
|
|
Execute the given function in main thread.
|
|
"""
|
|
def execute_main_thread(target_function, args=None):
|
|
if args:
|
|
GLib.idle_add(lambda: target_function(args))
|
|
else:
|
|
GLib.idle_add(lambda: target_function())
|
|
|
|
|
|
"""
|
|
Check for full-screen applications.
|
|
This method must be executed by the main thread. If not, it will cause to random failure.
|
|
"""
|
|
def is_active_window_skipped(skip_break_window_classes, take_break_window_classes, unfullscreen_allowed=False):
|
|
logging.info("Searching for full-screen application")
|
|
screen = Gdk.Screen.get_default()
|
|
active_xid = str(screen.get_active_window().get_xid())
|
|
cmdlist = ['xprop', '-root', '-notype','-id',active_xid, 'WM_CLASS', '_NET_WM_STATE']
|
|
|
|
try:
|
|
stdout = subprocess.check_output(cmdlist)
|
|
except subprocess.CalledProcessError:
|
|
logging.warning("Error in finding full-screen application")
|
|
pass
|
|
else:
|
|
if stdout:
|
|
is_fullscreen = 'FULLSCREEN' in stdout
|
|
# Extract the process name
|
|
process_names = re.findall('"(.+?)"', stdout)
|
|
if process_names:
|
|
process = process_names[1].lower()
|
|
if process in skip_break_window_classes:
|
|
return True
|
|
elif process in take_break_window_classes:
|
|
if is_fullscreen and unfullscreen_allowed:
|
|
try:
|
|
screen.get_active_window().unfullscreen()
|
|
except:
|
|
logging.error('Error in unfullscreen the window ' + process)
|
|
pass
|
|
return False
|
|
|
|
return is_fullscreen
|
|
|
|
|
|
"""
|
|
Format time based on the system time.
|
|
"""
|
|
def format_time(time):
|
|
locale.setlocale(locale.LC_ALL, '')
|
|
system_locale = locale.getlocale(locale.LC_TIME)[0]
|
|
if not system_locale:
|
|
system_locale = 'en_US.UTF-8'
|
|
return babel.dates.format_time(time, format='short', locale=system_locale)
|
|
|
|
|
|
"""
|
|
Create directory if not exists.
|
|
"""
|
|
def mkdir(path):
|
|
try:
|
|
os.makedirs(path)
|
|
except OSError as exc:
|
|
if exc.errno == errno.EEXIST and os.path.isdir(path):
|
|
pass
|
|
else:
|
|
logging.error('Error while creating ' + str(path))
|
|
raise
|