joshtronic/tmpufw
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added --status flag

Browse source
This commit is contained in:
Josh Sherman 2013-10-23 23:07:53 -04:00
parent 96648fb212
commit 00ea5b516e
1 changed files with 21 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

26
tmpufw
View file

@ -21,6 +21,7 @@ __status__ = 'Production'
__version__ = '1.0.0' __version__ = '1.0.0'
from argparse import ArgumentParser from argparse import ArgumentParser
from datetime import datetime
from os import getpid, makedirs, path, remove from os import getpid, makedirs, path, remove
from parsedatetime import Calendar from parsedatetime import Calendar
from shutil import move from shutil import move
@ -45,7 +46,22 @@ class tmpufw(object):
tmp_rules_file = '/tmp/' + __file__ + '-rules' tmp_rules_file = '/tmp/' + __file__ + '-rules'
if args.status: if args.status:
exit('TODO display rules and expirations') if path.exists(rules_file):
try:
print("Expiration\t\tRule")
print('=' * 80)
# Loops through the rules lines
for line in open(rules_file, 'r'):
# Breaks apart line into expiration timestamp and rule
timestamp, rule = line.strip("\n").split(' ', 1)
timestamp, decimal = timestamp.split('.')
print(str(datetime.fromtimestamp(int(timestamp))) + "\t" + rule)
except IOError:
self.error('unable to read from the rules file: ' + rules_file)
else:
self.error('there are no rules to display')
elif args.clean: elif args.clean:
# Checks for PID file # Checks for PID file
if not path.exists(pid_file): if not path.exists(pid_file):
@ -55,7 +71,7 @@ class tmpufw(object):
handle.write(str(getpid())) handle.write(str(getpid()))
handle.close(); handle.close();
except IOError: except IOError:
self.error('Unable to create PID file: ' + pid_file) self.error('unable to create PID file: ' + pid_file)
# Checks for the rules file # Checks for the rules file
if path.exists(rules_file): if path.exists(rules_file):
@ -63,7 +79,7 @@ class tmpufw(object):
try: try:
handle = open(tmp_rules_file, 'a') handle = open(tmp_rules_file, 'a')
except IOError: except IOError:
self.error('Unable to write to the tmp rules file: ' + tmp_rules_file) self.error('unable to write to the tmp rules file: ' + tmp_rules_file)
try: try:
current_time = time() current_time = time()
@ -87,7 +103,7 @@ class tmpufw(object):
# Moves the tmp file to the rules file # Moves the tmp file to the rules file
move(tmp_rules_file, rules_file) move(tmp_rules_file, rules_file)
except IOError: except IOError:
self.error('Unable to read rule file: ' + rules_file) self.error('unable to from the read rules file: ' + rules_file)
# Removes the PID # Removes the PID
remove(pid_file) remove(pid_file)
@ -110,7 +126,7 @@ class tmpufw(object):
handle.write("\n") handle.write("\n")
handle.close() handle.close()
except IOError: except IOError:
self.error('Unable to write to the rules file: ' + rules_file) self.error('unable to write to the rules file: ' + rules_file)
# Attempts to add the rule to `ufw` # Attempts to add the rule to `ufw`
try: try: