aboutsummaryrefslogtreecommitdiff
path: root/lmlib.py
blob: 2b0d6f9e054b07cc6265ecbcfac96543d0cd0858 (plain)
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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
#!/usr/bin/env python3
# File: lmlib.py
# License: CC-BY-SA 4.0
# Author: bgstack15
# Startdate: 2019-06-12
# Title: Python libs for logout-manager
# Purpose: Store the common elements for operating a logout-manager
# History:
# Usage:
#    In a logout-manager-gtk.py program
# Reference:
#    platform info https://stackoverflow.com/questions/110362/how-can-i-find-the-current-os-in-python/10091465#10091465
# Improve:
# Documentation:

import configparser, platform

logout_manager_version="2019-06-14b"

class Actions:

   @staticmethod
   def hibernate(config, event=None):
      #print("need to run the /sys/power/state trick, if available")
      print(config.get_hibernate_command())

   @staticmethod
   def lock(config, event=None):
      #print("please lock the screen.")
      print(config.get_lock_command())

   @staticmethod
   def logout(config, event=None):
      #print("please log out of current session!")
      print(config.get_logout_command())

   @staticmethod
   def reboot(config, event=None):
      #print("please reboot.")
      print(config.get_reboot_command())

   @staticmethod
   def shutdown(config, event=None):
      #print("please shut yourself down!")
      print(config.get_shutdown_command())

class Config:
   def __init__(self):
      # load defaults which can be overwritten
      self.hibernate_command = ""
      self.lock_command = ""
      self.logout_command = ""
      self.reboot_command = ""
      self.shutdown_command = ""
      self.hibernate_icon = "system-hibernate"
      self.hibernate_fallback_icon = "system-hibernate"
      self.lock_icon = "system-lock-screen"
      self.lock_fallback_icon = "system-lock-screen"
      self.logout_icon = "system-log-out"
      self.logout_fallback_icon = "system-log-out"
      self.reboot_icon = "system-reboot"
      self.reboot_fallback_icon = "system-reboot"
      self.shutdown_icon = "system-shutdown"
      self.shutdown_fallback_icon = "system-shutdown"
      self.icon_size = 24
      self.icon_theme = "default"
      self.icon_category = "actions"
      self.can_hibernate = False
      self.application_icon=("system-log-out")

   def set_hibernate_command(self,hibernate_command):
      self.hibernate_command = hibernate_command

   def set_lock_command(self,lock_command):
      self.lock_command = lock_command

   def set_logout_command(self,logout_command):
      self.logout_command = logout_command

   def set_reboot_command(self,reboot_command):
      self.reboot_command = reboot_command

   def set_shutdown_command(self,shutdown_command):
      self.shutdown_command = shutdown_command

   def set_hibernate_icon(self,hibernate_icon):
      self.hibernate_icon = hibernate_icon

   def set_lock_icon(self,lock_icon):
      self.lock_icon = lock_icon

   def set_logout_icon(self,logout_icon):
      self.logout_icon = logout_icon

   def set_reboot_icon(self,reboot_icon):
      self.reboot_icon = reboot_icon

   def set_shutdown_icon(self,shutdown_icon):
      self.shutdown_icon = shutdown_icon

   def set_icon_size(self,icon_size):
      self.icon_size = int(icon_size)

   def set_icon_theme(self,icon_theme):
      self.icon_theme = icon_theme

   def set_icon_category(self,icon_category):
      self.icon_category = icon_category

   def set_can_hibernate(self,can_hibernate):
      print("Setting can_hibernate:",can_hibernate)
      self.can_hibernate = bool(can_hibernate)

   def get_hibernate_command(self):
      return self.hibernate_command

   def get_lock_command(self):
      return self.lock_command

   def get_logout_command(self):
      return self.logout_command

   def get_reboot_command(self):
      return self.reboot_command

   def get_shutdown_command(self):
      return self.shutdown_command

   def get_hibernate_icon(self):
      return self.hibernate_icon

   def get_lock_icon(self):
      return self.lock_icon

   def get_logout_icon(self):
      return self.logout_icon

   def get_reboot_icon(self):
      return self.reboot_icon

   def get_shutdown_icon(self):
      return self.shutdown_icon

   def get_hibernate_fallback_icon(self):
      return self.hibernate_fallback_icon

   def get_lock_fallback_icon(self):
      return self.lock_fallback_icon

   def get_logout_fallback_icon(self):
      return self.logout_fallback_icon

   def get_reboot_fallback_icon(self):
      return self.reboot_fallback_icon

   def get_shutdown_fallback_icon(self):
      return self.shutdown_fallback_icon

   def get_icon_size(self):
      return self.icon_size

   def get_icon_theme(self):
      return self.icon_theme

   def get_icon_category(self):
      return self.icon_category

   def get_can_hibernate(self):
      return self.can_hibernate

def Initialize_config():
   # Read config
   config_in = configparser.ConfigParser()
   config_in.read('logout-manager.conf')
   config = Config()
   try:
      ci = config_in['logout-manager']
   except:
      # no definition
      print("Using default commands")

   try:
      ci_icons = config_in['icons']
   except:
      # no definition
      print("Using default icons")

   # load up our custom class, which stores the defaults in case we do not set them here
   if 'hibernate_command' in ci:
      config.set_hibernate_command(ci['hibernate_command'])
   if 'lock_command' in ci:
      config.set_lock_command(ci['lock_command'])
   if 'logout_command' in ci:
      config.set_logout_command(ci['logout_command'])
   if 'reboot_command' in ci:
      config.set_reboot_command(ci['reboot_command'])
   if 'shutdown_command' in ci:
      config.set_shutdown_command(ci['shutdown_command'])
   if 'hibernate' in ci_icons:
      config.set_hibernate_icon(ci_icons['hibernate'])
   if 'lock' in ci_icons:
      config.set_lock_icon(ci_icons['lock'])
   if 'logout' in ci_icons:
      config.set_logout_icon(ci_icons['logout'])
   if 'reboot' in ci_icons:
      config.set_reboot_icon(ci_icons['reboot'])
   if 'shutdown' in ci_icons:
      config.set_shutdown_icon(ci_icons['shutdown'])
   if 'size' in ci_icons:
      config.set_icon_size(ci_icons['size'])
   if 'theme' in ci_icons:
      config.set_icon_theme(ci_icons['theme'])
   # store the info about if hibernate is an option
   can_hibernate = False
   try:
      with open('/sys/power/state') as r:
         line = r.read()
         if 'disk' in line: can_hibernate = True
   except:
      pass
   config.set_can_hibernate(can_hibernate)

   # set icon category
   # written primarily for el7 which uses "app" for the system-reboot icons, etc.
   a = platform.dist()
   try:
      # WORKHERE: after moving in the gtk3 default icon theme check to lmlib, do a check here, for of the theme=default and gtk3_default=hicolor, to make category=apps
      if a[0] == "redhat" and int(a[1].split(".")[0]) <= 7:
         config.set_icon_category("apps")
   except:
      pass

   # DEBUG, raw from conf file and system status
   print("Raw values:")
   for item in config_in.sections():
      print("["+item+"]")
      for key in config_in[item]:
         print(key+" = "+config_in[item][key])
   print("Can hibernate:",can_hibernate)

   # DEBUG, stored values
   print("Stored values:")
   print(config.get_hibernate_command())
   print(config.get_lock_command())
   print(config.get_logout_command())
   print(config.get_reboot_command())
   print(config.get_shutdown_command())
   print(config.get_hibernate_icon())
   print(config.get_lock_icon())
   print(config.get_logout_icon())
   print(config.get_reboot_icon())
   print(config.get_shutdown_icon())
   print(config.get_icon_size())
   print(config.get_icon_theme())
   print(config.get_icon_category())
   print("Can hibernate:",config.get_can_hibernate())

   return config
bgstack15