123 lines
3.9 KiB
Python
123 lines
3.9 KiB
Python
#!/usr/bin/env python
|
|
|
|
import psycopg, pwd, grp, smtplib, os
|
|
from email.MIMEText import MIMEText
|
|
import Settings, PasswordTools
|
|
|
|
class InvalidDomain(Exception):
|
|
"""This exception is raised if an invalid domain is used."""
|
|
def __init__(self, domain):
|
|
self.domain = domain
|
|
|
|
def __str__(self):
|
|
return repr("Invalid domain %s" % (self.domain))
|
|
|
|
class NoSysuserForDomain(Exception):
|
|
"""This exception is raised if no system user is associated with a domain."""
|
|
def __init__(self, domain):
|
|
self.domain = domain
|
|
|
|
def __str__(self):
|
|
return repr("No system user for domain %s" % (self.domain))
|
|
|
|
class Domain:
|
|
"""A Domain representation object with service methods."""
|
|
def __init__(self, domain):
|
|
self.cnx = psycopg.connect("user=%(dbuser)s password=%(dbpassword)s dbname=%(dbname)s" % Settings.dbsettings)
|
|
self.domain = domain
|
|
self.validate_domain()
|
|
|
|
def validate_domain(self):
|
|
"""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."""
|
|
cr = self.cnx.cursor()
|
|
|
|
cr.execute("SELECT * FROM domain WHERE domainname=%(name)s" %
|
|
{'name': psycopg.QuotedString(self.domain)})
|
|
self.cnx.commit()
|
|
|
|
result = cr.fetchall()
|
|
if (not result):
|
|
raise InvalidDomain(self)
|
|
|
|
def __str__(self):
|
|
return repr(self.domain)
|
|
|
|
def getSysuser(self):
|
|
"""Gets the system user id of the domain."""
|
|
cr = self.cnx.cursor()
|
|
|
|
cr.execute("""SELECT sysuser.name FROM domain, sysuser
|
|
WHERE domain.domainname=%(name)s
|
|
AND domain.sysuserid=sysuser.sysuserid""" %
|
|
{'name': psycopg.QuotedString(self.domain)})
|
|
self.cnx.commit()
|
|
|
|
result = cr.fetchall()
|
|
if (not result):
|
|
raise NoSysuserForDomain(self)
|
|
# return row 0, field 0
|
|
return result[0][0]
|
|
|
|
def getNextPopUser(self):
|
|
"""Gets the user id of the next available POP3/IMAP-user for the domain."""
|
|
cr = self.cnx.cursor()
|
|
|
|
sysuser = self.getSysuser()
|
|
|
|
cr.execute("""SELECT max(id) FROM mailpasswd WHERE
|
|
id LIKE %(username)s""" %
|
|
{'username': psycopg.QuotedString(sysuser+'%')})
|
|
self.cnx.commit()
|
|
|
|
result = cr.fetchall()
|
|
if (not result):
|
|
return sysuser + "p01"
|
|
|
|
maxpopuser = result[0][0]
|
|
num = int(maxpopuser[len(sysuser)+1:])+1
|
|
return "%sp%d" % (sysuser, num)
|
|
|
|
def makePopUser(self, password):
|
|
"""Creates a new POP3/IMAP-user for the domain using the given password."""
|
|
cr = self.cnx.cursor()
|
|
|
|
sysuser = self.getSysuser()
|
|
popaccount = self.getNextPopUser()
|
|
crypted = PasswordTools.md5_crypt_password(password)
|
|
uid = pwd.getpwnam(sysuser)[2]
|
|
gid = grp.getgrnam(Settings.popgroup)[2]
|
|
homedir = Settings.pophome + popaccount
|
|
|
|
os.mkdir(homedir, 0755)
|
|
os.system("maildirmake \"%s/Maildir\"" % (homedir))
|
|
os.system("chown -R %s.%s %s" % ( sysuser, Settings.popgroup, homedir ))
|
|
|
|
cr = self.cnx.cursor()
|
|
cr.execute("""INSERT INTO mailpasswd (id, crypt, clear, uid, gid, home)
|
|
VALUES (%(id)s, %(crypt)s, %(clear)s, %(uid)d, %(gid)d, %(home)s)""" % {
|
|
'id': psycopg.QuotedString(popaccount),
|
|
'crypt': psycopg.QuotedString(crypted),
|
|
'clear': psycopg.QuotedString(password),
|
|
'uid': uid, 'gid': gid,
|
|
'home': psycopg.QuotedString(homedir)})
|
|
self.cnx.commit()
|
|
|
|
text = """A new POP3/IMAP account has been created
|
|
Domain: %(domain)s
|
|
User: %(user)s
|
|
Password: %(password)s""" % {'domain': self.domain,
|
|
'user': popaccount,
|
|
'password': password}
|
|
themail = MIMEText(text)
|
|
themail['Subject'] = "A new POP3/IMAP account has been created"
|
|
themail['From'] = Settings.mailsender
|
|
themail['To'] = Settings.mailreceiver
|
|
|
|
s = smtplib.SMTP()
|
|
s.connect()
|
|
s.sendmail(Settings.mailsender, [Settings.mailreceiver], themail.as_string())
|
|
s.close()
|