Exemplo n.º 1
0
    def _initMailer(self):
        """Initialise :class:`~utils.Mailer` object and connect to SMTP server
        """
        if not self.config.smtp_host:
            raise RuntimeError("No SMTP server specified\n")

        if not self.config.smtp_user:
            raise RuntimeError("Username for SMTP server not specified\n")

        if not self.config.sender:
            self.warn("Sender email address not specified! Using username instead\n")
            self.config.sender= self.config.smtp_user

        self.mailer= Mailer( logStream= self.config.logStream,
                             verbose= self.config.verbose )

        self.log("Reading input message from {0}...\n"
                 .format(self.config.inputFiles[0]), verbose=1 )
                 
        with io.open(self.config.inputFiles[0]) as ifile:
            self.mailer.setMessage( "\n".join(ifile) )
        

        self.mailer.connect( self.config.smtp_host,
                             self.config.smtp_user,
                             self.config.smtp_password )
Exemplo n.º 2
0
class UpdateUsers(ToolBase):
    """Create user accounts for web interface
    
    Arguments:
        parent (:class:`~.tools.ToolBase`): Parent tool
    """
    
    def __init__(self, parent):
        description=str("Creates a user account for each pilot in database"
        "One input file containing the message to be sent to each user shall"
        "be provided. To personalise the message, the fields ${first_name}, "
        "${last_name}, ${email}, ${username} and ${password} can be used.")
        
        super(UpdateUsers, self).__init__( description= description,
                                           parent=parent )

        self.config= self.defaultConfiguration(self.parent.config)
        self._initCmdLineArguments()        
        
        #internal variables
        self.mailer= None


    def _exec(self):
        """Execute the tool
        """
        self.parent.connectDatabase()
        self._initMailer()
        
        self.createUserAccounts()        
        

    def _initCmdLineArguments(self):
        """Initialise all command line arguments.
        """
        
        self.parser.add_argument("inputFiles", nargs=1)
                                  
        self.parser.add_argument("-c", "--club",
                                 help="Restrict records to pilots of the given"
                                      " club",
                                 default= self.config.club)

        self.parser.add_argument("-H", "--smtp-host",
                                 help="SMTP server address",
                                 default= self.config.smtp_host)

        self.parser.add_argument("-U", "--smtp-user",
                                 help="username for login to SMTP server",
                                 default= self.config.smtp_user)

        self.parser.add_argument("-P", "--smtp-password",
                                 help="password for login to SMTP server",
                                 default= self.config.smtp_password)
    
        self.parser.add_argument("-s", "--subject",
                                 help="subject used for Email messages",
                                 default= self.config.subject)

        self.parser.add_argument("-S", "--sender",
                                 help="Senders email address",
                                 default= self.config.sender)


    def _initMailer(self):
        """Initialise :class:`~utils.Mailer` object and connect to SMTP server
        """
        if not self.config.smtp_host:
            raise RuntimeError("No SMTP server specified\n")

        if not self.config.smtp_user:
            raise RuntimeError("Username for SMTP server not specified\n")

        if not self.config.sender:
            self.warn("Sender email address not specified! Using username instead\n")
            self.config.sender= self.config.smtp_user

        self.mailer= Mailer( logStream= self.config.logStream,
                             verbose= self.config.verbose )

        self.log("Reading input message from {0}...\n"
                 .format(self.config.inputFiles[0]), verbose=1 )
                 
        with io.open(self.config.inputFiles[0]) as ifile:
            self.mailer.setMessage( "\n".join(ifile) )
        

        self.mailer.connect( self.config.smtp_host,
                             self.config.smtp_user,
                             self.config.smtp_password )
        

    def createUserAccounts(self):
        """Create user accounts and sent emails.
        """
        clubMessage="for all clubs"
        if self.config.club:
            clubMessage="for club '{0}'".format(self.config.club)
            
        self.log("Creating user accounts {0} ...\n".format(clubMessage),
                 verbose=1)
        
        for pilot, user, pwd in self.parent.db.createUsersFromPilots():
       
            self.log("Creating account for pilot {0} ...\n".format(pilot),
                      verbose=3)

            if self.config.club and self.config.club.lower() != pilot.club.lower():
                self.log("Skipped (account exists)\n", verbose=3)
                continue
            
            pilot.email= pilot.getCommentField("email")
            pilot.username= user.username
            
            if not pilot.email:
                self.warn("Could not create account for {0}: No email found\n"
                          .format(pilot) )
                continue
            
            pilot.password= pwd            
            self.log("Sending notification email to {0} ...\n"
                     .format(pilot.email),
                     verbose=3)
            self.mailer( recipients=[pilot],
                         subject= self.config.subject,
                         sender= self.config.sender )
            
            self.log("Adding user '{0}' to database ...\n"
                     .format(user.username),
                     verbose=3)

            self.parent.db.insertUsers([user])
            self.log("Successfully added user {0}\n".format(user.username),
                     verbose=2)        
        
        self.parent.db.commit()


    @staticmethod
    def defaultConfiguration(config=ToolBase.defaultConfiguration()):
        """Get default configuration options
        
        Arguments:
            config (object): Input configuration. Existing attributes will be
               overwritten.
        
        Return:
            Default configuration object
        """
        config.club         = None
        config.smtp_host    = "smtp.gmail.com"
        config.smtp_user    = None
        config.smtp_password= None
        config.subject      = "Your new Startkladde Account"
        config.sender       = None

        return config