lib: lxcf-erase-setup clean up and bug fixes

author MUNEDA Takahiro <mux003@gmail.com>

Mon, 3 Mar 2014 03:16:19 +0000 (22:16 -0500)

committer Hideyuki Niwa <niwa.hideyuki@jp.fujitsu.com>

Mon, 3 Mar 2014 06:44:06 +0000 (15:44 +0900)
author MUNEDA Takahiro <mux003@gmail.com>
Mon, 3 Mar 2014 03:16:19 +0000 (22:16 -0500)
committer Hideyuki Niwa <niwa.hideyuki@jp.fujitsu.com>
Mon, 3 Mar 2014 06:44:06 +0000 (15:44 +0900)
diff --git a/lxcf/lib/lxcf-erase-setup b/lxcf/lib/lxcf-erase-setup

index 772d6c7..e01625f 100755 (executable)
--- a/lxcf/lib/lxcf-erase-setup
+++ b/lxcf/lib/lxcf-erase-setup
@@ -5,24 +5,27 @@
  # modify it under the terms of the GNU General Public License
  # as published by the Free Software Foundation; version 2
  # of the License.
-# 
+#
  # This program is distributed in the hope that it will be useful,
  # but WITHOUT ANY WARRANTY; without even the implied warranty of
  # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  # GNU General Public License for more details.
-# 
+#
  # You should have received a copy of the GNU General Public License
  # along with this program; if not, write to the Free Software
-# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  # 02110-1301, USA.
  
  # -*- coding: utf-8 -*-
  
-import sys, os, shutil
+import sys
+import os
  import fcntl
  import IPy
-from optparse import OptionParser,OptionGroup
+import ConfigParser
+from optparse import OptionParser
  
+CONFIG_FILE = '/etc/lxcf/lxcf.conf'
  HOSTS_FILE = '/etc/hosts'
  
  parser = OptionParser()
@@ -30,9 +33,9 @@ parser = OptionParser()
  
  argc = len(args)
  
-if (argc != 2) :
-  print "usage : lxcf-erase-setup  LXCNAME"
-  quit()
+if (argc != 2):
+    print "usage : lxcf-erase-setup  LXCNAME"
+    quit()
  
  domain_name = args[1]
  
@@ -40,133 +43,125 @@ domain_name = args[1]
  # Function for IP address
  #
  
-#
-# erase IP addres from /etc/hosts
-#
  
-def erase_ipaddr(name, hosts_file) :
-  # read hosts file and check ip addr 
-  try : 
-    f = open(hosts_file,"r")
-  except :
-    return
+def erase_ipaddr(name, hosts_file):
+    """erase IP addres from /etc/hosts"""
+    # read hosts file and check ip addr
+    try:
+        f = open(hosts_file, "r")
+    except:
+        return
  
-  Lines = f.readlines()
+    Lines = f.readlines()
  
-  f = open(hosts_file, "w")
-  fcntl.flock(f.fileno(), fcntl.LOCK_EX)
+    f = open(hosts_file, "w")
+    fcntl.flock(f.fileno(), fcntl.LOCK_EX)
  
-  for l in Lines :
-    S = l.split()
-    if (len(S) >= 1) :
-      if ((S[0])[0] == '#') :
-        continue
-      if (name != S[1]) :
-        f.write(l)
+    for l in Lines:
+        S = l.split()
+        if (len(S) >= 1):
+            if ((S[0])[0] == '#'):
+                continue
+            if (name != S[1]):
+                f.write(l)
  
-  # unlock hosts file
-  fcntl.flock(f.fileno(), fcntl.LOCK_UN)
+    # unlock hosts file
+    fcntl.flock(f.fileno(), fcntl.LOCK_UN)
  
-  f.close()
+    f.close()
  
-#
-# find IP addres from /etc/hosts
-#
  
-def find_ipaddr(name, hosts_file) :
-  # read hosts file and check ip addr 
-  try :
-    f = open(hosts_file,"r")
-  except :
-    return IPy.IP(1)
+def find_ipaddr(name, hosts_file):
+    """find IP addres from /etc/hosts"""
+    # read hosts file and check ip addr
+    try:
+        f = open(hosts_file, "r")
+    except:
+        return IPy.IP(1)
+
+    while True:
+        Line = f.readline()
+        if (Line):
+            S = Line.split()
  
-  while True :
-      Line = f.readline()
-      if (Line) :
-        S = Line.split()
+            if (len(S) >= 1):
+                if ((S[0])[0] == '#'):
+                    continue
  
-        if (len(S) >= 1) :
-          if ((S[0])[0] == '#') :
-             continue
+                try:
+                    ip_s = IPy.IP(S[0])
+                except:
+                    continue
  
-          try:
-            ip_s = IPy.IP(S[0])
-          except:
-            continue 
+                if (name == S[1]):
+                    f.close()
+                    return ip_s
+        else:
+            break
  
-          if (name == S[1]) :
-            f.close()
-            return ip_s
-      else:
-        break
+    return IPy.IP(1)
  
-  return IPy.IP(1)
  
-#
-# check IP address in /etc/hosts
-#
+def check_ipaddr(ip, hosts_file):
+    """check IP address in /etc/hosts"""
+    # read hosts file and check ip addr
+    try:
+        f = open(hosts_file, "r")
+    except:
+        return False
+
+    while True:
+        Line = f.readline()
+        if (Line):
+            S = Line.split()
  
-def check_ipaddr(ip, hosts_file) :
-  # read hosts file and check ip addr 
-  try :
-    f = open(hosts_file,"r")
-  except :
-    return False
-
-  while True :
-      Line = f.readline()
-      if (Line) :
-        S = Line.split()
-
-        if (len(S) >= 1) :
-          if ((S[0])[0] == '#') :
-             continue
-
-          try:
-            ip_s = IPy.IP(S[0])
-          except:
-            continue 
-
-          if (ip_s == ip) :
-            f.close()
+            if (len(S) >= 1):
+                if ((S[0])[0] == '#'):
+                    continue
+
+                try:
+                    ip_s = IPy.IP(S[0])
+                except:
+                    continue
+
+            if (ip_s == ip):
+                f.close()
              return False
-      else:
-        break
+        else:
+            break
  
-  return True
+    return True
  
-#
-# get new IP address 
-#
  
  def get_ipaddr(hosts_file):
-  # read the config file
-  conf = ConfigParser.SafeConfigParser()
-  conf.read(CONFIG_FILE)
+    """get new IP address"""
+    # read the config file
+    conf = ConfigParser.SafeConfigParser()
+    conf.read(CONFIG_FILE)
  
-  ipaddr_start = IPy.IP(conf.get('ipaddr_range', 'ipaddr_start'))
-  ipaddr_end = IPy.IP(conf.get('ipaddr_range', 'ipaddr_end'))
+    ipaddr_start = IPy.IP(conf.get('ipaddr_range', 'ipaddr_start'))
+    ipaddr_end = IPy.IP(conf.get('ipaddr_range', 'ipaddr_end'))
  
-  ip1 = ipaddr_start.int()
-  ip2 = ipaddr_end.int()
+    ip1 = ipaddr_start.int()
+    ip2 = ipaddr_end.int()
  
-  # looks for empty IP.address  
-  for i in range(ip1, ip2+1) :
-    ipa = IPy.IP(i)
-    if (check_ipaddr(ipa, hosts_file)) :
-       break
+    # looks for empty IP.address
+    for i in range(ip1, ip2+1):
+        ipa = IPy.IP(i)
+        if (check_ipaddr(ipa, hosts_file)):
+            break
  
-  try:
-    with open(hosts_file, 'a') as f :
-      f.write(ipa.strNormal()+'\t'+domain_name+'\n')
-  except:
-      pass
+    try:
+        with open(hosts_file, 'a') as f:
+            f.write(ipa.strNormal()+'\t'+domain_name+'\n')
+    except:
+        pass
  
-  return ipa
+    return ipa
  
  erase_ipaddr(domain_name, HOSTS_FILE)
-for hfile in os.listdir('/opt/lxcf/') :
-  hf = '/opt/lxcf/'+hfile+'/etc/hosts'
-  erase_ipaddr(domain_name, hf)
+for hfile in os.listdir('/opt/lxcf/'):
+    hf = '/opt/lxcf/'+hfile+'/etc/hosts'
+    erase_ipaddr(domain_name, hf)
  
  print "erase ip address from /etc/hosts"
author	MUNEDA Takahiro <mux003@gmail.com>
	Mon, 3 Mar 2014 03:16:19 +0000 (22:16 -0500)
committer	Hideyuki Niwa <niwa.hideyuki@jp.fujitsu.com>
	Mon, 3 Mar 2014 06:44:06 +0000 (15:44 +0900)