Exemplo n.º 1
0
if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='generates .html from database')
    parser.add_argument('--database', default=config.DATABASE, help='sqlite database')
    parser.add_argument('--verbose', action='store_true', default=False, help='verbose infos')
    args = parser.parse_args()

    # setup logging
    logging.basicConfig(
        stream=sys.stdout,
        level=logging.DEBUG if args.verbose else logging.INFO,
        format=config.LOG_FORMAT,
        datefmt=config.LOG_DATETIME_FORMAT
    )

    logging.info('setup_db...')
    setup_db(args.database)

    datas = []
    for entry in UrandomizationStorage().get_urandomizations():
        datas.append({
            "startTime": str(entry.start_time),
            "stopTime": str(entry.stop_time),
            "totalTime": str(entry.stop_time - entry.start_time),
            "byteCount": entry.byte_count,
            "ipAddress": entry.ip_address
        })

    html_doc = """<!DOCTYPE html>
<html>
    <head>
        <title>Urandomizater</title>
Exemplo n.º 2
0
    logging.basicConfig(
        filename=args.logfile,
        filemode='w',
        level=logging.DEBUG,
        format=config.LOG_FORMAT,
        datefmt=config.LOG_DATETIME_FORMAT
    )

    # console logging
    console = logging.StreamHandler(sys.stdout)
    console.setLevel(logging.DEBUG if args.verbose else logging.INFO)
    console.setFormatter(logging.Formatter(config.LOG_FORMAT, datefmt=config.LOG_DATETIME_FORMAT))
    logging.getLogger().addHandler(console)

    logger = logging.getLogger("Urandomizer")
    logger.info('setup db...')
    setup_db(config.DATABASE)

    logger.info('creating server...')
    server = UrandomServer(args.host, args.port)
    logger.info('starting server...')
    server.start()
    logger.info('serve forever... ctrl+c to stop')
    while True:
        try:
            time.sleep(0.5)
        except KeyboardInterrupt:
            logger.info('hit ctrl+c')
            break
    server.stop()