1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
# utils.py -- utility functions source
# This file is part of redshift-ng.
# redshift-ng 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.
# redshift-ng 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 redshift-ng. If not, see <http://www.gnu.org/licenses/>.
# Copyright (c) 2010 Francesco Marella <francesco.marella@gmail.com>
# Copyright (c) 2011 Jon Lund Steffensen <jonlst@gmail.com>
import ctypes
import os
import sys
try:
from xdg import BaseDirectory
from xdg import DesktopEntry
has_xdg = True
except ImportError:
has_xdg = False
REDSHIFT_DESKTOP = 'redshift-gtk.desktop'
# Keys to set when enabling/disabling autostart.
# Only first one is checked on "get".
AUTOSTART_KEYS = (('Hidden', ('true', 'false')),
('X-GNOME-Autostart-enabled', ('false', 'true')))
def open_autostart_file():
autostart_dir = BaseDirectory.save_config_path("autostart")
autostart_file = os.path.join(autostart_dir, REDSHIFT_DESKTOP)
if not os.path.exists(autostart_file):
desktop_files = list(
BaseDirectory.load_data_paths(
"applications", REDSHIFT_DESKTOP))
if not desktop_files:
raise IOError("Installed redshift desktop file not found!")
desktop_file_path = desktop_files[0]
# Read installed file
dfile = DesktopEntry.DesktopEntry(desktop_file_path)
for key, values in AUTOSTART_KEYS:
dfile.set(key, values[False])
dfile.write(filename=autostart_file)
else:
dfile = DesktopEntry.DesktopEntry(autostart_file)
return dfile, autostart_file
def supports_autostart():
return has_xdg
def get_autostart():
if not has_xdg:
return False
dfile, path = open_autostart_file()
check_key, check_values = AUTOSTART_KEYS[0]
return dfile.get(check_key) == check_values[True]
def set_autostart(active):
if not has_xdg:
return
dfile, path = open_autostart_file()
for key, values in AUTOSTART_KEYS:
dfile.set(key, values[active])
dfile.write(filename=path)
def setproctitle(title):
"""Set process title."""
title_bytes = title.encode(sys.getdefaultencoding(), 'replace')
buf = ctypes.create_string_buffer(title_bytes)
if 'linux' in sys.platform:
try:
libc = ctypes.cdll.LoadLibrary("libc.so.6")
except OSError:
return
try:
libc.prctl(15, buf, 0, 0, 0)
except AttributeError:
return # Strange libc, just skip this
elif 'bsd' in sys.platform:
try:
libc = ctypes.cdll.LoadLibrary("libc.so.7")
except OSError:
return
try:
libc.setproctitle(ctypes.create_string_buffer(b"-%s"), buf)
except AttributeError:
return
|
