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
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
|
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
'''
This script modify the "Startup Timestamp", "Shutdown Timestamp" and
"End Status" on Tuptime database preserving the other values around in sync.
Increase 60 secs the startup datetime on register number 1:
tuptime_modify.py -c startup -r 1 -s 60
Decrease 100 secs the startup datetime on register number 4 on other file:
tuptime_modify.py -c startup -r 4 -s -100 -f /tmp/test.db
Increase 30 secs the shutdown datetime on register number 12:
tuptime_modify.py -c shutdown -r 12 -s 60
Decrease 300 secs the shutdown datetime on register number 47 with verbose:
tuptime_modify.py -c shutdown -r 47 -s -300 -v
Swich end status value on register 54:
tuptime_modify.py -c endst -r 54
'''
import sys, argparse, locale, signal, logging, sqlite3, tempfile
from shutil import copyfile
DB_FILE = '/var/lib/tuptime/tuptime.db'
DATE_FORMAT = '%X %x'
__version__ = '1.0.1'
# Terminate when SIGPIPE signal is received
signal.signal(signal.SIGPIPE, signal.SIG_DFL)
# Set locale to the users default settings (LANG env. var)
locale.setlocale(locale.LC_ALL, '')
def get_arguments():
"""Get arguments from command line"""
parser = argparse.ArgumentParser()
parser.add_argument(
'-c', '--change',
dest='change',
default=False,
action='store',
type=str,
required=True,
choices=['startup', 'shutdown', 'endst'],
help='change startup or shutdown datetime [<startup|shutdown|endst>]'
)
parser.add_argument(
'-f', '--filedb',
dest='db_file',
default=DB_FILE,
action='store',
help='database file (' + DB_FILE + ')',
metavar='FILE'
)
parser.add_argument(
'-n', '--nobackup',
dest='nobackup',
default=False,
action='store_true',
help='avoid create backup db file'
)
parser.add_argument(
'-r', '--register',
dest='register',
default=False,
action='store',
type=int,
required=True,
help='startup register to modify'
)
parser.add_argument(
'-s', '--seconds',
dest='seconds',
default=0,
action='store',
type=int,
help='seconds to add or remove (+/-)'
)
parser.add_argument(
'-v', '--verbose',
dest='verbose',
default=False,
action='store_true',
help='verbose output'
)
arg = parser.parse_args()
if arg.verbose:
logging.basicConfig(format='%(levelname)s:%(message)s', level=logging.DEBUG)
logging.info('Version: %s', (__version__))
if arg.change == 'endst' and arg.seconds:
parser.error('Operator \"seconds\" can\'t be combined with \"endst\"')
logging.info('Arguments: %s', vars(arg))
return arg
def backup_dbf(arg):
"""Backup db file before modify"""
try:
tmp_file = tempfile.NamedTemporaryFile(prefix="tuptime_", suffix=".db").name
copyfile(arg.db_file, tmp_file)
print('Backup file:\t' + tmp_file)
except Exception as exp:
logging.error('Can\'t create backup file. %s', exp)
sys.exit(1)
def fix_endst(arg, reg, conn, modt, orgt):
"""Swich end status register"""
# Swich between values 1 to 0 and 0 to 1
modt['endst'] = 1 - orgt['endst']
print('\t modified\tendst: ' + str(modt['endst']))
# Update values
conn.execute('update tuptime set endst =? where rowid =?', (modt['endst'], reg['target']))
def fix_shutdown(arg, reg, conn, modt, orgt):
"""Modify shutdown datetime register"""
# Last row have None values
if orgt['offbtime'] is None and orgt['downtime'] is None:
modt['offbtime'] = None
modt['downtime'] = None
else:
modt['offbtime'] = orgt['offbtime'] + arg.seconds
modt['downtime'] = orgt['downtime'] - arg.seconds
modt['uptime'] = orgt['uptime'] + arg.seconds
if orgt['rntime'] + arg.seconds > 0:
modt['rntime'] = orgt['rntime'] + arg.seconds
modt['slptime'] = orgt['slptime']
elif orgt['slptime'] + arg.seconds >= 0:
modt['rntime'] = orgt['rntime']
modt['slptime'] = orgt['slptime'] + arg.seconds
else:
modt['rntime'] = modt['uptime']
modt['slptime'] = 0
print('\t modified\tbtime: --n/a-- | uptime: ' + str(modt['uptime']) + ' | rntime: ' + str(modt['rntime']) + ' | slptime: ' + str(modt['slptime']) + ' | offbtime: ' + str(modt['offbtime']) + ' | downtime: ' + str(modt['downtime']))
# Limit check
if modt['downtime'] is None or modt['downtime'] < 0 or \
modt['offbtime'] is None or modt['offbtime'] < 0 or \
modt['uptime'] < 1 or modt['rntime'] < 1 or modt['slptime'] < 0:
logging.error('modified values can\'t be None or under limits')
sys.exit(1)
# Update values
conn.execute('update tuptime set uptime =? where rowid =?', (modt['uptime'], reg['target']))
conn.execute('update tuptime set rntime =? where rowid =?', (modt['rntime'], reg['target']))
conn.execute('update tuptime set slptime =? where rowid =?', (modt['slptime'], reg['target']))
conn.execute('update tuptime set downtime =? where rowid =?', (modt['downtime'], reg['target']))
conn.execute('update tuptime set offbtime =? where rowid =?', (modt['offbtime'], reg['target']))
def fix_startup(arg, reg, conn, modt, orgt, modp, orgp):
"""Modify startup datetime register"""
modt['btime'] = orgt['btime'] + arg.seconds
modt['uptime'] = orgt['uptime'] - arg.seconds
if orgt['rntime'] - arg.seconds > 0:
modt['rntime'] = orgt['rntime'] - arg.seconds
modt['slptime'] = orgt['slptime']
elif orgt['slptime'] - arg.seconds >= 0:
modt['rntime'] = orgt['rntime']
modt['slptime'] = orgt['slptime'] - arg.seconds
else:
modt['rntime'] = modt['uptime']
modt['slptime'] = 0
print('\t modified\tbtime: ' + str(modt['btime']) + ' | uptime: ' + str(modt['uptime']) + ' | rntime: ' + str(modt['rntime']) + ' | slptime: ' + str(modt['slptime']) + ' | offbtime: --n/a-- | downtime: --n/a-- ')
# Limit check
if modt['uptime'] < 1 or modt['rntime'] < 1 or modt['slptime'] < 0:
logging.error('modified values can\'t be under limits')
sys.exit(1)
# Get previous registers to modify except from first row
if reg['prev'] > 0:
# Get values from previous row
conn.execute('select downtime from tuptime where rowid =?', (reg['prev'],))
orgp['downtime'] = conn.fetchone()[0]
modp['downtime'] = orgp['downtime'] + arg.seconds
print('\tTarget row-1 \'' + str(reg['prev']) + '\':')
print('\t original\tdowntime: ' + str(orgp['downtime']))
print('\t modified\tdowntime: ' + str(modp['downtime']))
# Limit check
if modp['downtime'] < 0:
logging.error('downtime can\'t be lower than 0')
sys.exit(1)
conn.execute('update tuptime set downtime =? where rowid =?', (modp['downtime'], reg['prev']))
# Update values
conn.execute('update tuptime set btime =? where rowid =?', (modt['btime'], reg['target']))
conn.execute('update tuptime set uptime =? where rowid =?', (modt['uptime'], reg['target']))
conn.execute('update tuptime set rntime =? where rowid =?', (modt['rntime'], reg['target']))
conn.execute('update tuptime set slptime =? where rowid =?', (modt['slptime'], reg['target']))
def main():
arg = get_arguments()
orgt = {} # Original target value
orgp = {} # Original previous (target-1) value
modt = {} # Modified target value
modp = {} # Modified previous (target-1) value
reg = {'target': arg.register, 'prev': (arg.register - 1)} # Target register to modify and previous one
print('Target file:\t' + arg.db_file)
if not arg.nobackup:
backup_dbf(arg)
db_conn = sqlite3.connect(arg.db_file)
db_conn.row_factory = sqlite3.Row
db_conn.set_trace_callback(logging.debug)
conn = db_conn.cursor()
# Check if DB have the old format
columns = [i[1] for i in conn.execute('PRAGMA table_info(tuptime)')]
if 'rntime' and 'slptime' and 'bootid' not in columns:
logging.error('DB format outdated')
sys.exit(1)
# Print raw rows
if arg.verbose:
print('\nTarget registers before: ')
conn.execute('select rowid as startup, * from tuptime where rowid between ? and ?', (reg['prev'], reg['target']))
db_rows = conn.fetchall()
for row in db_rows:
print("\t", end='')
print(dict(row))
# Validate register to modify
conn.execute('select max(rowid) from tuptime')
max_register = conn.fetchone()[0]
if reg['target'] > max_register or reg['target'] < 1:
logging.error('Invalid register to modify. Out of range')
sys.exit(1)
# Get values from target row
conn.execute('select btime, uptime, rntime, slptime, offbtime, endst, downtime from tuptime where rowid =?', (reg['target'],))
orgt['btime'], orgt['uptime'], orgt['rntime'], orgt['slptime'], orgt['offbtime'], orgt['endst'], orgt['downtime'] = conn.fetchone()
print('\nValues:')
print('\tTarget row \'' + str(reg['target']) + '\':')
# Choose value to modify: endst / startup / shutdown
if arg.change == 'endst':
print('\t original\tendst: ' + str(orgt['endst']))
fix_endst(arg, reg, conn, modt, orgt)
else:
print('\t original\tbtime: ' + str(orgt['btime']) + ' | uptime: ' + str(orgt['uptime']) + ' | rntime: ' + str(orgt['rntime']) + ' | slptime: ' + str(orgt['slptime']) + ' | offbtime: ' + str(orgt['offbtime']) + ' | downtime: ' + str(orgt['downtime']))
if arg.change == 'startup':
fix_startup(arg, reg, conn, modt, orgt, modp, orgp)
if arg.change == 'shutdown':
fix_shutdown(arg, reg, conn, modt, orgt)
db_conn.commit()
# Print raw rows
if arg.verbose:
print('\nTarget registers after: ')
conn.execute('select rowid as startup, * from tuptime where rowid between ? and ?', (reg['prev'], reg['target']))
db_rows = conn.fetchall()
for row in db_rows:
print("\t", end='')
print(dict(row))
db_conn.close()
print('\nDone.')
if __name__ == "__main__":
main()
|
