Exemplo n.º 1
0
    def __init__(self, config):
        self.config = config
        self.database = DemoData.get_session(self.config)

        # Security Email
        if self.config.security_type == "email":
            self.mail = DemoMail(
                host=self.config.mail_host,
                port=self.config.mail_port,
                from_mail=self.config.mail_from,
                user=self.config.mail_user,
                password=self.config.mail_password,
                tls=self.config.mail_tls,
            )
        else:
            self.mail = None

        # Security Auth
        if self.config.security_type.startswith('auth_'):
            path = os.path.join(os.path.dirname(__file__), "auth")
            mod_name = self.config.security_type
            auth_mod = __import__(
                'auth.'+mod_name,
                globals(), locals(),
                fromlist=[mod_name]
            )
            auth_class = getattr(auth_mod, Demo.get_class_name(mod_name))
            self.auth = auth_class(self.config.auth)
        else:
            self.auth = False

        # Provider
        path = os.path.join(os.path.dirname(__file__), "provider")
        prov_name = 'prov_'+self.config.provider
        prov_mod = __import__(
            'provider.'+prov_name,
            globals(), locals(),
            fromlist=[prov_name]
        )
        prov_class = getattr(
            prov_mod, Demo.get_class_name(self.config.provider)
        )
        self.provider = prov_class(self.config.provider_data)
Exemplo n.º 2
0
def cli_entrypoint():
    parser = argparse.ArgumentParser()
    parser.add_argument('-c', help='config file', default='./config.ini')
    args = parser.parse_args()

    DemoConfig.config_file = args.c
    config = DemoConfig()

    # create table
    session = DemoData.get_session(config)
    session.close()

    logging.basicConfig(level=config.log_level)
    vacuum = Vacuum()
    pool = Pool()
    try:
        vacuum.start()
        pool.start()
        server = ThreadedHTTPServer(('0.0.0.0', config.http_port), Handler)
        server.serve_forever()
    except (KeyboardInterrupt, SystemExit):
        logging.info("Exit signal catched")
        vacuum.stop = True