コード例 #1
0
ファイル: create.py プロジェクト: gclawes/eecis-accounts
def main():
    # message to use for sending mail logs
    MESSAGE = ""

    MESSAGE += "MAKING ACCOUNTS ON %s FOR %s\n\n" % (HOST, DOMAIN)
    
    # get authentication modules
    modules = auth_modules.load_modules()
    
    # fancy python list comprehension to:
    # select users with the pending_create status that don't have
    # the DOMAIN_created flag for this domain
    pending_users = [u for u in User.query.filter_by(status='pending_create').all()
            if u.in_domain(DOMAIN) and "%s_created" % DOMAIN not in u.get_flags() ]
    usernames = [u.username for u in pending_users]

    # if no pending accounts, do nothing
    if pending_users == []: 
        sys.exit()

    MESSAGE += "New Accounts\n------------\n"

    # create login entries for each login method
    MESSAGE += "Creating logins for:\n%s\n\n" % ", ".join(usernames)
    for name, module in modules.iteritems():
        MESSAGE += module.create_logins(pending_users)
        MESSAGE += "\n\n"

    # call ZFS create scripts, and mail scripts on louie if it's research
    MESSAGE += "Creating %s ZFS filesystems for:\n%s\n\n" % (DOMAIN, ", ".join(usernames))
    for user in pending_users:
        MESSAGE += "DEBUG: zfs_create(user)\n"
        #zfs_create(user)
        if user.in_domain('research'):
            MESSAGE += "DEBUG: research_mail(user)\n"
            #research_mail(user)

    # Creating users is complete, add the flag for creation in the DB
    for user in pending_users:
        user.add_flag(DOMAIN+"_created")

    MESSAGE += "\n\nDONE MAKING ACCOUNTS ON %s FOR %s\n" % (HOST, DOMAIN)

    print MESSAGE 
    mail.send("*****@*****.**", "Super-User", MAILTO, \
            "New Accounts - %s" % DOMAIN, MESSAGE, MAILHOST)
コード例 #2
0
ファイル: remind.py プロジェクト: gclawes/eecis-accounts
def main():
    # Security checks
    # if this is run as root for some reason, drop to "account"
    if os.getuid() == 0:
        os.setgid(UID)
        os.setuid(UID)
        stderr.write("ERROR: remind.py must be run as root!")

    # very fancy list comprehensions to get dict of users 
    # by sponsor (maps sponsors to the users they sponsor)
    # collisions are not possible, a user can not be in multiple sponsor lists
    users = User.query.filter_by(status='pending_sponsor').all()
    sponsors = map(lambda s: User.query.get(s), {user.sponsor for user in users})
    users_by_sponsor = {sponsor : [user for user in users if user.sponsor == sponsor.username] for sponsor in sponsors}

    for sponsor, users in users_by_sponsor.iteritems(): # for key, value in dict.iteritems()
        subject, message = generate_email(users)
        mail.send("*****@*****.**", "ECE/CIS Labstaff", \
                "*****@*****.**" % sponsor.username,
                subject, message, MAILHOST)
コード例 #3
0
ファイル: create.py プロジェクト: gclawes/eecis-accounts
    # call ZFS create scripts, and mail scripts on louie if it's research
    MESSAGE += "Creating %s ZFS filesystems for:\n%s\n\n" % (DOMAIN, ", ".join(usernames))
    for user in pending_users:
        MESSAGE += "DEBUG: zfs_create(user)\n"
        #zfs_create(user)
        if user.in_domain('research'):
            MESSAGE += "DEBUG: research_mail(user)\n"
            #research_mail(user)

    # Creating users is complete, add the flag for creation in the DB
    for user in pending_users:
        user.add_flag(DOMAIN+"_created")

    MESSAGE += "\n\nDONE MAKING ACCOUNTS ON %s FOR %s\n" % (HOST, DOMAIN)

    print MESSAGE 
    mail.send("*****@*****.**", "Super-User", MAILTO, \
            "New Accounts - %s" % DOMAIN, MESSAGE, MAILHOST)

if __name__ == "__main__":
    if not os.path.isfile(LOCKFILE):
        os.mknod(LOCKFILE)
        main()
        os.remove(LOCKFILE)
    else:
        error = "ERROR: %s still exists, create.py is still running on %s\n\n" % (LOCKFILE, HOST)
        mail.send("*****@*****.**", "Super-User", MAILTO, \
                "New Accounts ERROR - %s" % DOMAIN, error, MAILHOST)


コード例 #4
0
ファイル: remind.py プロジェクト: gclawes/eecis-accounts
    # Security checks
    # if this is run as root for some reason, drop to "account"
    if os.getuid() == 0:
        os.setgid(UID)
        os.setuid(UID)
        stderr.write("ERROR: remind.py must be run as root!")

    # very fancy list comprehensions to get dict of users 
    # by sponsor (maps sponsors to the users they sponsor)
    # collisions are not possible, a user can not be in multiple sponsor lists
    users = User.query.filter_by(status='pending_sponsor').all()
    sponsors = map(lambda s: User.query.get(s), {user.sponsor for user in users})
    users_by_sponsor = {sponsor : [user for user in users if user.sponsor == sponsor.username] for sponsor in sponsors}

    for sponsor, users in users_by_sponsor.iteritems(): # for key, value in dict.iteritems()
        subject, message = generate_email(users)
        mail.send("*****@*****.**", "ECE/CIS Labstaff", \
                "*****@*****.**" % sponsor.username,
                subject, message, MAILHOST)

if __name__ == "__main__":
    if not os.path.isfile(LOCKFILE):
        os.mknod(LOCKFILE)
        main()
        os.remove(LOCKFILE)
    else:
        error = "ERROR: %s still exists, remind.py is still running on %s\n\n" % (LOCKFILE, HOST)
        mail.send("*****@*****.**", "Account-user", MAILTO, \
                "Sponsor Reminder ERROR - %s" % DOMAIN, error, MAILHOST)