def handle(self, *args, **options):

        creator_msg = "load_host_activity_test_data"
        (conninfo,
         creds) = qlogin(settings.QUMULO_devcluster['ipaddr'], 'admin',
                         settings.QUMULO_devcluster['adminpassword'], 8000)
        if not conninfo:
            msg = "could not connect to dev cluster  ... exiting"
            logger.critical(msg)
            sys.exit(-1)

        qr = Cluster.objects.filter(name=settings.QUMULO_devcluster['name'])
        if qr.count() == 0:
            cluster = Cluster(
                name=settings.QUMULO_devcluster['name'],
                ipaddr=settings.QUMULO_devcluster['ipaddr'],
                adminpassword=settings.QUMULO_devcluster['adminpassword'],
                port=8000)
            cluster.save()
        else:
            cluster = qr[0]

        data = ''
        filename = options['filename']
        for line in fileinput.input(filename):
            data = data + line
        data = json.loads(data)
        sample = request.RestResponse(data, 'etag')
        dt = datetime.datetime.utcnow()
        dt = dt.replace(tzinfo=pytz.UTC)
        cluster.load_activity_sample(conninfo=conninfo,
                                     creds=creds,
                                     sample=sample,
                                     validtime=dt)
예제 #2
0
    def handle(self, *args, **options):

        creator_msg = "load_host_activity_test_data"
        (conninfo,
         creds) = qlogin(settings.QUMULO_devcluster['ipaddr'], 'admin',
                         settings.QUMULO_devcluster['adminpassword'], 8000)
        if not conninfo:
            msg = "could not connect to dev cluster  ... exiting"
            logger.critical(msg)
            sys.exit(-1)

        qr = Cluster.objects.filter(name=settings.QUMULO_devcluster['name'])
        if qr.count() == 0:
            cluster = Cluster(
                name=settings.QUMULO_devcluster['name'],
                ipaddr=settings.QUMULO_devcluster['ipaddr'],
                adminpassword=settings.QUMULO_devcluster['adminpassword'],
                port=8000)
            cluster.save()
        else:
            cluster = qr[0]

        filename = options['filename']
        numdays = options['days_in_dataset']
        for d in range(int(numdays), 0, -1):
            data = ''
            for line in fileinput.input(filename):
                if "rate" in str(line):
                    # "rate": 1,
                    val = str(line).split(':')
                    val = val[1].replace(",", "")
                    val = float(val) * random.random()
                    line = '"rate": ' + str(val) + ","
                data = data + line
            data = json.loads(data)
            sample = request.RestResponse(data, 'etag')
            dt = datetime.datetime.utcnow()
            dt = dt.replace(tzinfo=pytz.UTC)
            dt = dt - datetime.timedelta(days=d)
            cluster.load_activity_sample(conninfo=conninfo,
                                         creds=creds,
                                         sample=sample,
                                         validtime=dt)