- separated password and domain logic into package gvadm

git-svn-id: file:///home/www/usr01/svn/gnuviechadmin/gnuviech.info/gnuviechadmin/trunk@121 a67ec6bc-e5d5-0310-a910-815c51eb3124

backend/addpopuser.py

@@ -1,66 +1,8 @@
 #!/usr/bin/env python
 
-import psycopg, crypt, crack, popen2, getopt, random, sys
+import psycopg, getopt, sys
 
-def generate_password():
+from gvadm import PasswordTools, DomainTools
-	(o, i, e) = popen2.popen3("apg -n 1 -m 8 -x 12 -a 0")
-
-	return "".join(o.readlines()).strip()
-
-def check_password(password):
-	try:
-		return crack.VeryFascistCheck(password)
-	except ValueError, ve:
-		print "Weak password:", ve
-		return None
-
-def md5_crypt_password(password):
-	salt = "".join([chr(letter) for letter in random.sample(range(ord('a'), ord('z')), 8)])
- 
-	return crypt.crypt(password, '$1$' + salt)
-
-class InvalidDomain(Exception):
-	def __init__(self, domain):
-		self.domain = domain
-	
-	def __str__(self):
-		return repr("Invalid domain %s" % (domain))
-
-def validate_domain(domain):
-	cnx = psycopg.connect("user=exim4 password=CotOgigmeIk5 dbname=gnuviechadmin")
-	cr = cnx.cursor()
-
-	cr.execute("SELECT * FROM domain WHERE domainname=%(name)s" %
-	           {'name': psycopg.QuotedString(domain)})
-	cnx.commit()
-
-	result = cr.fetchall()
-	if (not result):
-		raise InvalidDomain(domain)
-#
-#print "-- Information from mailalias --"
-#cr.execute('SELECT * FROM mailalias')
-#cnx.commit()
-#
-#print cr.description
-#
-#result = cr.fetchall()
-#for line in result:
-#	print line
-#
-#print "-- Information from mailpasswd --"
-#cr.execute('SELECT * FROM mailpasswd')
-#cnx.commit()
-#result = cr.fetchall()
-#for line in result:
-#	print line
-#
-#print "-- Domains --"
-#cr.execute('SELECT DISTINCT domain FROM mailalias')
-#cnx.commit()
-#result = cr.fetchall()
-#for line in result:
-#	print line
 
 def usage():
   print """Usage information:
@@ -93,8 +35,8 @@ if __name__ == "__main__":
   # specify the domain
   domain = dict(options[0])['--domain']
   try:
-		validate_domain(domain)
+    DomainTools.validate_domain(domain)
-	except InvalidDomain, iv:
+  except DomainTools.InvalidDomain, iv:
     print iv
     sys.exit(1)
 
@@ -102,11 +44,11 @@ if __name__ == "__main__":
   password = None
 
   if dict(options[0]).has_key('--password'):
-		password = check_password(dict(options[0])['--password'])
+    password = PasswordTools.check_password(dict(options[0])['--password'])
   if (password == None):
-		password = generate_password()
+    password = PasswordTools.generate_password()
   
   print domain
   print password
 
-	print md5_crypt_password(password)
+  print PasswordTools.md5_crypt_password(password)

backend/gvadm/DomainTools.py

@@ -0,0 +1,30 @@
+#!/usr/bin/env python
+
+import psycopg
+import Settings
+
+class InvalidDomain(Exception):
+  """This Exception is thrown if an invalid domain is used."""
+  def __init__(self, domain):
+    self.domain = domain
+  
+  def __str__(self):
+    return repr("Invalid domain %s" % (self.domain))
+
+def validate_domain(domain):
+  """This function validates whether the given domain is allowed.
+  That means that the domain needs to be registered in the database.
+  
+  If the domain is invalid InvalidDomain is raised."""
+  cnx = psycopg.connect("user=%(dbuser)s password=%(dbpassword)s dbname=%(dbname)s" % Settings.dbsettings)
+  cr = cnx.cursor()
+
+  cr.execute("SELECT * FROM domain WHERE domainname=%(name)s" %
+             {'name': psycopg.QuotedString(domain)})
+  cnx.commit()
+
+  result = cr.fetchall()
+  if (not result):
+    raise InvalidDomain(domain)
+
+

backend/gvadm/PasswordTools.py

@@ -0,0 +1,25 @@
+#!/usr/bin/env python
+
+import crypt, crack, popen2, random
+
+def generate_password():
+	(o, i, e) = popen2.popen3("apg -n 1 -m 8 -x 12 -a 0")
+
+	return "".join(o.readlines()).strip()
+
+def check_password(password):
+	try:
+		return crack.VeryFascistCheck(password)
+	except ValueError, ve:
+		print "Weak password:", ve
+		return None
+
+def md5_crypt_password(password):
+	salt = "".join([chr(letter) for letter in random.sample(range(ord('a'), ord('z')), 8)])
+ 
+	return crypt.crypt(password, '$1$' + salt)
+
+if __name__ == '__main__':
+	print check_password("test")
+	print generate_password()
+	print md5_crypt_password("test")

backend/gvadm/Settings.py

@@ -0,0 +1,5 @@
+#!/usr/bin/env python
+
+dbsettings = { 'dbuser': 'exim4',
+               'dbpassword' : 'CotOgigmeIk5',
+	       'dbname' : 'gnuviechadmin' }

backend/gvadm/__init__.py

@@ -0,0 +1,6 @@
+#!/usr/bin/env python
+
+"""This package contains classes for the gnuviech administration tool
+backend."""
+
+