Skip to content

Commit

Permalink
various fix
Browse files Browse the repository at this point in the history
  • Loading branch information
Michio Honda committed Nov 4, 2020
1 parent e868551 commit da6cd64
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 4 deletions.
1 change: 1 addition & 0 deletions tm-cleanup.service
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ Type=oneshot
ExecStart=/usr/local/bin/tm reservation clean
User=root
ProtectSystem=strict
ReadWritePaths=/var/lib/tftpboot
ReadWritePaths=/usr/local/tm
Nice=19
IOSchedulingClass=best-effort
Expand Down
11 changes: 7 additions & 4 deletions tm.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,6 +83,7 @@ def reservation(self, argv):
continue
if (args.noexpire or
today > datetime.strptime(v['expire'], dtfmt).date()):
self.power(split('tm power poweroff {}'.format(v['node'])))
self.reset_node(v['node'], v['mac'])
return

Expand All @@ -106,19 +107,20 @@ def reservation(self, argv):

if args.func == 'reserve' or args.func == 'update':
try:
dt = datetime.strptime(args.expire, dtfmt)
dt = datetime.strptime(args.expire, dtfmt).date()
except(ValueError):
self.log('expiration date format must be dd/mm/yy'
', not {}'.format(args.expire))
return
if dt < datetime.now():
today = datetime.now().date()
if dt < today:
self.log('date must be on or later than today')
return
else:
latest = datetime.now() + timedelta(days=MAXDAYS)
latest = (datetime.now() + timedelta(days=MAXDAYS)).date()
if dt > latest:
dt = latest
self.log('14 days of the maximum reservation is set')
print('14 days of the maximum reservation is set')

if args.func == 'reserve':
if not self.set_loader(r['mac'], self.user, args.node):
Expand All @@ -128,6 +130,7 @@ def reservation(self, argv):
self.db.update({'user': getpass.getuser(),
'expire': dt.strftime(dtfmt)}, Query().node == args.node)
else:
self.power(split('tm power poweroff {}'.format(args.node)))
self.reset_node(args.node, r['mac'])
self.log('{}: {} successful'.format(args.node, args.func))

Expand Down

0 comments on commit da6cd64

Please sign in to comment.