Annotation of /rpms/denyhosts/contribs10/denyhosts-dh_reenable

#!/usr/bin/python
# This program help sysadmin to reenable a hosts blocked by denyhosts.
#
# dh_reenable (c) 2008 Marco Bertorello <marco@bertorello.ns0.it> and is
# free software. You can use, modify and redistribute it under terms of
# GNU General Public License version 2 or later, as you whish, as published by
# Free Software Foundation.
#
# You can get a full copy of license here:
#
# http://www.gnu.org/licenses/gpl-2.0.txt
#
# and
#
# http://www.gnu.org/licenses/gpl-3.0.txt


from __future__ import with_statement
import os
import sys
import fileinput
import re

# file definition:

HOSTSFILE='/var/lib/denyhosts/hosts'
HOSTRESTFILE='/var/lib/denyhosts/hosts-restricted'
HOSTROOTFILE='/var/lib/denyhosts/hosts-root'
HOSTVALIDFILE='/var/lib/denyhosts/hosts-valid'
HOSTSDENY='/etc/hosts.deny'
#TEST='/etc/hosts.deny.tmp'
CONFIGFILE='/etc/denyhosts.conf'

# Parse the configuration file for the location of the HOSTS_DENY file.
# If it exists, overwrite the hard-coded value for HOSTSDENY from the
# top of the file.
if os.path.isfile(CONFIGFILE):
    with open(CONFIGFILE) as file:
        for line in file:
            result = re.search('^(HOSTS_DENY\s*=\s*)(.*)', line)
            if result != None:
                HOSTSDENY = result.group(2)

def usage():
         print "Usage:"
         print sys.argv[0]+"     --help: Show this help"
         print sys.argv[0]+"       <IP>: check if the ip address was denied and reenable it"
         print sys.argv[0]+" <HOSTNAME>: check if the hostname was denied and reenable it"

try:
        host=sys.argv[1]
except:
        print sys.argv[0]+" need a hostname or a ip address input. See --help."
        sys.exit(1)

if host == "--help":
        usage()
        sys.exit(1)

def search(file_txt,host):
        for lines in fileinput.FileInput(file_txt, inplace=1):  
                lines = lines.strip()
                if lines.find(host) != -1: 
                        continue
                else:
                        print lines


try:
        search(HOSTSFILE,host)
except:
        print "Problem parsing file "+HOSTSFILE
        sys.exit(1)
try:
        search(HOSTRESTFILE,host)
except:
        print "Problem parsing file "+HOSTRESTFILE
        sys.exit(1)
try:
        search(HOSTROOTFILE,host)
except:
        print "Problem parsing file "+HOSTROOTFILE
        sys.exit(1)
try:
        search(HOSTVALIDFILE,host)
except:
        print "Problem parsing file "+HOSTVALIDFILE
        sys.exit(1)
try:
        search(HOSTSDENY,host)
except:
        print "Problem parsing file "+HOSTSDENY
        sys.exit(1)

print "Done!"
print "Please restart denyhosts"

1	jpp	1.1	#!/usr/bin/python
2			# This program help sysadmin to reenable a hosts blocked by denyhosts.
3			#
4			# dh_reenable (c) 2008 Marco Bertorello <marco@bertorello.ns0.it> and is
5			# free software. You can use, modify and redistribute it under terms of
6			# GNU General Public License version 2 or later, as you whish, as published by
7			# Free Software Foundation.
8			#
9			# You can get a full copy of license here:
10			#
11			# http://www.gnu.org/licenses/gpl-2.0.txt
12			#
13			# and
14			#
15			# http://www.gnu.org/licenses/gpl-3.0.txt
16
17
18			from __future__ import with_statement
19			import os
20			import sys
21			import fileinput
22			import re
23
24			# file definition:
25
26			HOSTSFILE='/var/lib/denyhosts/hosts'
27			HOSTRESTFILE='/var/lib/denyhosts/hosts-restricted'
28			HOSTROOTFILE='/var/lib/denyhosts/hosts-root'
29			HOSTVALIDFILE='/var/lib/denyhosts/hosts-valid'
30			HOSTSDENY='/etc/hosts.deny'
31			#TEST='/etc/hosts.deny.tmp'
32			CONFIGFILE='/etc/denyhosts.conf'
33
34			# Parse the configuration file for the location of the HOSTS_DENY file.
35			# If it exists, overwrite the hard-coded value for HOSTSDENY from the
36			# top of the file.
37			if os.path.isfile(CONFIGFILE):
38			with open(CONFIGFILE) as file:
39			for line in file:
40			result = re.search('^(HOSTS_DENY\s=\s)(.*)', line)
41			if result != None:
42			HOSTSDENY = result.group(2)
43
44			def usage():
45			print "Usage:"
46			print sys.argv[0]+" --help: Show this help"
47			print sys.argv[0]+" <IP>: check if the ip address was denied and reenable it"
48			print sys.argv[0]+" <HOSTNAME>: check if the hostname was denied and reenable it"
49
50			try:
51			host=sys.argv[1]
52			except:
53			print sys.argv[0]+" need a hostname or a ip address input. See --help."
54			sys.exit(1)
55
56			if host == "--help":
57			usage()
58			sys.exit(1)
59
60			def search(file_txt,host):
61			for lines in fileinput.FileInput(file_txt, inplace=1):
62			lines = lines.strip()
63			if lines.find(host) != -1:
64			continue
65			else:
66			print lines
67
68
69			try:
70			search(HOSTSFILE,host)
71			except:
72			print "Problem parsing file "+HOSTSFILE
73			sys.exit(1)
74			try:
75			search(HOSTRESTFILE,host)
76			except:
77			print "Problem parsing file "+HOSTRESTFILE
78			sys.exit(1)
79			try:
80			search(HOSTROOTFILE,host)
81			except:
82			print "Problem parsing file "+HOSTROOTFILE
83			sys.exit(1)
84			try:
85			search(HOSTVALIDFILE,host)
86			except:
87			print "Problem parsing file "+HOSTVALIDFILE
88			sys.exit(1)
89			try:
90			search(HOSTSDENY,host)
91			except:
92			print "Problem parsing file "+HOSTSDENY
93			sys.exit(1)
94
95			print "Done!"
96			print "Please restart denyhosts"
97